Fix LIVE mode

Azure · Sep 4, 2020 · f618bc9 · f618bc9
1 parent ae88645
commit f618bc9
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 14 deletions.
diff --git a/...es/azure-data-tables/src/test/java/com/azure/data/tables/TableServiceAsyncClientTest.java b/...es/azure-data-tables/src/test/java/com/azure/data/tables/TableServiceAsyncClientTest.java
@@ -49,9 +49,11 @@ protected void beforeTest() {
         if (interceptorManager.isPlaybackMode()) {
             builder.httpClient(interceptorManager.getPlaybackClient());
         } else {
-            builder.httpClient(HttpClient.createDefault())
-                .addPolicy(interceptorManager.getRecordPolicy())
-                .addPolicy(new RetryPolicy());
+            builder.httpClient(HttpClient.createDefault());
+            if (!interceptorManager.isLiveMode()) {
+                builder.addPolicy(interceptorManager.getRecordPolicy());
+            }
+            builder.addPolicy(new RetryPolicy());
         }
 
         serviceClient = builder.buildAsyncClient();
@@ -233,7 +235,7 @@ void serviceListTablesWithTopAsync() {
             serviceClient.createTable(tableName2),
             serviceClient.createTable(tableName3)
         ).block(TIMEOUT);
-        
+
         // Act & Assert
         StepVerifier.create(serviceClient.listTables(options))
             .expectNextCount(2)

diff --git a/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TableServiceClientTest.java b/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TableServiceClientTest.java
@@ -25,9 +25,11 @@ protected void beforeTest() {
         if (interceptorManager.isPlaybackMode()) {
             builder.httpClient(interceptorManager.getPlaybackClient());
         } else {
-            builder.httpClient(HttpClient.createDefault())
-                .addPolicy(interceptorManager.getRecordPolicy())
-                .addPolicy(new RetryPolicy());
+            builder.httpClient(HttpClient.createDefault());
+            if (!interceptorManager.isLiveMode()) {
+                builder.addPolicy(interceptorManager.getRecordPolicy());
+            }
+            builder.addPolicy(new RetryPolicy());
         }
 
         serviceClient = builder.buildClient();

diff --git a/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TablesAsyncClientTest.java b/sdk/tables/azure-data-tables/src/test/java/com/azure/data/tables/TablesAsyncClientTest.java
@@ -51,10 +51,11 @@ protected void beforeTest() {
             playbackClient = interceptorManager.getPlaybackClient();
             builder.httpClient(playbackClient);
         } else {
-            recordPolicy = interceptorManager.getRecordPolicy();
-            builder.httpClient(HttpClient.createDefault())
-                .addPolicy(recordPolicy)
-                .addPolicy(new RetryPolicy());
+            builder.httpClient(HttpClient.createDefault());
+            if (!interceptorManager.isLiveMode()) {
+                builder.addPolicy(interceptorManager.getRecordPolicy());
+            }
+            builder.addPolicy(new RetryPolicy());
         }
 
         tableClient = builder.buildAsyncClient();

diff --git a/...re-data-tables/src/test/java/com/azure/data/tables/implementation/AzureTableImplTest.java b/...re-data-tables/src/test/java/com/azure/data/tables/implementation/AzureTableImplTest.java
@@ -99,8 +99,10 @@ protected void beforeTest() {
             httpClientToUse = interceptorManager.getPlaybackClient();
         } else {
             httpClientToUse = HttpClient.createDefault();
-            HttpPipelinePolicy recordPolicy = interceptorManager.getRecordPolicy();
-            policies.add(recordPolicy);
+            if (!interceptorManager.isLiveMode()) {
+                HttpPipelinePolicy recordPolicy = interceptorManager.getRecordPolicy();
+                policies.add(recordPolicy);
+            }
             policies.add(new RetryPolicy());
         }
 

diff --git a/sdk/tables/tests.yml b/sdk/tables/tests.yml
@@ -7,7 +7,7 @@ jobs:
       Timeout: 60
       MaxParallel: 12
       EnvVars:
-        AZURE_TEST_MODE: RECORD
+        AZURE_TEST_MODE: LIVE
       Artifacts:
         - name: azure-data-tables
           groupId: com.azure