up

tests/reachability/StellaOps.Signals.Reachability.Tests/ReachabilityScoringTests.cs

@@ -7,8 +7,10 @@ using System.Threading;
 using System.Threading.Tasks;
 using FluentAssertions;
 using Microsoft.Extensions.Logging.Abstractions;
+using Microsoft.Extensions.Options;
 using MongoDB.Bson;
 using StellaOps.Signals.Models;
+using StellaOps.Signals.Options;
 using StellaOps.Signals.Parsing;
 using StellaOps.Signals.Persistence;
 using StellaOps.Signals.Services;
@@ -56,7 +58,19 @@ public sealed class ReachabilityScoringTests
         var callgraph = await LoadCallgraphAsync(caseId, variant, variantPath);
         var callgraphRepo = new InMemoryCallgraphRepository(callgraph);
         var factRepo = new InMemoryReachabilityFactRepository();
-        var scoringService = new ReachabilityScoringService(callgraphRepo, factRepo, TimeProvider.System, NullLogger<ReachabilityScoringService>.Instance);
+        var options = new SignalsOptions();
+        var cache = new InMemoryReachabilityCache();
+        var eventsPublisher = new NullEventsPublisher();
+        var unknowns = new InMemoryUnknownsRepository();
+        var scoringService = new ReachabilityScoringService(
+            callgraphRepo,
+            factRepo,
+            TimeProvider.System,
+            Options.Create(options),
+            cache,
+            unknowns,
+            eventsPublisher,
+            NullLogger<ReachabilityScoringService>.Instance);
 
         var request = BuildRequest(casePath, variant, sinks, entryPoints);
         request.CallgraphId = callgraph.Id;
@@ -218,6 +232,47 @@ public sealed class ReachabilityScoringTests
             return Task.FromResult(document);
         }
     }
+
+    private sealed class InMemoryReachabilityCache : IReachabilityCache
+    {
+        private readonly Dictionary<string, ReachabilityFactDocument> storage = new(StringComparer.Ordinal);
+
+        public Task<ReachabilityFactDocument?> GetAsync(string subjectKey, CancellationToken cancellationToken)
+        {
+            storage.TryGetValue(subjectKey, out var doc);
+            return Task.FromResult(doc);
+        }
+
+        public Task SetAsync(ReachabilityFactDocument document, CancellationToken cancellationToken)
+        {
+            storage[document.SubjectKey] = document;
+            return Task.CompletedTask;
+        }
+
+        public Task InvalidateAsync(string subjectKey, CancellationToken cancellationToken)
+        {
+            storage.Remove(subjectKey);
+            return Task.CompletedTask;
+        }
+    }
+
+    private sealed class InMemoryUnknownsRepository : IUnknownsRepository
+    {
+        public Task UpsertAsync(string subjectKey, IEnumerable<UnknownSymbolDocument> items, CancellationToken cancellationToken) =>
+            Task.CompletedTask;
+
+        public Task<IReadOnlyList<UnknownSymbolDocument>> GetBySubjectAsync(string subjectKey, CancellationToken cancellationToken) =>
+            Task.FromResult((IReadOnlyList<UnknownSymbolDocument>)Array.Empty<UnknownSymbolDocument>());
+
+        public Task<int> CountBySubjectAsync(string subjectKey, CancellationToken cancellationToken) =>
+            Task.FromResult(0);
+    }
+
+    private sealed class NullEventsPublisher : IEventsPublisher
+    {
+        public Task PublishFactUpdatedAsync(ReachabilityFactDocument fact, CancellationToken cancellationToken) => Task.CompletedTask;
+    }
+
     private static string LocateRepoRoot()
     {
         var current = new DirectoryInfo(AppContext.BaseDirectory);