[Perf] Call configureClientBuilder() in DataLake and FileShare tests (#…

…31324)
Azure · Oct 11, 2022 · 1c2a0d9 · 1c2a0d9
1 parent c85090e
commit 1c2a0d9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/...ure-storage-perf/src/main/java/com/azure/storage/file/datalake/perf/core/ServiceTest.java b/...ure-storage-perf/src/main/java/com/azure/storage/file/datalake/perf/core/ServiceTest.java
@@ -33,16 +33,14 @@ public ServiceTest(TOptions options) {
             = StorageConnectionString.create(connectionString, null);
         StorageEndpoint endpoint = storageConnectionString.getBlobEndpoint();
 
-        dataLakeServiceClient = new DataLakeServiceClientBuilder()
+        DataLakeServiceClientBuilder builder = new DataLakeServiceClientBuilder()
             .endpoint(endpoint.getPrimaryUri())
             .credential(new StorageSharedKeyCredential(storageConnectionString.getAccountName(),
-                storageConnectionString.getStorageAuthSettings().getAccount().getAccessKey()))
-            .buildClient();
+                storageConnectionString.getStorageAuthSettings().getAccount().getAccessKey()));
 
-        dataLakeServiceAsyncClient = new DataLakeServiceClientBuilder()
-            .endpoint(endpoint.getPrimaryUri())
-            .credential(new StorageSharedKeyCredential(storageConnectionString.getAccountName(),
-                storageConnectionString.getStorageAuthSettings().getAccount().getAccessKey()))
-            .buildAsyncClient();
+        configureClientBuilder(builder);
+
+        dataLakeServiceClient = builder.buildClient();
+        dataLakeServiceAsyncClient = builder.buildAsyncClient();
     }
 }
diff --git a/.../azure-storage-perf/src/main/java/com/azure/storage/file/share/perf/core/ServiceTest.java b/.../azure-storage-perf/src/main/java/com/azure/storage/file/share/perf/core/ServiceTest.java
@@ -26,10 +26,12 @@ public ServiceTest(TOptions options) {
             throw new IllegalStateException("Environment variable STORAGE_CONNECTION_STRING must be set");
         }
 
-        shareServiceClient = new ShareServiceClientBuilder().connectionString(connectionString).
-            buildClient();
+        ShareServiceClientBuilder builder = new ShareServiceClientBuilder()
+            .connectionString(connectionString);
 
-        shareServiceAsyncClient = new ShareServiceClientBuilder().connectionString(connectionString).
-            buildAsyncClient();
+        configureClientBuilder(builder);
+
+        shareServiceClient = builder.buildClient();
+        shareServiceAsyncClient = builder.buildAsyncClient();
     }
 }