up

src/Concelier/__Tests/StellaOps.Concelier.Connector.Vndr.Oracle.Tests/Oracle/OracleConnectorTests.cs

@@ -1,27 +1,27 @@
-using System;
-using System.Collections.Generic;
-using System.IO;
-using System.Linq;
-using System.Net;
-using System.Net.Http;
-using System.Net.Http.Headers;
-using System.Text;
-using System.Threading;
-using System.Threading.Tasks;
-using Microsoft.Extensions.DependencyInjection;
-using Microsoft.Extensions.Http;
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Net;
+using System.Net.Http;
+using System.Net.Http.Headers;
+using System.Text;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Http;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Logging.Abstractions;
 using Microsoft.Extensions.Options;
 using Microsoft.Extensions.Time.Testing;
-using StellaOps.Concelier.Bson;
+using StellaOps.Concelier.Documents;
 using StellaOps.Concelier.Models;
 using StellaOps.Concelier.Connector.Common;
 using StellaOps.Concelier.Connector.Common.Http;
 using StellaOps.Concelier.Connector.Common.Testing;
-using StellaOps.Concelier.Connector.Vndr.Oracle;
-using StellaOps.Concelier.Connector.Vndr.Oracle.Configuration;
-using StellaOps.Concelier.Connector.Vndr.Oracle.Internal;
+using StellaOps.Concelier.Connector.Vndr.Oracle;
+using StellaOps.Concelier.Connector.Vndr.Oracle.Configuration;
+using StellaOps.Concelier.Connector.Vndr.Oracle.Internal;
 using StellaOps.Concelier.Storage;
 using StellaOps.Concelier.Storage.Advisories;
 using StellaOps.Concelier.Storage;
@@ -30,82 +30,82 @@ using StellaOps.Concelier.Storage.Postgres;
 using StellaOps.Concelier.Storage.PsirtFlags;
 using StellaOps.Concelier.Testing;
 using Xunit.Abstractions;
-
-namespace StellaOps.Concelier.Connector.Vndr.Oracle.Tests;
-
+
+namespace StellaOps.Concelier.Connector.Vndr.Oracle.Tests;
+
 [Collection(ConcelierFixtureCollection.Name)]
-public sealed class OracleConnectorTests : IAsyncLifetime
-{
-    private readonly ConcelierPostgresFixture _fixture;
-    private readonly FakeTimeProvider _timeProvider;
-    private readonly CannedHttpMessageHandler _handler;
-    private readonly ITestOutputHelper _output;
-
-    private static readonly Uri AdvisoryOne = new("https://www.oracle.com/security-alerts/cpuapr2024-01.html");
-    private static readonly Uri AdvisoryTwo = new("https://www.oracle.com/security-alerts/cpuapr2024-02.html");
-    private static readonly Uri CalendarUri = new("https://www.oracle.com/security-alerts/cpuapr2024.html");
-
-    public OracleConnectorTests(ConcelierPostgresFixture fixture, ITestOutputHelper output)
-    {
-        _fixture = fixture;
-        _timeProvider = new FakeTimeProvider(new DateTimeOffset(2024, 4, 18, 0, 0, 0, TimeSpan.Zero));
-        _handler = new CannedHttpMessageHandler();
-        _output = output;
-    }
-
-    [Fact]
-    public async Task FetchParseMap_EmitsOraclePsirtSnapshot()
-    {
-        await using var provider = await BuildServiceProviderAsync();
-        SeedDetails();
-
-        var calendarFetcher = provider.GetRequiredService<OracleCalendarFetcher>();
-        var discovered = await calendarFetcher.GetAdvisoryUrisAsync(CancellationToken.None);
-        _output.WriteLine("Calendar URIs: " + string.Join(", ", discovered.Select(static uri => uri.AbsoluteUri)));
-        Assert.Equal(2, discovered.Count);
-
-        // Re-seed fixtures because calendar fetch consumes canned responses.
-        SeedDetails();
-
-        var connector = provider.GetRequiredService<OracleConnector>();
-        await connector.FetchAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.ParseAsync(provider, CancellationToken.None);
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        var advisoryStore = provider.GetRequiredService<IAdvisoryStore>();
-        var advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
-        _output.WriteLine("Advisories fetched: " + string.Join(", ", advisories.Select(static a => a.AdvisoryKey)));
-        _output.WriteLine($"Advisory count: {advisories.Count}");
-        Assert.Equal(2, advisories.Count);
-
-        var first = advisories.Single(advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-01-html");
-        var second = advisories.Single(advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-02-html");
-        Assert.Equal(new DateTimeOffset(2024, 4, 18, 12, 30, 0, TimeSpan.Zero), first.Published);
-        Assert.Equal(new DateTimeOffset(2024, 4, 19, 8, 15, 0, TimeSpan.Zero), second.Published);
-        Assert.All(advisories, advisory =>
-        {
-            Assert.True(advisory.Aliases.Any(alias => alias.StartsWith("CVE-", StringComparison.Ordinal)), $"Expected CVE alias for {advisory.AdvisoryKey}");
-            Assert.NotEmpty(advisory.AffectedPackages);
-        });
-
-        var snapshot = SnapshotSerializer.ToSnapshot(advisories.OrderBy(static a => a.AdvisoryKey, StringComparer.Ordinal).ToArray());
-        var expected = ReadFixture("oracle-advisories.snapshot.json");
-        var normalizedSnapshot = Normalize(snapshot);
-        var normalizedExpected = Normalize(expected);
-        if (!string.Equals(normalizedExpected, normalizedSnapshot, StringComparison.Ordinal))
-        {
-            var actualPath = Path.Combine(AppContext.BaseDirectory, "Source", "Vndr", "Oracle", "Fixtures", "oracle-advisories.actual.json");
-            var actualDirectory = Path.GetDirectoryName(actualPath);
-            if (!string.IsNullOrEmpty(actualDirectory))
-            {
-                Directory.CreateDirectory(actualDirectory);
-            }
-            File.WriteAllText(actualPath, snapshot);
-        }
-
-        Assert.Equal(normalizedExpected, normalizedSnapshot);
-
+public sealed class OracleConnectorTests : IAsyncLifetime
+{
+    private readonly ConcelierPostgresFixture _fixture;
+    private readonly FakeTimeProvider _timeProvider;
+    private readonly CannedHttpMessageHandler _handler;
+    private readonly ITestOutputHelper _output;
+
+    private static readonly Uri AdvisoryOne = new("https://www.oracle.com/security-alerts/cpuapr2024-01.html");
+    private static readonly Uri AdvisoryTwo = new("https://www.oracle.com/security-alerts/cpuapr2024-02.html");
+    private static readonly Uri CalendarUri = new("https://www.oracle.com/security-alerts/cpuapr2024.html");
+
+    public OracleConnectorTests(ConcelierPostgresFixture fixture, ITestOutputHelper output)
+    {
+        _fixture = fixture;
+        _timeProvider = new FakeTimeProvider(new DateTimeOffset(2024, 4, 18, 0, 0, 0, TimeSpan.Zero));
+        _handler = new CannedHttpMessageHandler();
+        _output = output;
+    }
+
+    [Fact]
+    public async Task FetchParseMap_EmitsOraclePsirtSnapshot()
+    {
+        await using var provider = await BuildServiceProviderAsync();
+        SeedDetails();
+
+        var calendarFetcher = provider.GetRequiredService<OracleCalendarFetcher>();
+        var discovered = await calendarFetcher.GetAdvisoryUrisAsync(CancellationToken.None);
+        _output.WriteLine("Calendar URIs: " + string.Join(", ", discovered.Select(static uri => uri.AbsoluteUri)));
+        Assert.Equal(2, discovered.Count);
+
+        // Re-seed fixtures because calendar fetch consumes canned responses.
+        SeedDetails();
+
+        var connector = provider.GetRequiredService<OracleConnector>();
+        await connector.FetchAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.ParseAsync(provider, CancellationToken.None);
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        var advisoryStore = provider.GetRequiredService<IAdvisoryStore>();
+        var advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
+        _output.WriteLine("Advisories fetched: " + string.Join(", ", advisories.Select(static a => a.AdvisoryKey)));
+        _output.WriteLine($"Advisory count: {advisories.Count}");
+        Assert.Equal(2, advisories.Count);
+
+        var first = advisories.Single(advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-01-html");
+        var second = advisories.Single(advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-02-html");
+        Assert.Equal(new DateTimeOffset(2024, 4, 18, 12, 30, 0, TimeSpan.Zero), first.Published);
+        Assert.Equal(new DateTimeOffset(2024, 4, 19, 8, 15, 0, TimeSpan.Zero), second.Published);
+        Assert.All(advisories, advisory =>
+        {
+            Assert.True(advisory.Aliases.Any(alias => alias.StartsWith("CVE-", StringComparison.Ordinal)), $"Expected CVE alias for {advisory.AdvisoryKey}");
+            Assert.NotEmpty(advisory.AffectedPackages);
+        });
+
+        var snapshot = SnapshotSerializer.ToSnapshot(advisories.OrderBy(static a => a.AdvisoryKey, StringComparer.Ordinal).ToArray());
+        var expected = ReadFixture("oracle-advisories.snapshot.json");
+        var normalizedSnapshot = Normalize(snapshot);
+        var normalizedExpected = Normalize(expected);
+        if (!string.Equals(normalizedExpected, normalizedSnapshot, StringComparison.Ordinal))
+        {
+            var actualPath = Path.Combine(AppContext.BaseDirectory, "Source", "Vndr", "Oracle", "Fixtures", "oracle-advisories.actual.json");
+            var actualDirectory = Path.GetDirectoryName(actualPath);
+            if (!string.IsNullOrEmpty(actualDirectory))
+            {
+                Directory.CreateDirectory(actualDirectory);
+            }
+            File.WriteAllText(actualPath, snapshot);
+        }
+
+        Assert.Equal(normalizedExpected, normalizedSnapshot);
+
         var psirtStore = provider.GetRequiredService<IPsirtFlagStore>();
         var flags = new List<PsirtFlagRecord>();
         foreach (var advisory in advisories)
@@ -119,124 +119,124 @@ public sealed class OracleConnectorTests : IAsyncLifetime
 
         Assert.Equal(2, flags.Count);
         Assert.All(flags, flag => Assert.Equal("Oracle", flag.Vendor));
-    }
-
-    [Fact]
-    public async Task FetchAsync_IdempotentForUnchangedAdvisories()
-    {
-        await using var provider = await BuildServiceProviderAsync();
-        SeedDetails();
-
-        var connector = provider.GetRequiredService<OracleConnector>();
-        await connector.FetchAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.ParseAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        // Second run with unchanged documents should rely on fetch cache.
-        SeedDetails();
-        await connector.FetchAsync(provider, CancellationToken.None);
-        await connector.ParseAsync(provider, CancellationToken.None);
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        var stateRepository = provider.GetRequiredService<ISourceStateRepository>();
-        var state = await stateRepository.TryGetAsync(VndrOracleConnectorPlugin.SourceName, CancellationToken.None);
-        Assert.NotNull(state);
-        var cursor = OracleCursor.FromBson(state!.Cursor);
-        Assert.Empty(cursor.PendingDocuments);
-        Assert.Empty(cursor.PendingMappings);
-        Assert.Equal(2, cursor.FetchCache.Count);
-        Assert.All(cursor.FetchCache.Values, entry => Assert.False(string.IsNullOrWhiteSpace(entry.Sha256)));
-
-        var documentStore = provider.GetRequiredService<IDocumentStore>();
-        var first = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryOne.ToString(), CancellationToken.None);
-        Assert.NotNull(first);
-        Assert.Equal(DocumentStatuses.Mapped, first!.Status);
-
-        var second = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryTwo.ToString(), CancellationToken.None);
-        Assert.NotNull(second);
-        Assert.Equal(DocumentStatuses.Mapped, second!.Status);
-
+    }
+
+    [Fact]
+    public async Task FetchAsync_IdempotentForUnchangedAdvisories()
+    {
+        await using var provider = await BuildServiceProviderAsync();
+        SeedDetails();
+
+        var connector = provider.GetRequiredService<OracleConnector>();
+        await connector.FetchAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.ParseAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        // Second run with unchanged documents should rely on fetch cache.
+        SeedDetails();
+        await connector.FetchAsync(provider, CancellationToken.None);
+        await connector.ParseAsync(provider, CancellationToken.None);
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        var stateRepository = provider.GetRequiredService<ISourceStateRepository>();
+        var state = await stateRepository.TryGetAsync(VndrOracleConnectorPlugin.SourceName, CancellationToken.None);
+        Assert.NotNull(state);
+        var cursor = OracleCursor.FromBson(state!.Cursor);
+        Assert.Empty(cursor.PendingDocuments);
+        Assert.Empty(cursor.PendingMappings);
+        Assert.Equal(2, cursor.FetchCache.Count);
+        Assert.All(cursor.FetchCache.Values, entry => Assert.False(string.IsNullOrWhiteSpace(entry.Sha256)));
+
+        var documentStore = provider.GetRequiredService<IDocumentStore>();
+        var first = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryOne.ToString(), CancellationToken.None);
+        Assert.NotNull(first);
+        Assert.Equal(DocumentStatuses.Mapped, first!.Status);
+
+        var second = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryTwo.ToString(), CancellationToken.None);
+        Assert.NotNull(second);
+        Assert.Equal(DocumentStatuses.Mapped, second!.Status);
+
         var dtoStore = provider.GetRequiredService<IDtoStore>();
         var dto1 = await dtoStore.FindByDocumentIdAsync(first!.Id, CancellationToken.None);
         Assert.NotNull(dto1);
         var dto2 = await dtoStore.FindByDocumentIdAsync(second!.Id, CancellationToken.None);
         Assert.NotNull(dto2);
-    }
-
-    [Fact]
-    public async Task FetchAsync_ResumeProcessesNewCalendarEntries()
-    {
-        await using var provider = await BuildServiceProviderAsync();
-
-        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024-single.html");
-        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
-
-        var connector = provider.GetRequiredService<OracleConnector>();
-        await connector.FetchAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.ParseAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        var advisoryStore = provider.GetRequiredService<IAdvisoryStore>();
-        var advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
-        Assert.Single(advisories);
-        Assert.Equal("oracle/cpuapr2024-01-html", advisories[0].AdvisoryKey);
-
-        _handler.Clear();
-        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
-        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
-        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
-
-        await connector.FetchAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.ParseAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
-        Assert.Equal(2, advisories.Count);
-        Assert.Contains(advisories, advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-02-html");
-    }
-
-    [Fact]
-    public async Task ParseAsync_InvalidDocumentIsQuarantined()
-    {
-        await using var provider = await BuildServiceProviderAsync();
-
-        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
-        AddDetailResponse(AdvisoryOne, "oracle-detail-invalid.html", "\"oracle-001\"");
-        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
-
-        var connector = provider.GetRequiredService<OracleConnector>();
-        await connector.FetchAsync(provider, CancellationToken.None);
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.ParseAsync(provider, CancellationToken.None);
-
-        var documentStore = provider.GetRequiredService<IDocumentStore>();
-        var invalidDocument = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryOne.ToString(), CancellationToken.None);
-        Assert.NotNull(invalidDocument);
-        _output.WriteLine($"Invalid document status: {invalidDocument!.Status}");
-
+    }
+
+    [Fact]
+    public async Task FetchAsync_ResumeProcessesNewCalendarEntries()
+    {
+        await using var provider = await BuildServiceProviderAsync();
+
+        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024-single.html");
+        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
+
+        var connector = provider.GetRequiredService<OracleConnector>();
+        await connector.FetchAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.ParseAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        var advisoryStore = provider.GetRequiredService<IAdvisoryStore>();
+        var advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
+        Assert.Single(advisories);
+        Assert.Equal("oracle/cpuapr2024-01-html", advisories[0].AdvisoryKey);
+
+        _handler.Clear();
+        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
+        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
+        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
+
+        await connector.FetchAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.ParseAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        advisories = await advisoryStore.GetRecentAsync(10, CancellationToken.None);
+        Assert.Equal(2, advisories.Count);
+        Assert.Contains(advisories, advisory => advisory.AdvisoryKey == "oracle/cpuapr2024-02-html");
+    }
+
+    [Fact]
+    public async Task ParseAsync_InvalidDocumentIsQuarantined()
+    {
+        await using var provider = await BuildServiceProviderAsync();
+
+        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
+        AddDetailResponse(AdvisoryOne, "oracle-detail-invalid.html", "\"oracle-001\"");
+        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
+
+        var connector = provider.GetRequiredService<OracleConnector>();
+        await connector.FetchAsync(provider, CancellationToken.None);
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.ParseAsync(provider, CancellationToken.None);
+
+        var documentStore = provider.GetRequiredService<IDocumentStore>();
+        var invalidDocument = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryOne.ToString(), CancellationToken.None);
+        Assert.NotNull(invalidDocument);
+        _output.WriteLine($"Invalid document status: {invalidDocument!.Status}");
+
         var dtoStore = provider.GetRequiredService<IDtoStore>();
         var invalidDto = await dtoStore.FindByDocumentIdAsync(invalidDocument.Id, CancellationToken.None);
         if (invalidDto is not null)
         {
-            _output.WriteLine("Validation unexpectedly succeeded. DTO: " + invalidDto.Payload.ToJson());
-        }
-        Assert.Equal(DocumentStatuses.Failed, invalidDocument.Status);
-        Assert.Null(invalidDto);
-
-        var validDocument = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryTwo.ToString(), CancellationToken.None);
-        Assert.NotNull(validDocument);
-        Assert.Equal(DocumentStatuses.PendingMap, validDocument!.Status);
-
-        _timeProvider.Advance(TimeSpan.FromMinutes(1));
-        await connector.MapAsync(provider, CancellationToken.None);
-
-        var advisories = await provider.GetRequiredService<IAdvisoryStore>().GetRecentAsync(10, CancellationToken.None);
+            _output.WriteLine("Validation unexpectedly succeeded. DTO: " + invalidDto.Payload.ToJson());
+        }
+        Assert.Equal(DocumentStatuses.Failed, invalidDocument.Status);
+        Assert.Null(invalidDto);
+
+        var validDocument = await documentStore.FindBySourceAndUriAsync(VndrOracleConnectorPlugin.SourceName, AdvisoryTwo.ToString(), CancellationToken.None);
+        Assert.NotNull(validDocument);
+        Assert.Equal(DocumentStatuses.PendingMap, validDocument!.Status);
+
+        _timeProvider.Advance(TimeSpan.FromMinutes(1));
+        await connector.MapAsync(provider, CancellationToken.None);
+
+        var advisories = await provider.GetRequiredService<IAdvisoryStore>().GetRecentAsync(10, CancellationToken.None);
         Assert.Single(advisories);
         Assert.Equal("oracle/cpuapr2024-02-html", advisories[0].AdvisoryKey);
 
@@ -246,109 +246,109 @@ public sealed class OracleConnectorTests : IAsyncLifetime
 
         var missingFlag = await psirtStore.FindAsync("oracle/cpuapr2024-01-html", CancellationToken.None);
         Assert.Null(missingFlag);
-
-        var stateRepository = provider.GetRequiredService<ISourceStateRepository>();
-        var state = await stateRepository.TryGetAsync(VndrOracleConnectorPlugin.SourceName, CancellationToken.None);
-        Assert.NotNull(state);
-        var cursor = OracleCursor.FromBson(state!.Cursor);
-        Assert.Empty(cursor.PendingDocuments);
-        Assert.Empty(cursor.PendingMappings);
-    }
-
+
+        var stateRepository = provider.GetRequiredService<ISourceStateRepository>();
+        var state = await stateRepository.TryGetAsync(VndrOracleConnectorPlugin.SourceName, CancellationToken.None);
+        Assert.NotNull(state);
+        var cursor = OracleCursor.FromBson(state!.Cursor);
+        Assert.Empty(cursor.PendingDocuments);
+        Assert.Empty(cursor.PendingMappings);
+    }
+
     private async Task<ServiceProvider> BuildServiceProviderAsync()
     {
         await _fixture.TruncateAllTablesAsync();
         _handler.Clear();
-
-        var services = new ServiceCollection();
-        services.AddLogging(builder => builder.AddProvider(NullLoggerProvider.Instance));
-        services.AddSingleton<TimeProvider>(_timeProvider);
-        services.AddSingleton(_handler);
-
+
+        var services = new ServiceCollection();
+        services.AddLogging(builder => builder.AddProvider(NullLoggerProvider.Instance));
+        services.AddSingleton<TimeProvider>(_timeProvider);
+        services.AddSingleton(_handler);
+
         services.AddConcelierPostgresStorage(options =>
         {
             options.ConnectionString = _fixture.ConnectionString;
             options.SchemaName = _fixture.SchemaName;
             options.CommandTimeoutSeconds = 5;
         });
-
-        services.AddSourceCommon();
-        services.AddOracleConnector(opts =>
-        {
-            opts.CalendarUris = new List<Uri> { CalendarUri };
-            opts.RequestDelay = TimeSpan.Zero;
-        });
-
-        services.Configure<HttpClientFactoryOptions>(OracleOptions.HttpClientName, builderOptions =>
-        {
-            builderOptions.HttpMessageHandlerBuilderActions.Add(builder =>
-            {
-                builder.PrimaryHandler = _handler;
-            });
-        });
-
+
+        services.AddSourceCommon();
+        services.AddOracleConnector(opts =>
+        {
+            opts.CalendarUris = new List<Uri> { CalendarUri };
+            opts.RequestDelay = TimeSpan.Zero;
+        });
+
+        services.Configure<HttpClientFactoryOptions>(OracleOptions.HttpClientName, builderOptions =>
+        {
+            builderOptions.HttpMessageHandlerBuilderActions.Add(builder =>
+            {
+                builder.PrimaryHandler = _handler;
+            });
+        });
+
         return services.BuildServiceProvider();
     }
-
-    private void SeedDetails()
-    {
-        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
-        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
-        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
-    }
-
-    private void AddCalendarResponse(Uri uri, string fixture)
-    {
-        _handler.AddResponse(uri, () =>
-        {
-            var response = new HttpResponseMessage(HttpStatusCode.OK)
-            {
-                Content = new StringContent(ReadFixture(fixture), Encoding.UTF8, "text/html"),
-            };
-
-            return response;
-        });
-    }
-
-    private void AddDetailResponse(Uri uri, string fixture, string? etag)
-    {
-        _handler.AddResponse(uri, () =>
-        {
-            var response = new HttpResponseMessage(HttpStatusCode.OK)
-            {
-                Content = new StringContent(ReadFixture(fixture), Encoding.UTF8, "text/html"),
-            };
-
-            if (!string.IsNullOrEmpty(etag))
-            {
-                response.Headers.ETag = new EntityTagHeaderValue(etag);
-            }
-
-            return response;
-        });
-    }
-
-    private static string ReadFixture(string filename)
-    {
-        var primary = Path.Combine(AppContext.BaseDirectory, "Source", "Vndr", "Oracle", "Fixtures", filename);
-        if (File.Exists(primary))
-        {
-            return File.ReadAllText(primary);
-        }
-
-        var fallback = Path.Combine(AppContext.BaseDirectory, "Oracle", "Fixtures", filename);
-        if (File.Exists(fallback))
-        {
-            return File.ReadAllText(fallback);
-        }
-
-        throw new FileNotFoundException($"Fixture '{filename}' not found in test output.", filename);
-    }
-
-    private static string Normalize(string value)
-        => value.Replace("\r\n", "\n", StringComparison.Ordinal);
-
-    public Task InitializeAsync() => Task.CompletedTask;
-
-    public Task DisposeAsync() => Task.CompletedTask;
-}
+
+    private void SeedDetails()
+    {
+        AddCalendarResponse(CalendarUri, "oracle-calendar-cpuapr2024.html");
+        AddDetailResponse(AdvisoryOne, "oracle-detail-cpuapr2024-01.html", "\"oracle-001\"");
+        AddDetailResponse(AdvisoryTwo, "oracle-detail-cpuapr2024-02.html", "\"oracle-002\"");
+    }
+
+    private void AddCalendarResponse(Uri uri, string fixture)
+    {
+        _handler.AddResponse(uri, () =>
+        {
+            var response = new HttpResponseMessage(HttpStatusCode.OK)
+            {
+                Content = new StringContent(ReadFixture(fixture), Encoding.UTF8, "text/html"),
+            };
+
+            return response;
+        });
+    }
+
+    private void AddDetailResponse(Uri uri, string fixture, string? etag)
+    {
+        _handler.AddResponse(uri, () =>
+        {
+            var response = new HttpResponseMessage(HttpStatusCode.OK)
+            {
+                Content = new StringContent(ReadFixture(fixture), Encoding.UTF8, "text/html"),
+            };
+
+            if (!string.IsNullOrEmpty(etag))
+            {
+                response.Headers.ETag = new EntityTagHeaderValue(etag);
+            }
+
+            return response;
+        });
+    }
+
+    private static string ReadFixture(string filename)
+    {
+        var primary = Path.Combine(AppContext.BaseDirectory, "Source", "Vndr", "Oracle", "Fixtures", filename);
+        if (File.Exists(primary))
+        {
+            return File.ReadAllText(primary);
+        }
+
+        var fallback = Path.Combine(AppContext.BaseDirectory, "Oracle", "Fixtures", filename);
+        if (File.Exists(fallback))
+        {
+            return File.ReadAllText(fallback);
+        }
+
+        throw new FileNotFoundException($"Fixture '{filename}' not found in test output.", filename);
+    }
+
+    private static string Normalize(string value)
+        => value.Replace("\r\n", "\n", StringComparison.Ordinal);
+
+    public Task InitializeAsync() => Task.CompletedTask;
+
+    public Task DisposeAsync() => Task.CompletedTask;
+}