Fixed client builder tests that became flaky with the new version of …

…Azure Core. (#27985)
Azure · Mar 31, 2022 · 3b14f98 · 3b14f98
1 parent 852c9ea
commit 3b14f98
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TableClientBuilderTest.java b/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TableClientBuilderTest.java
@@ -8,6 +8,7 @@
 import com.azure.core.exception.HttpResponseException;
 import com.azure.core.http.HttpPipeline;
 import com.azure.core.http.policy.ExponentialBackoffOptions;
+import com.azure.core.http.policy.FixedDelayOptions;
 import com.azure.core.http.policy.HttpLogOptions;
 import com.azure.core.http.policy.RetryOptions;
 import com.azure.core.http.policy.RetryPolicy;
@@ -21,6 +22,7 @@
 
 import java.net.MalformedURLException;
 import java.security.SecureRandom;
+import java.time.Duration;
 import java.util.Collections;
 
 import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
@@ -107,6 +109,7 @@ public void serviceClientFreshDateOnRetry() throws MalformedURLException {
             .connectionString(connectionString)
             .tableName(tableName)
             .httpClient(new TestUtils.FreshDateTestClient())
+            .retryOptions(new RetryOptions(new FixedDelayOptions(3, Duration.ofSeconds(1))))
             .buildAsyncClient();
 
         StepVerifier.create(tableAsyncClient.getHttpPipeline().send(

diff --git a/.../azure-data-tables/src/test/java/com/azure/data/tables/TableServiceClientBuilderTest.java b/.../azure-data-tables/src/test/java/com/azure/data/tables/TableServiceClientBuilderTest.java
@@ -8,6 +8,7 @@
 import com.azure.core.exception.HttpResponseException;
 import com.azure.core.http.HttpPipeline;
 import com.azure.core.http.policy.ExponentialBackoffOptions;
+import com.azure.core.http.policy.FixedDelayOptions;
 import com.azure.core.http.policy.HttpLogOptions;
 import com.azure.core.http.policy.RetryOptions;
 import com.azure.core.http.policy.RetryPolicy;
@@ -21,6 +22,7 @@
 
 import java.net.MalformedURLException;
 import java.security.SecureRandom;
+import java.time.Duration;
 import java.util.Collections;
 
 import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
@@ -102,6 +104,7 @@ public void serviceClientFreshDateOnRetry() throws MalformedURLException {
         TableServiceAsyncClient tableServiceAsyncClient = new TableServiceClientBuilder()
             .connectionString(connectionString)
             .httpClient(new TestUtils.FreshDateTestClient())
+            .retryOptions(new RetryOptions(new FixedDelayOptions(3, Duration.ofSeconds(1))))
             .buildAsyncClient();
 
         StepVerifier.create(tableServiceAsyncClient.getHttpPipeline().send(