Update test since EnsureValidClient does not retry 2 times now.

Azure · Feb 12, 2022 · 64681b7 · 64681b7
1 parent 26b9b08
commit 64681b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/Microsoft.Azure.Cosmos/src/DocumentClient.cs b/Microsoft.Azure.Cosmos/src/DocumentClient.cs
@@ -1254,6 +1254,7 @@ public void Dispose()
             if (this.initTaskCache != null)
             {
                 this.initTaskCache.Dispose();
+                this.initTaskCache = null;
             }
 
             DefaultTrace.TraceInformation("DocumentClient with id {0} disposed.", this.traceId);

diff --git a/...soft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/CosmosGatewayTimeoutTests.cs b/...soft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/CosmosGatewayTimeoutTests.cs
@@ -106,7 +106,7 @@ public async Task CosmosHttpClientRetryValidation()
                 }
                 catch (CosmosException rte)
                 {
-                    Assert.IsTrue(handler.Count >= 6);
+                    Assert.IsTrue(handler.Count >= 3, $"HandlerCount: {handler.Count}; Expecte 6");
                     string message = rte.ToString();
                     Assert.IsTrue(message.Contains("Start Time"), "Start Time:" + message);
                     Assert.IsTrue(message.Contains("Total Duration"), "Total Duration:" + message);
@@ -129,9 +129,10 @@ private class TransientHttpClientCreatorHandler : DelegatingHandler
 
             protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
             {
-                if (this.Count++ <= 3)
+                this.Count++;
+                if (this.Count < 3)
                 {
-                    throw new WebException();
+                    throw new WebException($"Mocked WebException {this.Count}");
                 }
 
                 throw new TaskCanceledException();