fix tests

Azure · Oct 18, 2023 · 8a3427f · 8a3427f
1 parent e2e8f7e
commit 8a3427f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/Microsoft.Azure.SignalR.Management.Tests/Resilient/HttpClientRetryFacts.cs b/test/Microsoft.Azure.SignalR.Management.Tests/Resilient/HttpClientRetryFacts.cs
@@ -156,8 +156,9 @@ private static async Task FixedDelayRetryTestCore(Func<HttpHandlerSetup, Moq.Lan
                 .ConfigurePrimaryHttpMessageHandler(sp => handlerMock.Object))
             .BuildServiceManager()
             .CreateHubContextAsync(HubName, default);
-        var exception = await Assert.ThrowsAnyAsync<AggregateException>(() => testAction(hubContext));
-        assert(exception);
+        var exception = await Assert.ThrowsAnyAsync<AzureSignalRRuntimeException>(() => testAction(hubContext));
+        var aggregationException = Assert.IsType<AggregateException>(exception.InnerException);
+        assert(aggregationException);
 
         handlerMock.Protected().Verify("SendAsync", Times.Exactly(4), ItExpr.IsAny<HttpRequestMessage>(), ItExpr.IsAny<CancellationToken>());
     }