use another more specific exception

specklesystems · Jan 8, 2025 · 81c1d32 · 81c1d32
1 parent 93a6447
commit 81c1d32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Speckle.Sdk.Tests.Integration/Api/GraphQL/GraphQLClientExceptionHandling.cs b/tests/Speckle.Sdk.Tests.Integration/Api/GraphQL/GraphQLClientExceptionHandling.cs
@@ -90,9 +90,9 @@ public async Task TestGraphQLLayer_BadInput()
   public async Task TestCancel()
   {
     using CancellationTokenSource cts = new();
-    cts.Cancel();
+    await cts.CancelAsync();
 
-    var ex = await Assert.ThrowsAsync<OperationCanceledException>(
+    var ex = await Assert.ThrowsAsync<TaskCanceledException>(
       async () => await _sut.ActiveUser.Get(cts.Token).ConfigureAwait(false)
     );