Refactor code structure and optimize performance across multiple modules

src/Policy/__Tests/StellaOps.Policy.Storage.Postgres.Tests/RecheckEvidenceMigrationTests.cs

@@ -5,6 +5,8 @@ using Npgsql;
 using StellaOps.Policy.Storage.Postgres;
 using Xunit;
 
+
+using StellaOps.TestKit;
 namespace StellaOps.Policy.Storage.Postgres.Tests;
 
 [Collection(PolicyPostgresCollection.Name)]
@@ -26,7 +28,8 @@ public sealed class RecheckEvidenceMigrationTests : IAsyncLifetime
 
     public Task DisposeAsync() => Task.CompletedTask;
 
-    [Fact]
+    [Trait("Category", TestCategories.Unit)]
+        [Fact]
     public async Task Migration_CreatesRecheckAndEvidenceTables()
     {
         await using var connection = await _dataSource.OpenConnectionAsync("default", "reader", CancellationToken.None);