up

src/Concelier/__Tests/StellaOps.Concelier.Exporter.Json.Tests/JsonExporterDependencyInjectionRoutineTests.cs

@@ -1,12 +1,11 @@
-using System.Collections.Generic;
-using System.Runtime.CompilerServices;
-using System.Threading.Tasks;
+using System.Collections.Generic;
+using System.Runtime.CompilerServices;
+using System.Threading.Tasks;
 using System.Collections.Immutable;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
-using MongoDB.Driver;
 using StellaOps.Concelier.Core.Jobs;
 using StellaOps.Concelier.Core.Events;
 using StellaOps.Concelier.Exporter.Json;
@@ -15,15 +14,16 @@ using StellaOps.Concelier.Storage.Mongo.Exporting;
 using StellaOps.Concelier.Models;
 using StellaOps.Cryptography;
 using StellaOps.Cryptography.DependencyInjection;
-
-namespace StellaOps.Concelier.Exporter.Json.Tests;
-
-public sealed class JsonExporterDependencyInjectionRoutineTests
-{
-    [Fact]
-    public void Register_AddsJobDefinitionAndServices()
-    {
-        var services = new ServiceCollection();
+using StellaOps.Provenance.Mongo;
+
+namespace StellaOps.Concelier.Exporter.Json.Tests;
+
+public sealed class JsonExporterDependencyInjectionRoutineTests
+{
+    [Fact]
+    public void Register_AddsJobDefinitionAndServices()
+    {
+        var services = new ServiceCollection();
         services.AddLogging();
         services.AddSingleton<IAdvisoryStore, StubAdvisoryStore>();
         services.AddSingleton<IExportStateStore, StubExportStateStore>();
@@ -32,64 +32,60 @@ public sealed class JsonExporterDependencyInjectionRoutineTests
         services.AddOptions<JobSchedulerOptions>();
         services.Configure<CryptoHashOptions>(_ => { });
         services.AddStellaOpsCrypto();
-
-        var configuration = new ConfigurationBuilder()
-            .AddInMemoryCollection(new Dictionary<string, string?>())
-            .Build();
-
-        var routine = new JsonExporterDependencyInjectionRoutine();
-        routine.Register(services, configuration);
-
-        using var provider = services.BuildServiceProvider();
-        var optionsAccessor = provider.GetRequiredService<IOptions<JobSchedulerOptions>>();
-        var options = optionsAccessor.Value;
-
-        Assert.True(options.Definitions.TryGetValue(JsonExportJob.JobKind, out var definition));
-        Assert.Equal(typeof(JsonExportJob), definition.JobType);
-        Assert.True(definition.Enabled);
-
-        var exporter = provider.GetRequiredService<JsonFeedExporter>();
-        Assert.NotNull(exporter);
-    }
-
-    private sealed class StubAdvisoryStore : IAdvisoryStore
-    {
-        public Task<IReadOnlyList<Advisory>> GetRecentAsync(int limit, CancellationToken cancellationToken, IClientSessionHandle? session = null)
+
+        var configuration = new ConfigurationBuilder()
+            .AddInMemoryCollection(new Dictionary<string, string?>())
+            .Build();
+
+        var routine = new JsonExporterDependencyInjectionRoutine();
+        routine.Register(services, configuration);
+
+        using var provider = services.BuildServiceProvider();
+        var optionsAccessor = provider.GetRequiredService<IOptions<JobSchedulerOptions>>();
+        var options = optionsAccessor.Value;
+
+        Assert.True(options.Definitions.TryGetValue(JsonExportJob.JobKind, out var definition));
+        Assert.Equal(typeof(JsonExportJob), definition.JobType);
+        Assert.True(definition.Enabled);
+
+        var exporter = provider.GetRequiredService<JsonFeedExporter>();
+        Assert.NotNull(exporter);
+    }
+
+    private sealed class StubAdvisoryStore : IAdvisoryStore
+    {
+        public Task<IReadOnlyList<Advisory>> GetRecentAsync(int limit, CancellationToken cancellationToken)
         {
-            _ = session;
             return Task.FromResult<IReadOnlyList<Advisory>>(Array.Empty<Advisory>());
         }
 
-        public Task<Advisory?> FindAsync(string advisoryKey, CancellationToken cancellationToken, IClientSessionHandle? session = null)
+        public Task<Advisory?> FindAsync(string advisoryKey, CancellationToken cancellationToken)
         {
-            _ = session;
             return Task.FromResult<Advisory?>(null);
         }
 
-        public Task UpsertAsync(Advisory advisory, CancellationToken cancellationToken, IClientSessionHandle? session = null)
+        public Task UpsertAsync(Advisory advisory, CancellationToken cancellationToken)
         {
-            _ = session;
             return Task.CompletedTask;
         }
 
-        public IAsyncEnumerable<Advisory> StreamAsync(CancellationToken cancellationToken, IClientSessionHandle? session = null)
+        public IAsyncEnumerable<Advisory> StreamAsync(CancellationToken cancellationToken)
         {
-            _ = session;
             return Enumerate(cancellationToken);
 
             static async IAsyncEnumerable<Advisory> Enumerate([EnumeratorCancellation] CancellationToken ct)
             {
-                ct.ThrowIfCancellationRequested();
-                await Task.Yield();
-                yield break;
-            }
-        }
-    }
-
+                ct.ThrowIfCancellationRequested();
+                await Task.Yield();
+                yield break;
+            }
+        }
+    }
+
     private sealed class StubExportStateStore : IExportStateStore
     {
         private ExportStateRecord? _record;
-
+
         public Task<ExportStateRecord?> FindAsync(string id, CancellationToken cancellationToken)
         {
             return Task.FromResult(_record);
@@ -107,6 +103,9 @@ public sealed class JsonExporterDependencyInjectionRoutineTests
         public ValueTask AppendAsync(AdvisoryEventAppendRequest request, CancellationToken cancellationToken)
             => throw new NotSupportedException();
 
+        public ValueTask AttachStatementProvenanceAsync(Guid statementId, DsseProvenance provenance, TrustInfo trust, CancellationToken cancellationToken)
+            => ValueTask.CompletedTask;
+
         public ValueTask<AdvisoryReplay> ReplayAsync(string vulnerabilityKey, DateTimeOffset? asOf, CancellationToken cancellationToken)
         {
             return ValueTask.FromResult(new AdvisoryReplay(