up

src/Scanner/__Tests/StellaOps.Scanner.Worker.Tests/EntropyStageExecutorTests.cs

@@ -0,0 +1,67 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging.Abstractions;
+using StellaOps.Scanner.Core.Contracts;
+using StellaOps.Scanner.Core.Entropy;
+using StellaOps.Scanner.Worker.Processing;
+using StellaOps.Scanner.Worker.Processing.Entropy;
+using Xunit;
+
+namespace StellaOps.Scanner.Worker.Tests;
+
+public class EntropyStageExecutorTests
+{
+    [Fact]
+    public async Task ExecuteAsync_WritesEntropyReportAndSummary()
+    {
+        // Arrange: create a temp file with random bytes to yield high entropy.
+        var tmp = Path.GetTempFileName();
+        var rng = new Random(1234);
+        var bytes = new byte[64 * 1024];
+        rng.NextBytes(bytes);
+        File.WriteAllBytes(tmp, bytes);
+
+        var fileEntries = new List<ScanFileEntry>
+        {
+            new ScanFileEntry(tmp, sizeBytes: bytes.LongLength, kind: "blob", metadata: new Dictionary<string, string>())
+        };
+
+        var lease = new StubLease("job-1", "scan-1", imageDigest: "sha256:test", layerDigest: "sha256:layer");
+        var context = new ScanJobContext(lease, TimeProvider.System, DateTimeOffset.UtcNow, CancellationToken.None);
+        context.Analysis.Set(ScanAnalysisKeys.FileEntries, (IReadOnlyList<ScanFileEntry>)fileEntries);
+
+        var executor = new EntropyStageExecutor(NullLogger<EntropyStageExecutor>.Instance);
+
+        // Act
+        await executor.ExecuteAsync(context, CancellationToken.None);
+
+        // Assert
+        Assert.True(context.Analysis.TryGet<EntropyReport>(ScanAnalysisKeys.EntropyReport, out var report));
+        Assert.NotNull(report);
+        Assert.Equal("sha256:layer", report!.LayerDigest);
+        Assert.NotEmpty(report.Files);
+
+        Assert.True(context.Analysis.TryGet<EntropyLayerSummary>(ScanAnalysisKeys.EntropyLayerSummary, out var summary));
+        Assert.NotNull(summary);
+        Assert.Equal("sha256:layer", summary!.LayerDigest);
+    }
+
+    private sealed class StubLease : IScanJobLease
+    {
+        public StubLease(string jobId, string scanId, string imageDigest, string layerDigest)
+        {
+            JobId = jobId;
+            ScanId = scanId;
+            ImageDigest = imageDigest;
+            LayerDigest = layerDigest;
+        }
+
+        public string JobId { get; }
+        public string ScanId { get; }
+        public string? ImageDigest { get; }
+        public string? LayerDigest { get; }
+    }
+}