up

src/Scanner/__Tests/StellaOps.Scanner.Reachability.Tests/FakeFileContentAddressableStore.cs

@@ -0,0 +1,39 @@
+using System;
+using System.Collections.Concurrent;
+using System.IO;
+using System.Threading;
+using System.Threading.Tasks;
+using StellaOps.Scanner.Cache.Abstractions;
+
+namespace StellaOps.Scanner.Reachability.Tests;
+
+internal sealed class FakeFileContentAddressableStore : IFileContentAddressableStore
+{
+    private readonly ConcurrentDictionary<string, byte[]> store = new();
+
+    public ValueTask<FileCasEntry?> TryGetAsync(string sha256, CancellationToken cancellationToken = default)
+    {
+        if (store.TryGetValue(sha256, out var bytes))
+        {
+            return ValueTask.FromResult<FileCasEntry?>(new FileCasEntry(sha256, bytes.LongLength, DateTimeOffset.UtcNow, DateTimeOffset.UtcNow, sha256 + ".zip"));
+        }
+
+        return ValueTask.FromResult<FileCasEntry?>(null);
+    }
+
+    public Task<FileCasEntry> PutAsync(FileCasPutRequest request, CancellationToken cancellationToken = default)
+    {
+        using var ms = new MemoryStream();
+        request.Content.CopyTo(ms);
+        store[request.Sha256] = ms.ToArray();
+        return Task.FromResult(new FileCasEntry(request.Sha256, ms.Length, DateTimeOffset.UtcNow, DateTimeOffset.UtcNow, request.Sha256 + ".zip"));
+    }
+
+    public Task<bool> RemoveAsync(string sha256, CancellationToken cancellationToken = default)
+        => Task.FromResult(store.TryRemove(sha256, out _));
+
+    public Task<int> EvictExpiredAsync(CancellationToken cancellationToken = default) => Task.FromResult(0);
+    public Task<int> ExportAsync(string destinationDirectory, CancellationToken cancellationToken = default) => Task.FromResult(0);
+    public Task<int> ImportAsync(string sourceDirectory, CancellationToken cancellationToken = default) => Task.FromResult(0);
+    public Task<int> CompactAsync(CancellationToken cancellationToken = default) => Task.FromResult(0);
+}