Correct various xUnit warnings relating to task cancellation in tests.

fgimian · Jan 14, 2025 · 23f5e77 · 23f5e77
1 parent 7a4e271
commit 23f5e77
Showing 1 changed file with 20 additions and 12 deletions.
diff --git a/src/TotalMixVC.Tests/TaskExtensionsTests.cs b/src/TotalMixVC.Tests/TaskExtensionsTests.cs
@@ -14,7 +14,7 @@ public async Task TimeoutAfter_Completes_DoesNotThrowException_Async()
             completed = true;
         };
 
-        await Task.Run(task).TimeoutAfter(1000);
+        await Task.Run(task, TestContext.Current.CancellationToken).TimeoutAfter(1000);
 
         Assert.True(completed);
     }
@@ -47,14 +47,19 @@ public async Task TimeoutAfter_Cancellation_ThrowsException_Async()
             completed = true;
         };
 
-        var cancelTask = Task.Run(async () =>
-        {
-            await Task.Delay(100);
-            await cancellationTokenSource.CancelAsync();
-        });
+        var cancelTask = Task.Run(
+            async () =>
+            {
+                await Task.Delay(100);
+                await cancellationTokenSource.CancelAsync();
+            },
+            TestContext.Current.CancellationToken
+        );
 
         await Assert.ThrowsAsync<OperationCanceledException>(
-            async () => await Task.Run(task).TimeoutAfter(1000, cancellationTokenSource)
+            async () =>
+                await Task.Run(task, TestContext.Current.CancellationToken)
+                    .TimeoutAfter(1000, cancellationTokenSource)
         );
         Assert.False(completed);
         await cancelTask;
@@ -107,11 +112,14 @@ public async Task TimeoutAfter_CancellationWithReturn_ThrowsException_Async()
             return "Hello";
         };
 
-        var cancelTask = Task.Run(async () =>
-        {
-            await Task.Delay(100);
-            await cancellationTokenSource.CancelAsync();
-        });
+        var cancelTask = Task.Run(
+            async () =>
+            {
+                await Task.Delay(100);
+                await cancellationTokenSource.CancelAsync();
+            },
+            TestContext.Current.CancellationToken
+        );
 
         await Assert.ThrowsAsync<OperationCanceledException>(
             async () => await Task.Run(task).TimeoutAfter(1000, cancellationTokenSource)