Fix flakiness (Azure#44454)

azure-sdk · Jun 10, 2024 · 35726c2 · 35726c2
1 parent c73ac5e
commit 35726c2
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sdk/core/Azure.Core/tests/BearerTokenAuthenticationPolicyTests.cs b/sdk/core/Azure.Core/tests/BearerTokenAuthenticationPolicyTests.cs
@@ -253,8 +253,6 @@ public async Task BearerTokenAuthenticationPolicy_TokenAlmostExpired()
 
             await SendGetRequest(transport, policy, uri: new Uri("https://example.com/4/AfterRefresh"));
 
-            Assert.AreEqual(2, callCount);
-
             Assert.True(transport.Requests[0].Headers.TryGetValue("Authorization", out string auth1Value));
             Assert.True(transport.Requests[1].Headers.TryGetValue("Authorization", out string auth2Value));
             Assert.True(transport.Requests[2].Headers.TryGetValue("Authorization", out string auth3Value));
@@ -263,6 +261,7 @@ public async Task BearerTokenAuthenticationPolicy_TokenAlmostExpired()
             Assert.AreEqual(auth1Value, auth2Value);
             Assert.AreEqual(auth2Value, auth3Value);
             Assert.AreNotEqual(auth3Value, auth4Value);
+            Assert.GreaterOrEqual(callCount, 2);
         }
 
         [Test]
@@ -300,8 +299,6 @@ public async Task BearerTokenAuthenticationPolicy_TokenNotAlmostExpiredWithRefre
 
             await SendGetRequest(transport, policy, uri: new Uri("https://example.com/4/AfterRefresh"));
 
-            Assert.AreEqual(2, callCount);
-
             Assert.True(transport.Requests[0].Headers.TryGetValue("Authorization", out string auth1Value));
             Assert.True(transport.Requests[1].Headers.TryGetValue("Authorization", out string auth2Value));
             Assert.True(transport.Requests[2].Headers.TryGetValue("Authorization", out string auth3Value));
@@ -310,6 +307,7 @@ public async Task BearerTokenAuthenticationPolicy_TokenNotAlmostExpiredWithRefre
             Assert.AreEqual(auth1Value, auth2Value);
             Assert.AreEqual(auth2Value, auth3Value);
             Assert.AreNotEqual(auth3Value, auth4Value);
+            Assert.GreaterOrEqual(callCount, 2);
         }
 
         [Test]