Refactor code structure and optimize performance across multiple modules

src/TaskRunner/StellaOps.TaskRunner/StellaOps.TaskRunner.Tests/PackRunIncidentModeTests.cs

@@ -3,11 +3,13 @@ using Microsoft.Extensions.Time.Testing;
 using StellaOps.TaskRunner.Core.Events;
 using StellaOps.TaskRunner.Core.IncidentMode;
 
+using StellaOps.TestKit;
 namespace StellaOps.TaskRunner.Tests;
 
 public sealed class PackRunIncidentModeTests
 {
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task ActivateAsync_ActivatesIncidentModeSuccessfully()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -34,7 +36,8 @@ public sealed class PackRunIncidentModeTests
         Assert.NotNull(result.Status.ExpiresAt);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task ActivateAsync_WithoutDuration_CreatesIndefiniteIncidentMode()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -57,7 +60,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Null(result.Status.ExpiresAt);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task ActivateAsync_EmitsTimelineEvent()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -88,7 +92,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Equal(PackRunIncidentEventTypes.IncidentModeActivated, evt.EventType);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task DeactivateAsync_DeactivatesIncidentMode()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -118,7 +123,8 @@ public sealed class PackRunIncidentModeTests
         Assert.False(status.Active);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task GetStatusAsync_ReturnsInactiveForUnknownRun()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -132,7 +138,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Equal(IncidentEscalationLevel.None, status.Level);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task GetStatusAsync_AutoDeactivatesExpiredIncidentMode()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -161,7 +168,8 @@ public sealed class PackRunIncidentModeTests
         Assert.False(status.Active);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task HandleSloBreachAsync_ActivatesIncidentModeFromBreach()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -190,7 +198,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Contains("error_rate_5m", result.Status.ActivationReason!);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task HandleSloBreachAsync_MapsSeverityToLevel()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -228,7 +237,8 @@ public sealed class PackRunIncidentModeTests
         }
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task HandleSloBreachAsync_ReturnsErrorForMissingResourceId()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -254,7 +264,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Contains("No resource ID", result.Error);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task EscalateAsync_IncreasesEscalationLevel()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -286,7 +297,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Contains("Escalated", result.Status.ActivationReason);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task EscalateAsync_FailsWhenNotInIncidentMode()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -304,7 +316,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Contains("not active", result.Error);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task EscalateAsync_FailsWhenNewLevelIsLowerOrEqual()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -333,7 +346,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Contains("Cannot escalate", result.Error);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void GetSettingsForLevel_ReturnsCorrectSettings()
     {
         var store = new InMemoryPackRunIncidentModeStore();
@@ -356,7 +370,8 @@ public sealed class PackRunIncidentModeTests
         Assert.Equal(365, criticalSettings.RetentionPolicy.LogRetentionDays);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void PackRunIncidentModeStatus_Inactive_ReturnsDefaultValues()
     {
         var inactive = PackRunIncidentModeStatus.Inactive();
@@ -371,7 +386,8 @@ public sealed class PackRunIncidentModeTests
         Assert.False(inactive.DebugCaptureSettings.CaptureActive);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void IncidentRetentionPolicy_Extended_HasLongerRetention()
     {
         var defaultPolicy = IncidentRetentionPolicy.Default();
@@ -382,7 +398,8 @@ public sealed class PackRunIncidentModeTests
         Assert.True(extendedPolicy.ArtifactRetentionDays > defaultPolicy.ArtifactRetentionDays);
     }
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public void IncidentTelemetrySettings_Enhanced_HasHigherSampling()
     {
         var defaultSettings = IncidentTelemetrySettings.Default();