fix test cases

Azure · Oct 24, 2023 · 7b4bbff · 7b4bbff
1 parent cd929a9
commit 7b4bbff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/Microsoft.Azure.SignalR.Common.Tests/Auth/AadAccessKeyTests.cs b/test/Microsoft.Azure.SignalR.Common.Tests/Auth/AadAccessKeyTests.cs
@@ -81,7 +81,7 @@ public async Task TestUpdateAccessKeyShouldSkip(bool isAuthorized, int timeElaps
             }
             else
             {
-                await Assert.ThrowsAsync<TaskCanceledException>(async () => await key.UpdateAccessKeyAsync(source.Token));
+                await Assert.ThrowsAsync<OperationCanceledException>(async () => await key.UpdateAccessKeyAsync(source.Token));
                 Assert.False((bool)isAuthorizedField.GetValue(key));
                 Assert.True(lastUpdatedTime < (DateTime)lastUpdatedTimeField.GetValue(key));
                 Assert.True(initializedTcs.Task.IsCompleted);
@@ -104,7 +104,7 @@ public async Task TestInitializeFailed()
             );
 
             var source = new CancellationTokenSource(TimeSpan.FromSeconds(1));
-            await Assert.ThrowsAnyAsync<TaskCanceledException>(
+            await Assert.ThrowsAnyAsync<OperationCanceledException>(
                 async () => await key.UpdateAccessKeyAsync(source.Token)
             );