save progress

src/Scanner/__Tests/StellaOps.Scanner.CallGraph.Tests/ValkeyCallGraphCacheServiceTests.cs

@@ -1,36 +1,73 @@
 using Microsoft.Extensions.Logging.Abstractions;
 using Microsoft.Extensions.Options;
-using StellaOps.Messaging.Testing.Fixtures;
+using Moq;
+using StackExchange.Redis;
 using StellaOps.Scanner.CallGraph;
 using StellaOps.Scanner.CallGraph.Caching;
 using Xunit;
 
 namespace StellaOps.Scanner.CallGraph.Tests;
 
-[Collection(nameof(ValkeyFixtureCollection))]
 public class ValkeyCallGraphCacheServiceTests : IAsyncLifetime
 {
-    private readonly ValkeyFixture _fixture;
     private ValkeyCallGraphCacheService _cache = null!;
 
-    public ValkeyCallGraphCacheServiceTests(ValkeyFixture fixture)
-    {
-        _fixture = fixture;
-    }
-
     public Task InitializeAsync()
     {
+        var store = new Dictionary<string, RedisValue>(StringComparer.Ordinal);
+
+        var database = new Mock<IDatabase>(MockBehavior.Loose);
+        database
+            .Setup(db => db.StringGetAsync(It.IsAny<RedisKey>(), It.IsAny<CommandFlags>()))
+            .ReturnsAsync((RedisKey key, CommandFlags _) =>
+                store.TryGetValue(key.ToString(), out var value) ? value : RedisValue.Null);
+
+        database
+            .Setup(db => db.StringSetAsync(
+                It.IsAny<RedisKey>(),
+                It.IsAny<RedisValue>(),
+                It.IsAny<TimeSpan?>(),
+                It.IsAny<When>(),
+                It.IsAny<CommandFlags>()))
+            .ReturnsAsync((RedisKey key, RedisValue value, TimeSpan? _, When _, CommandFlags _) =>
+            {
+                store[key.ToString()] = value;
+                return true;
+            });
+
+        database
+            .Setup(db => db.StringSetAsync(
+                It.IsAny<RedisKey>(),
+                It.IsAny<RedisValue>(),
+                It.IsAny<TimeSpan?>(),
+                It.IsAny<bool>(),
+                It.IsAny<When>(),
+                It.IsAny<CommandFlags>()))
+            .ReturnsAsync((RedisKey key, RedisValue value, TimeSpan? _, bool _, When _, CommandFlags _) =>
+            {
+                store[key.ToString()] = value;
+                return true;
+            });
+
+        var connection = new Mock<IConnectionMultiplexer>(MockBehavior.Loose);
+        connection
+            .Setup(c => c.GetDatabase(It.IsAny<int>(), It.IsAny<object?>()))
+            .Returns(database.Object);
+
         var options = Options.Create(new CallGraphCacheConfig
         {
             Enabled = true,
-            ConnectionString = _fixture.ConnectionString,
+            ConnectionString = "localhost:6379",
             KeyPrefix = "test:callgraph:",
             TtlSeconds = 60,
             EnableGzip = true,
             CircuitBreaker = new CircuitBreakerConfig { FailureThreshold = 3, TimeoutSeconds = 30, HalfOpenTimeout = 10 }
         });
 
-        _cache = new ValkeyCallGraphCacheService(options, NullLogger<ValkeyCallGraphCacheService>.Instance);
+        _cache = new ValkeyCallGraphCacheService(
+            options,
+            NullLogger<ValkeyCallGraphCacheService>.Instance,
+            connectionFactory: _ => Task.FromResult(connection.Object));
         return Task.CompletedTask;
     }