Enable async for test wait signals (#111)

Make TaskCompletionSource used in wait signals run completions asynchronously to avoid potential deadlocks during test runs.
nats-io · Aug 11, 2023 · 31797cf · 31797cf
1 parent 638e2d3
commit 31797cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/NATS.Client.TestUtilities/WaitSignal.cs b/tests/NATS.Client.TestUtilities/WaitSignal.cs
@@ -56,7 +56,7 @@ public WaitSignal(TimeSpan timeout, int count = 1)
     {
         _timeout = timeout;
         _count = count;
-        _tcs = new TaskCompletionSource();
+        _tcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
     }
 
     public TimeSpan Timeout => _timeout;
@@ -106,7 +106,7 @@ public WaitSignal(TimeSpan timeout, int count = 1)
     {
         _timeout = timeout;
         _count = count;
-        _tcs = new TaskCompletionSource<T>();
+        _tcs = new TaskCompletionSource<T>(TaskCreationOptions.RunContinuationsAsynchronously);
     }
 
     public TimeSpan Timeout => _timeout;