Refactor code structure and optimize performance across multiple modules

2025-12-26 20:03:22 +02:00
parent c786faae84
commit b4fc66feb6
3353 changed files with 88254 additions and 1590657 deletions
--- a/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/CrossModuleEvidenceLinkingTests.cs
+++ b/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/CrossModuleEvidenceLinkingTests.cs
@@ -16,6 +16,7 @@ using StellaOps.Evidence.Storage.Postgres.Tests.Fixtures;
 using Xunit;
 using Xunit.Abstractions;

+using StellaOps.TestKit;
 namespace StellaOps.Evidence.Storage.Postgres.Tests;

 /// <summary>
@@ -52,7 +53,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Multi-Module Evidence for Same Subject

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task SameSubject_MultipleEvidenceTypes_AllLinked()
    {
        // Arrange - A container image subject with evidence from multiple modules
@@ -87,7 +89,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime
        _output.WriteLine($"Subject {subjectNodeId} has {allEvidence.Count} evidence records from different modules");
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task SameSubject_FilterByType_ReturnsCorrectEvidence()
    {
        // Arrange
@@ -111,7 +114,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Evidence Chain Scenarios

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task EvidenceChain_ScanToVexToPolicy_LinkedCorrectly()
    {
        // Scenario: Vulnerability scan → VEX assessment → Policy decision
@@ -147,7 +151,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime
        _output.WriteLine($"Evidence chain: Scan({scan.EvidenceId}) → VEX({vex.EvidenceId}) → Policy({policy.EvidenceId})");
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task EvidenceChain_ReachabilityToEpssToPolicy_LinkedCorrectly()
    {
        // Scenario: Reachability analysis + EPSS score → Policy decision
@@ -178,7 +183,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Multi-Tenant Evidence Isolation

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task MultiTenant_SameSubject_IsolatedByTenant()
    {
        // Arrange - Two tenants with evidence for the same subject
@@ -214,7 +220,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Evidence Graph Queries

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task EvidenceGraph_AllTypesForArtifact_ReturnsComplete()
    {
        // Arrange - Simulate a complete evidence graph for a container artifact
@@ -251,7 +258,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime
        }
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task EvidenceGraph_ExistsCheck_ForAllTypes()
    {
        // Arrange
@@ -272,7 +280,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Cross-Module Evidence Correlation

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task Correlation_SameCorrelationId_FindsRelatedEvidence()
    {
        // Arrange - Evidence from different modules with same correlation ID
@@ -295,7 +304,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime
        allEvidence.Should().OnlyContain(e => e.Provenance.CorrelationId == correlationId);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task Generators_MultiplePerSubject_AllPreserved()
    {
        // Arrange - Evidence from different generators
@@ -322,7 +332,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime

    #region Evidence Count and Statistics

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task CountBySubject_AfterMultiModuleInserts_ReturnsCorrectCount()
    {
        // Arrange
@@ -339,7 +350,8 @@ public sealed class CrossModuleEvidenceLinkingTests : IAsyncLifetime
        count.Should().Be(3);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByType_AcrossSubjects_ReturnsAll()
    {
        // Arrange - Multiple subjects with same evidence type
--- a/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/PostgresEvidenceStoreIntegrationTests.cs
+++ b/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/PostgresEvidenceStoreIntegrationTests.cs
@@ -12,6 +12,7 @@ using StellaOps.Evidence.Storage.Postgres.Tests.Fixtures;
 using Xunit;
 using Xunit.Abstractions;

+using StellaOps.TestKit;
 namespace StellaOps.Evidence.Storage.Postgres.Tests;

 /// <summary>
@@ -47,7 +48,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Store Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task StoreAsync_NewEvidence_ReturnsEvidenceId()
    {
        // Arrange
@@ -61,7 +63,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        _output.WriteLine($"Stored evidence: {storedId}");
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task StoreAsync_DuplicateEvidence_IsIdempotent()
    {
        // Arrange
@@ -80,7 +83,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        count.Should().Be(1);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task StoreBatchAsync_MultipleRecords_StoresAllSuccessfully()
    {
        // Arrange
@@ -98,7 +102,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        count.Should().Be(5);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task StoreBatchAsync_WithDuplicates_StoresOnlyUnique()
    {
        // Arrange
@@ -116,7 +121,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region GetById Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByIdAsync_ExistingEvidence_ReturnsEvidence()
    {
        // Arrange
@@ -136,7 +142,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved.Provenance.GeneratorId.Should().Be(evidence.Provenance.GeneratorId);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByIdAsync_NonExistingEvidence_ReturnsNull()
    {
        // Arrange
@@ -149,7 +156,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved.Should().BeNull();
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByIdAsync_WithSignatures_PreservesSignatures()
    {
        // Arrange
@@ -170,7 +178,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region GetBySubject Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetBySubjectAsync_MultipleEvidence_ReturnsAll()
    {
        // Arrange
@@ -196,7 +205,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
            .Contain(new[] { EvidenceType.Scan, EvidenceType.Reachability, EvidenceType.Policy });
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetBySubjectAsync_WithTypeFilter_ReturnsFiltered()
    {
        // Arrange
@@ -213,7 +223,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved[0].EvidenceType.Should().Be(EvidenceType.Scan);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetBySubjectAsync_NoEvidence_ReturnsEmptyList()
    {
        // Arrange
@@ -230,7 +241,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region GetByType Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByTypeAsync_MultipleEvidence_ReturnsMatchingType()
    {
        // Arrange
@@ -246,7 +258,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved.Should().OnlyContain(e => e.EvidenceType == EvidenceType.Scan);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task GetByTypeAsync_WithLimit_RespectsLimit()
    {
        // Arrange
@@ -266,7 +279,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Exists Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task ExistsAsync_ExistingEvidence_ReturnsTrue()
    {
        // Arrange
@@ -280,7 +294,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        exists.Should().BeTrue();
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task ExistsAsync_NonExistingEvidence_ReturnsFalse()
    {
        // Arrange
@@ -294,7 +309,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        exists.Should().BeFalse();
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task ExistsAsync_NonExistingSubject_ReturnsFalse()
    {
        // Arrange
@@ -311,7 +327,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Delete Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task DeleteAsync_ExistingEvidence_ReturnsTrue()
    {
        // Arrange
@@ -329,7 +346,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved.Should().BeNull();
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task DeleteAsync_NonExistingEvidence_ReturnsFalse()
    {
        // Arrange
@@ -346,7 +364,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Count Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task CountBySubjectAsync_MultipleEvidence_ReturnsCorrectCount()
    {
        // Arrange
@@ -362,7 +381,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        count.Should().Be(3);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task CountBySubjectAsync_NoEvidence_ReturnsZero()
    {
        // Arrange
@@ -379,7 +399,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Integrity Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task RoundTrip_EvidenceRecord_PreservesIntegrity()
    {
        // Arrange
@@ -394,7 +415,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved!.VerifyIntegrity().Should().BeTrue("evidence ID should match computed hash");
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task RoundTrip_BinaryPayload_PreservesData()
    {
        // Arrange
@@ -417,7 +439,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime
        retrieved!.Payload.ToArray().Should().BeEquivalentTo(binaryPayload);
    }

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public async Task RoundTrip_UnicodePayload_PreservesData()
    {
        // Arrange
@@ -446,7 +469,8 @@ public sealed class PostgresEvidenceStoreIntegrationTests : IAsyncLifetime

    #region Factory Tests

-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
    public void Factory_CreateStore_ReturnsTenantScopedStore()
    {
        // Arrange
--- a/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/StellaOps.Evidence.Storage.Postgres.Tests.csproj
+++ b/src/Libraries/Tests/StellaOps.Evidence.Storage.Postgres.Tests/StellaOps.Evidence.Storage.Postgres.Tests.csproj
@@ -17,5 +17,6 @@
  <ItemGroup>
    <ProjectReference Include="..\..\StellaOps.Evidence.Storage.Postgres\StellaOps.Evidence.Storage.Postgres.csproj" />
    <ProjectReference Include="..\..\StellaOps.Infrastructure.Postgres\StellaOps.Infrastructure.Postgres.csproj" />
+    <ProjectReference Include="../../StellaOps.TestKit/StellaOps.TestKit.csproj" />
  </ItemGroup>
 </Project>