[Purview-Scanning] TestProxy Migration (Azure#35851)

Author: mssfang

sdk/purview/azure-analytics-purview-scanning/assets.json

@@ -0,0 +1,6 @@
+{
+  "AssetsRepo": "Azure/azure-sdk-assets",
+  "AssetsRepoPrefixPath": "java",
+  "TagPrefix": "java/purview/azure-analytics-purview-scanning",
+  "Tag": "java/purview/azure-analytics-purview-scanning_fea6bf1c51"
+}

sdk/purview/azure-analytics-purview-scanning/pom.xml

@@ -34,7 +34,7 @@
     <jacoco.min.linecoverage>0.1</jacoco.min.linecoverage>
     <jacoco.min.branchcoverage>0.05</jacoco.min.branchcoverage>
     <javaModulesSurefireArgLine>
-      --add-exports com.azure.core/com.azure.core.implementation.jackson=ALL-UNNAMED
+      --add-exports com.azure.core/com.azure.core.implementation.util=ALL-UNNAMED
     </javaModulesSurefireArgLine>
   </properties>
 

sdk/purview/azure-analytics-purview-scanning/src/test/java/com/azure/analytics/purview/scanning/PurviewScanningClientTestBase.java

@@ -3,20 +3,15 @@
 
 package com.azure.analytics.purview.scanning;
 
-import com.azure.core.credential.AccessToken;
 import com.azure.core.http.HttpClient;
-import com.azure.core.http.policy.HttpLogDetailLevel;
-import com.azure.core.http.policy.HttpLogOptions;
-import com.azure.core.test.TestBase;
-import com.azure.core.test.TestMode;
+import com.azure.core.test.TestProxyTestBase;
+import com.azure.core.test.utils.MockTokenCredential;
 import com.azure.core.util.Configuration;
 import com.azure.identity.DefaultAzureCredentialBuilder;
-import reactor.core.publisher.Mono;
 
-import java.time.OffsetDateTime;
 import java.util.Objects;
 
-public class PurviewScanningClientTestBase extends TestBase {
+public class PurviewScanningClientTestBase extends TestProxyTestBase {
 
     protected String getEndpoint() {
         String endpoint = interceptorManager.isPlaybackMode()
@@ -27,22 +22,23 @@ protected String getEndpoint() {
     }
 
     PurviewScanningClientBuilder builderSetUp() {
-        PurviewScanningClientBuilder builder =
-            new PurviewScanningClientBuilder()
+        PurviewScanningClientBuilder builder = new PurviewScanningClientBuilder();
+        if (interceptorManager.isPlaybackMode()) {
+            builder
+                .httpClient(interceptorManager.getPlaybackClient())
+                .credential(new MockTokenCredential());
+        } else {
+            builder
                 .httpClient(HttpClient.createDefault())
-                .httpLogOptions(new HttpLogOptions().setLogLevel(HttpLogDetailLevel.BODY_AND_HEADERS));
-        if (getTestMode() == TestMode.PLAYBACK) {
-            builder.httpClient(interceptorManager.getPlaybackClient())
-                .credential(request -> Mono.just(new AccessToken("this_is_a_token", OffsetDateTime.MAX)));
-        } else if (getTestMode() == TestMode.RECORD) {
-            builder.addPolicy(interceptorManager.getRecordPolicy())
                 .credential(new DefaultAzureCredentialBuilder().build());
-        } else if (getTestMode() == TestMode.LIVE) {
-            builder.credential(new DefaultAzureCredentialBuilder().build());
         }
-        return Objects.requireNonNull(builder);
-    }
 
+        if (interceptorManager.isRecordMode()) {
+            builder.addPolicy(interceptorManager.getRecordPolicy());
+        }
 
+        builder.endpoint(getEndpoint());
 
+        return Objects.requireNonNull(builder);
+    }
 }

sdk/purview/azure-analytics-purview-scanning/src/test/java/com/azure/analytics/purview/scanning/SystemScanRulesetsClientTests.java

@@ -17,11 +17,9 @@ public class SystemScanRulesetsClientTests extends PurviewScanningClientTestBase
 
     @Override
     protected void beforeTest() {
-        PurviewScanningClientBuilder builder = builderSetUp();
-        client = builder.endpoint(getEndpoint()).buildSystemScanRulesetsClient();
+        client = builderSetUp().buildSystemScanRulesetsClient();
     }
 
-
     @Test
     public void testListAll() {
         PagedIterable<BinaryData> response = client.listAll(null);