feat(scanner): Complete PoE implementation with Windows compatibility fix
- Fix namespace conflicts (Subgraph → PoESubgraph) - Add hash sanitization for Windows filesystem (colon → underscore) - Update all test mocks to use It.IsAny<>() - Add direct orchestrator unit tests - All 8 PoE tests now passing (100% success rate) - Complete SPRINT_3500_0001_0001 documentation Fixes compilation errors and Windows filesystem compatibility issues. Tests: 8/8 passing Files: 8 modified, 1 new test, 1 completion report 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -12,7 +12,6 @@ using StellaOps.Attestor;
|
||||
using StellaOps.Scanner.Core.Configuration;
|
||||
using StellaOps.Scanner.Core.Contracts;
|
||||
using StellaOps.Scanner.Reachability;
|
||||
using StellaOps.Attestor;
|
||||
using StellaOps.Scanner.Worker.Orchestration;
|
||||
using StellaOps.Scanner.Worker.Processing;
|
||||
using StellaOps.Scanner.Worker.Processing.PoE;
|
||||
@@ -47,7 +46,7 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
);
|
||||
|
||||
_configMonitorMock = new Mock<IOptionsMonitor<PoEConfiguration>>();
|
||||
_configMonitorMock.Setup(m => m.CurrentValue).Returns(PoEConfiguration.Enabled);
|
||||
_configMonitorMock.Setup(m => m.CurrentValue).Returns(PoEConfiguration.EnabledDefault);
|
||||
|
||||
_executor = new PoEGenerationStageExecutor(
|
||||
_orchestrator,
|
||||
@@ -118,15 +117,15 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
.ReturnsAsync(new Dictionary<string, PoESubgraph?> { ["CVE-2021-44228"] = subgraph });
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<Subgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<PoESubgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(poeBytes);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.ComputePoEHash(poeBytes))
|
||||
.Setup(x => x.ComputePoEHash(It.IsAny<byte[]>()))
|
||||
.Returns(poeHash);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.SignPoEAsync(poeBytes, It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.SignPoEAsync(It.IsAny<byte[]>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(dsseBytes);
|
||||
|
||||
// Act
|
||||
@@ -136,7 +135,7 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
Assert.True(context.Analysis.TryGet<IReadOnlyList<PoEResult>>(ScanAnalysisKeys.PoEResults, out var results));
|
||||
Assert.Single(results!);
|
||||
Assert.Equal("CVE-2021-44228", results[0].VulnId);
|
||||
Assert.Equal(poeHash, results[0].PoeHash);
|
||||
Assert.Equal(poeHash, results[0].PoEHash);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
@@ -172,15 +171,15 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
.ReturnsAsync(new Dictionary<string, PoESubgraph?> { ["CVE-2021-44228"] = subgraph });
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<Subgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<PoESubgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(poeBytes);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.ComputePoEHash(poeBytes))
|
||||
.Setup(x => x.ComputePoEHash(It.IsAny<byte[]>()))
|
||||
.Returns(poeHash);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.SignPoEAsync(poeBytes, It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.SignPoEAsync(It.IsAny<byte[]>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(dsseBytes);
|
||||
|
||||
// Act
|
||||
@@ -226,7 +225,7 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
});
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<Subgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<PoESubgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(poeBytes);
|
||||
|
||||
_emitterMock
|
||||
@@ -273,15 +272,15 @@ public class PoEGenerationStageExecutorTests : IDisposable
|
||||
.ReturnsAsync(new Dictionary<string, PoESubgraph?> { ["CVE-2021-44228"] = subgraph });
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<Subgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<PoESubgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(poeBytes);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.ComputePoEHash(poeBytes))
|
||||
.Setup(x => x.ComputePoEHash(It.IsAny<byte[]>()))
|
||||
.Returns(poeHash);
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.SignPoEAsync(poeBytes, It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.Setup(x => x.SignPoEAsync(It.IsAny<byte[]>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(dsseBytes);
|
||||
|
||||
// Act
|
||||
|
||||
@@ -0,0 +1,175 @@
|
||||
// Copyright (c) StellaOps. Licensed under AGPL-3.0-or-later.
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.Extensions.Logging.Abstractions;
|
||||
using Moq;
|
||||
using StellaOps.Attestor;
|
||||
using StellaOps.Scanner.Core.Configuration;
|
||||
using StellaOps.Scanner.Reachability;
|
||||
using StellaOps.Scanner.Worker.Orchestration;
|
||||
using StellaOps.Signals.Storage;
|
||||
using Xunit;
|
||||
using Xunit.Abstractions;
|
||||
|
||||
namespace StellaOps.Scanner.Worker.Tests.PoE;
|
||||
|
||||
/// <summary>
|
||||
/// Direct tests for PoEOrchestrator to debug mock setup issues.
|
||||
/// </summary>
|
||||
public class PoEOrchestratorDirectTests : IDisposable
|
||||
{
|
||||
private readonly ITestOutputHelper _output;
|
||||
private readonly string _tempCasRoot;
|
||||
private readonly Mock<IReachabilityResolver> _resolverMock;
|
||||
private readonly Mock<IProofEmitter> _emitterMock;
|
||||
private readonly PoECasStore _casStore;
|
||||
private readonly PoEOrchestrator _orchestrator;
|
||||
|
||||
public PoEOrchestratorDirectTests(ITestOutputHelper output)
|
||||
{
|
||||
_output = output;
|
||||
_tempCasRoot = Path.Combine(Path.GetTempPath(), $"poe-direct-test-{Guid.NewGuid()}");
|
||||
Directory.CreateDirectory(_tempCasRoot);
|
||||
|
||||
_resolverMock = new Mock<IReachabilityResolver>();
|
||||
_emitterMock = new Mock<IProofEmitter>();
|
||||
_casStore = new PoECasStore(_tempCasRoot, NullLogger<PoECasStore>.Instance);
|
||||
|
||||
var logger = new XunitLogger<PoEOrchestrator>(_output);
|
||||
_orchestrator = new PoEOrchestrator(
|
||||
_resolverMock.Object,
|
||||
_emitterMock.Object,
|
||||
_casStore,
|
||||
logger
|
||||
);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task DirectTest_ShouldGeneratePoE()
|
||||
{
|
||||
// Arrange
|
||||
var vulnerabilities = new List<VulnerabilityMatch>
|
||||
{
|
||||
new VulnerabilityMatch(
|
||||
VulnId: "CVE-2021-44228",
|
||||
ComponentRef: "pkg:maven/log4j@2.14.1",
|
||||
IsReachable: true,
|
||||
Severity: "Critical")
|
||||
};
|
||||
|
||||
var subgraph = new PoESubgraph(
|
||||
BuildId: "gnu-build-id:test",
|
||||
ComponentRef: "pkg:maven/log4j@2.14.1",
|
||||
VulnId: "CVE-2021-44228",
|
||||
Nodes: new List<FunctionId>
|
||||
{
|
||||
new FunctionId("sha256:mod1", "main", "0x401000", null, null),
|
||||
new FunctionId("sha256:mod2", "vulnerable", "0x402000", null, null)
|
||||
},
|
||||
Edges: new List<Edge>
|
||||
{
|
||||
new Edge("main", "vulnerable", Array.Empty<string>(), 0.95)
|
||||
},
|
||||
EntryRefs: new[] { "main" },
|
||||
SinkRefs: new[] { "vulnerable" },
|
||||
PolicyDigest: "sha256:policy123",
|
||||
ToolchainDigest: "sha256:tool123"
|
||||
);
|
||||
|
||||
var poeBytes = System.Text.Encoding.UTF8.GetBytes("{\"test\":\"poe\"}");
|
||||
var dsseBytes = System.Text.Encoding.UTF8.GetBytes("{\"test\":\"dsse\"}");
|
||||
var poeHash = "blake3:abc123";
|
||||
|
||||
_output.WriteLine("Setting up resolver mock...");
|
||||
_resolverMock
|
||||
.Setup(x => x.ResolveBatchAsync(It.IsAny<IReadOnlyList<ReachabilityResolutionRequest>>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(new Dictionary<string, PoESubgraph?> { ["CVE-2021-44228"] = subgraph })
|
||||
.Verifiable();
|
||||
|
||||
_output.WriteLine("Setting up emitter mocks...");
|
||||
_emitterMock
|
||||
.Setup(x => x.EmitPoEAsync(It.IsAny<PoESubgraph>(), It.IsAny<ProofMetadata>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(poeBytes)
|
||||
.Verifiable();
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.ComputePoEHash(It.IsAny<byte[]>()))
|
||||
.Returns(poeHash)
|
||||
.Verifiable();
|
||||
|
||||
_emitterMock
|
||||
.Setup(x => x.SignPoEAsync(It.IsAny<byte[]>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
|
||||
.ReturnsAsync(dsseBytes)
|
||||
.Verifiable();
|
||||
|
||||
var context = new PoEScanContext(
|
||||
ScanId: "scan-test-123",
|
||||
GraphHash: "blake3:graphhash",
|
||||
BuildId: "gnu-build-id:test",
|
||||
ImageDigest: "sha256:imagehash",
|
||||
PolicyId: "default-policy",
|
||||
PolicyDigest: "sha256:policyhash",
|
||||
ScannerVersion: "1.0.0",
|
||||
ConfigPath: "etc/scanner.yaml"
|
||||
);
|
||||
|
||||
var configuration = PoEConfiguration.EnabledDefault;
|
||||
|
||||
// Act
|
||||
_output.WriteLine("Calling GeneratePoEArtifactsAsync...");
|
||||
var results = await _orchestrator.GeneratePoEArtifactsAsync(
|
||||
context,
|
||||
vulnerabilities,
|
||||
configuration,
|
||||
CancellationToken.None);
|
||||
|
||||
// Assert
|
||||
_output.WriteLine($"Results count: {results.Count}");
|
||||
Assert.NotEmpty(results);
|
||||
Assert.Single(results);
|
||||
Assert.Equal("CVE-2021-44228", results[0].VulnId);
|
||||
Assert.Equal(poeHash, results[0].PoEHash);
|
||||
|
||||
// Verify mocks were called
|
||||
_resolverMock.Verify();
|
||||
_emitterMock.Verify();
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
if (Directory.Exists(_tempCasRoot))
|
||||
{
|
||||
Directory.Delete(_tempCasRoot, recursive: true);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// XUnit logger adapter for testing.
|
||||
/// </summary>
|
||||
public class XunitLogger<T> : ILogger<T>
|
||||
{
|
||||
private readonly ITestOutputHelper _output;
|
||||
|
||||
public XunitLogger(ITestOutputHelper output)
|
||||
{
|
||||
_output = output;
|
||||
}
|
||||
|
||||
public IDisposable BeginScope<TState>(TState state) => null!;
|
||||
|
||||
public bool IsEnabled(LogLevel logLevel) => true;
|
||||
|
||||
public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Exception? exception, Func<TState, Exception?, string> formatter)
|
||||
{
|
||||
_output.WriteLine($"[{logLevel}] {formatter(state, exception)}");
|
||||
if (exception != null)
|
||||
{
|
||||
_output.WriteLine($"Exception: {exception}");
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user