Trying to fix the regression

Author: sakno

src/DotNext.Tests/Runtime/CompilerServices/SpawningAsyncTaskMethodBuilderTests.cs

@@ -6,13 +6,18 @@ namespace DotNext.Runtime.CompilerServices;
 public sealed class SpawningAsyncTaskMethodBuilderTests : Test
 {
     [Fact]
-    public static Task ForkAsyncMethodWithResult()
+    public static async Task ForkAsyncMethodWithResult()
     {
-        return Sum(40, 2, Thread.CurrentThread.ManagedThreadId);
+        using var resetEvent = new ManualResetEventSlim(initialState: false);
+        var task = Sum(40, 2, Thread.CurrentThread.ManagedThreadId);
+        True(resetEvent.Wait(DefaultTimeout));
+
+        Equal(42, await task);
 
         [AsyncMethodBuilder(typeof(SpawningAsyncTaskMethodBuilder<>))]
-        static async Task<int> Sum(int x, int y, int callerThreadId)
+        async Task<int> Sum(int x, int y, int callerThreadId)
         {
+            resetEvent.Set();
             NotEqual(callerThreadId, Thread.CurrentThread.ManagedThreadId);
 
             await Task.CompletedTask;
@@ -21,13 +26,18 @@ static async Task<int> Sum(int x, int y, int callerThreadId)
     }
 
     [Fact]
-    public static Task ForkAsyncMethodWithoutResult()
+    public static async Task ForkAsyncMethodWithoutResult()
     {
-        return CheckThreadId(Thread.CurrentThread.ManagedThreadId);
+        using var resetEvent = new ManualResetEventSlim(initialState: false);
+        var task = CheckThreadId(Thread.CurrentThread.ManagedThreadId);
+        True(resetEvent.Wait(DefaultTimeout));
+
+        await task;
 
         [AsyncMethodBuilder(typeof(SpawningAsyncTaskMethodBuilder))]
-        static async Task CheckThreadId(int callerThreadId)
+        async Task CheckThreadId(int callerThreadId)
         {
+            resetEvent.Set();
             NotEqual(callerThreadId, Thread.CurrentThread.ManagedThreadId);
 
             await Task.CompletedTask;
@@ -37,14 +47,17 @@ static async Task CheckThreadId(int callerThreadId)
     [Fact]
     public static async Task CancellationOfSpawnedMethod()
     {
+        using var resetEvent = new ManualResetEventSlim(initialState: false);
         var task = CheckThreadId(Thread.CurrentThread.ManagedThreadId, new(true));
+        True(resetEvent.Wait(DefaultTimeout));
 
         await task.ConfigureAwait(ConfigureAwaitOptions.ContinueOnCapturedContext | ConfigureAwaitOptions.SuppressThrowing);
         True(task.IsCanceled);
 
         [AsyncMethodBuilder(typeof(SpawningAsyncTaskMethodBuilder))]
-        static async Task CheckThreadId(int callerThreadId, CancellationToken token)
+        async Task CheckThreadId(int callerThreadId, CancellationToken token)
         {
+            resetEvent.Set();
             NotEqual(callerThreadId, Thread.CurrentThread.ManagedThreadId);
 
             await Task.Delay(DefaultTimeout, token);