Refactor code structure and optimize performance across multiple modules

src/AirGap/__Tests/StellaOps.AirGap.Importer.Tests/AirGapControllerContractTests.cs

@@ -26,7 +26,8 @@ public sealed class AirGapControllerContractTests
 {
     #region AIRGAP-5100-010: Contract Tests
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_ExportEndpoint_ExpectedRequestStructure()
     {
         // Arrange - Define expected request structure
@@ -56,7 +57,8 @@ public sealed class AirGapControllerContractTests
         feeds.GetArrayLength().Should().BeGreaterThan(0);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_ExportEndpoint_ExpectedResponseStructure()
     {
         // Arrange - Define expected response structure
@@ -87,7 +89,8 @@ public sealed class AirGapControllerContractTests
         parsed.RootElement.TryGetProperty("manifest", out _).Should().BeTrue();
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_ImportEndpoint_ExpectedRequestStructure()
     {
         // Arrange - Import request (typically multipart form or bundle URL)
@@ -107,7 +110,8 @@ public sealed class AirGapControllerContractTests
         parsed.RootElement.TryGetProperty("bundleDigest", out _).Should().BeTrue();
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_ImportEndpoint_ExpectedResponseStructure()
     {
         // Arrange
@@ -131,7 +135,8 @@ public sealed class AirGapControllerContractTests
         parsed.RootElement.TryGetProperty("feedsImported", out _).Should().BeTrue();
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_ListBundlesEndpoint_ExpectedResponseStructure()
     {
         // Arrange
@@ -164,7 +169,8 @@ public sealed class AirGapControllerContractTests
         parsed.RootElement.TryGetProperty("total", out _).Should().BeTrue();
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Contract_StateEndpoint_ExpectedResponseStructure()
     {
         // Arrange - AirGap state response
@@ -197,7 +203,8 @@ public sealed class AirGapControllerContractTests
 
     #region AIRGAP-5100-011: Auth Tests
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_RequiredScopes_ForExport()
     {
         // Arrange - Expected scopes for export operation
@@ -207,7 +214,8 @@ public sealed class AirGapControllerContractTests
         requiredScopes.Should().Contain("airgap:export");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_RequiredScopes_ForImport()
     {
         // Arrange - Expected scopes for import operation
@@ -217,7 +225,8 @@ public sealed class AirGapControllerContractTests
         requiredScopes.Should().Contain("airgap:import");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_RequiredScopes_ForList()
     {
         // Arrange - Expected scopes for list operation
@@ -227,7 +236,8 @@ public sealed class AirGapControllerContractTests
         requiredScopes.Should().Contain("airgap:read");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_DenyByDefault_NoTokenReturnsUnauthorized()
     {
         // Arrange - Request without token
@@ -237,7 +247,8 @@ public sealed class AirGapControllerContractTests
         expectedStatusCode.Should().Be(HttpStatusCode.Unauthorized);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_TenantIsolation_CannotAccessOtherTenantBundles()
     {
         // Arrange - Claims for tenant A
@@ -256,7 +267,8 @@ public sealed class AirGapControllerContractTests
         // Requests for tenant-B bundles should be rejected
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void Auth_TokenExpiry_ExpiredTokenReturnsForbidden()
     {
         // Arrange - Expired token scenario
@@ -272,7 +284,8 @@ public sealed class AirGapControllerContractTests
 
     #region AIRGAP-5100-012: OTel Trace Assertions
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_ExportOperation_IncludesBundleIdTag()
     {
         // Arrange
@@ -289,7 +302,8 @@ public sealed class AirGapControllerContractTests
         expectedTags.Should().Contain("operation");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_ImportOperation_IncludesOperationTag()
     {
         // Arrange
@@ -305,7 +319,8 @@ public sealed class AirGapControllerContractTests
         expectedTags["operation"].Should().Be("airgap.import");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_Metrics_TracksExportCount()
     {
         // Arrange
@@ -317,7 +332,8 @@ public sealed class AirGapControllerContractTests
         metricName.Should().NotBeNullOrEmpty();
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_Metrics_TracksImportCount()
     {
         // Arrange
@@ -329,7 +345,8 @@ public sealed class AirGapControllerContractTests
         expectedDimensions.Should().Contain("status");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_ActivitySource_HasCorrectName()
     {
         // Arrange
@@ -339,11 +356,13 @@ public sealed class AirGapControllerContractTests
         expectedSourceName.Should().StartWith("StellaOps.");
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void OTel_Spans_PropagateTraceContext()
     {
         // Arrange - Create a trace context
         using var activity = new Activity("test-airgap-operation");
+using StellaOps.TestKit;
         activity.Start();
 
         // Act