up

src/Concelier/__Tests/StellaOps.Concelier.Connector.CertBund.Tests/CertBundConnectorTests.cs

@@ -1,12 +1,12 @@
-using System;
-using System.Net.Http;
-using System.Net.Http.Headers;
+using System;
+using System.Net.Http;
+using System.Net.Http.Headers;
 using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
 using FluentAssertions;
 using Microsoft.Extensions.DependencyInjection;
-using StellaOps.Concelier.Bson;
+using StellaOps.Concelier.Documents;
 using StellaOps.Concelier.Connector.CertBund.Configuration;
 using StellaOps.Concelier.Connector.Common;
 using StellaOps.Concelier.Connector.Common.Testing;
@@ -15,14 +15,14 @@ using StellaOps.Concelier.Storage;
 using StellaOps.Concelier.Storage.Advisories;
 using StellaOps.Concelier.Testing;
 using Xunit;
-
-namespace StellaOps.Concelier.Connector.CertBund.Tests;
-
+
+namespace StellaOps.Concelier.Connector.CertBund.Tests;
+
 [Collection(ConcelierFixtureCollection.Name)]
 public sealed class CertBundConnectorTests
 {
-    private static readonly Uri FeedUri = new("https://test.local/content/public/securityAdvisory/rss");
-    private static readonly Uri PortalUri = new("https://test.local/portal/");
+    private static readonly Uri FeedUri = new("https://test.local/content/public/securityAdvisory/rss");
+    private static readonly Uri PortalUri = new("https://test.local/portal/");
     private static readonly Uri DetailUri = new("https://test.local/portal/api/securityadvisory?name=WID-SEC-2025-2264");
 
     private readonly ConcelierPostgresFixture _fixture;
@@ -31,7 +31,7 @@ public sealed class CertBundConnectorTests
     {
         _fixture = fixture;
     }
-
+
     [Fact]
     public async Task FetchParseMap_ProducesCanonicalAdvisory()
     {
@@ -46,35 +46,35 @@ public sealed class CertBundConnectorTests
         var advisoryStore = harness.ServiceProvider.GetRequiredService<IAdvisoryStore>();
         var advisories = await advisoryStore.GetRecentAsync(5, CancellationToken.None);
         advisories.Should().HaveCount(1);
-
-        var advisory = advisories[0];
-        advisory.AdvisoryKey.Should().Be("WID-SEC-2025-2264");
-        advisory.Aliases.Should().Contain("CVE-2025-1234");
-        advisory.AffectedPackages.Should().Contain(package => package.Identifier.Contains("Ivanti"));
-        advisory.References.Should().Contain(reference => reference.Url == DetailUri.ToString());
-        advisory.Language.Should().Be("de");
-
-        var endpoint = advisory.AffectedPackages.Should().ContainSingle(p => p.Identifier.Contains("Endpoint Manager") && !p.Identifier.Contains("Cloud"))
-            .Subject;
-        endpoint.VersionRanges.Should().ContainSingle(range =>
-            range.RangeKind == NormalizedVersionSchemes.SemVer &&
-            range.IntroducedVersion == "2023.1" &&
-            range.FixedVersion == "2024.2");
-        endpoint.NormalizedVersions.Should().ContainSingle(rule =>
-            rule.Min == "2023.1" &&
-            rule.Max == "2024.2" &&
-            rule.Notes == "certbund:WID-SEC-2025-2264:ivanti");
-
+
+        var advisory = advisories[0];
+        advisory.AdvisoryKey.Should().Be("WID-SEC-2025-2264");
+        advisory.Aliases.Should().Contain("CVE-2025-1234");
+        advisory.AffectedPackages.Should().Contain(package => package.Identifier.Contains("Ivanti"));
+        advisory.References.Should().Contain(reference => reference.Url == DetailUri.ToString());
+        advisory.Language.Should().Be("de");
+
+        var endpoint = advisory.AffectedPackages.Should().ContainSingle(p => p.Identifier.Contains("Endpoint Manager") && !p.Identifier.Contains("Cloud"))
+            .Subject;
+        endpoint.VersionRanges.Should().ContainSingle(range =>
+            range.RangeKind == NormalizedVersionSchemes.SemVer &&
+            range.IntroducedVersion == "2023.1" &&
+            range.FixedVersion == "2024.2");
+        endpoint.NormalizedVersions.Should().ContainSingle(rule =>
+            rule.Min == "2023.1" &&
+            rule.Max == "2024.2" &&
+            rule.Notes == "certbund:WID-SEC-2025-2264:ivanti");
+
         var stateRepository = harness.ServiceProvider.GetRequiredService<ISourceStateRepository>();
         var state = await stateRepository.TryGetAsync(CertBundConnectorPlugin.SourceName, CancellationToken.None);
         state.Should().NotBeNull();
-        state!.Cursor.Should().NotBeNull();
-        state.Cursor.TryGetValue("pendingDocuments", out var pendingDocs).Should().BeTrue();
-        pendingDocs!.AsBsonArray.Should().BeEmpty();
-        state.Cursor.TryGetValue("pendingMappings", out var pendingMappings).Should().BeTrue();
-        pendingMappings!.AsBsonArray.Should().BeEmpty();
-    }
-
+        state!.Cursor.Should().NotBeNull();
+        state.Cursor.TryGetValue("pendingDocuments", out var pendingDocs).Should().BeTrue();
+        pendingDocs!.AsDocumentArray.Should().BeEmpty();
+        state.Cursor.TryGetValue("pendingMappings", out var pendingMappings).Should().BeTrue();
+        pendingMappings!.AsDocumentArray.Should().BeEmpty();
+    }
+
     [Fact]
     public async Task Fetch_PersistsDocumentWithMetadata()
     {
@@ -87,17 +87,17 @@ public sealed class CertBundConnectorTests
         var documentStore = harness.ServiceProvider.GetRequiredService<IDocumentStore>();
         var document = await documentStore.FindBySourceAndUriAsync(CertBundConnectorPlugin.SourceName, DetailUri.ToString(), CancellationToken.None);
         document.Should().NotBeNull();
-        document!.Metadata.Should().ContainKey("certbund.advisoryId").WhoseValue.Should().Be("WID-SEC-2025-2264");
-        document.Metadata.Should().ContainKey("certbund.category");
+        document!.Metadata.Should().ContainKey("certbund.advisoryId").WhoseValue.Should().Be("WID-SEC-2025-2264");
+        document.Metadata.Should().ContainKey("certbund.category");
         document.Metadata.Should().ContainKey("certbund.published");
         document.Status.Should().Be(DocumentStatuses.PendingParse);
 
         var stateRepository = harness.ServiceProvider.GetRequiredService<ISourceStateRepository>();
         var state = await stateRepository.TryGetAsync(CertBundConnectorPlugin.SourceName, CancellationToken.None);
         state.Should().NotBeNull();
-        state!.Cursor.Should().NotBeNull();
+        state!.Cursor.Should().NotBeNull();
         state.Cursor.TryGetValue("pendingDocuments", out var pendingDocs).Should().BeTrue();
-        pendingDocs!.AsBsonArray.Should().HaveCount(1);
+        pendingDocs!.AsDocumentArray.Should().HaveCount(1);
     }
 
     private async Task<ConnectorTestHarness> BuildHarnessAsync()
@@ -133,13 +133,13 @@ public sealed class CertBundConnectorTests
             var response = new HttpResponseMessage(System.Net.HttpStatusCode.OK)
             {
                 Content = new StringContent(json, Encoding.UTF8, "application/json"),
-            };
-            if (!string.IsNullOrWhiteSpace(etag))
-            {
-                response.Headers.ETag = new EntityTagHeaderValue(etag);
-            }
-
-            return response;
+            };
+            if (!string.IsNullOrWhiteSpace(etag))
+            {
+                response.Headers.ETag = new EntityTagHeaderValue(etag);
+            }
+
+            return response;
         });
     }
 
@@ -158,7 +158,7 @@ public sealed class CertBundConnectorTests
             Content = new StringContent(html, Encoding.UTF8, "text/html"),
         });
     }
-
+
     private static string ReadFixture(string fileName)
         => System.IO.File.ReadAllText(System.IO.Path.Combine(AppContext.BaseDirectory, "Fixtures", fileName));
 }