feat: implement IAsyncDiscoveryInitializer and related classes for im…

…proved test discovery handling

TUnit.TestProject/Bugs/3992/BugRecreation.cs → TUnit.TestProject/Bugs/3992/RuntimeInitializeTests.cs

@@ -1,20 +1,18 @@
-using TUnit.TestProject.Attributes;
+using TUnit.Core.Interfaces;
+using TUnit.TestProject.Attributes;
 
 namespace TUnit.TestProject.Bugs._3992;
 
 /// <summary>
 /// Once this is discovered during test discovery, containers spin up
 /// </summary>
 [EngineTest(ExpectedResult.Pass)]
-public sealed class BugRecreation
+public sealed class RuntimeInitializeTests
 {
     //Docker container
     [ClassDataSource<DummyContainer>(Shared = SharedType.PerClass)]
     public required DummyContainer Container { get; init; }
 
-    public IEnumerable<Func<int>> Executions
-        => Container.Ints.Select(e => new Func<int>(() => e));
-
     [Before(Class)]
     public static Task BeforeClass(ClassHookContext context) => NotInitialised(context.Tests);
 
@@ -23,7 +21,7 @@ public IEnumerable<Func<int>> Executions
 
     public static async Task NotInitialised(IEnumerable<TestContext> tests)
     {
-        var bugRecreations = tests.Select(x => x.Metadata.TestDetails.ClassInstance).OfType<BugRecreation>();
+        var bugRecreations = tests.Select(x => x.Metadata.TestDetails.ClassInstance).OfType<RuntimeInitializeTests>();
 
         foreach (var bugRecreation in bugRecreations)
         {
@@ -38,4 +36,21 @@ public async Task Test(int value, CancellationToken token)
         await Assert.That(value).IsNotDefault();
         await Assert.That(DummyContainer.NumberOfInits).IsEqualTo(1);
     }
+
+    public class DummyContainer : IAsyncInitializer, IAsyncDisposable
+    {
+        public Task InitializeAsync()
+        {
+            NumberOfInits++;
+            return Task.CompletedTask;
+        }
+
+        public static int NumberOfInits { get; private set; }
+
+        public ValueTask DisposeAsync()
+        {
+            return default;
+        }
+    }
+
 }

TUnit.TestProject/Bugs/3992/RuntimeInitializeTests2.cs

@@ -0,0 +1,56 @@
+using TUnit.Core.Interfaces;
+using TUnit.TestProject.Attributes;
+
+namespace TUnit.TestProject.Bugs._3992;
+
+/// <summary>
+/// Once this is discovered during test discovery, containers spin up
+/// </summary>
+[EngineTest(ExpectedResult.Pass)]
+public sealed class DiscoveryInitializeTests
+{
+    //Docker container
+    [ClassDataSource<DummyContainer>(Shared = SharedType.PerClass)]
+    public required DummyContainer Container { get; init; }
+
+    [Before(Class)]
+    public static Task BeforeClass(ClassHookContext context) => Initialised(context.Tests);
+
+    [After(TestDiscovery)]
+    public static Task AfterDiscovery(TestDiscoveryContext context) => Initialised(context.AllTests);
+
+    public static async Task Initialised(IEnumerable<TestContext> tests)
+    {
+        var bugRecreations = tests.Select(x => x.Metadata.TestDetails.ClassInstance).OfType<RuntimeInitializeTests>();
+
+        foreach (var bugRecreation in bugRecreations)
+        {
+            await Assert.That(bugRecreation.Container).IsNotNull();
+            await Assert.That(DummyContainer.NumberOfInits).IsEqualTo(1);
+        }
+    }
+
+    [Test, Arguments(1)]
+    public async Task Test(int value, CancellationToken token)
+    {
+        await Assert.That(value).IsNotDefault();
+        await Assert.That(DummyContainer.NumberOfInits).IsEqualTo(1);
+    }
+
+    public class DummyContainer : IAsyncDiscoveryInitializer, IAsyncDisposable
+    {
+        public Task InitializeAsync()
+        {
+            NumberOfInits++;
+            return Task.CompletedTask;
+        }
+
+        public static int NumberOfInits { get; private set; }
+
+        public ValueTask DisposeAsync()
+        {
+            return default;
+        }
+    }
+
+}