Migrate Digital Twins tests to TestProxy (Azure#35997)

srnagar · Jul 20, 2023 · 4faf9ee · 4faf9ee
1 parent 3914164
commit 4faf9ee
Show file tree

Hide file tree

Showing 64 changed files with 12 additions and 23,965 deletions.
diff --git a/sdk/digitaltwins/azure-digitaltwins-core/assets.json b/sdk/digitaltwins/azure-digitaltwins-core/assets.json
@@ -0,0 +1,6 @@
+{
+  "AssetsRepo": "Azure/azure-sdk-assets",
+  "AssetsRepoPrefixPath": "java",
+  "TagPrefix": "java/digitaltwins/azure-digitaltwins-core",
+  "Tag": "java/digitaltwins/azure-digitaltwins-core_329ebe0bf8"
+}
diff --git a/...ure-digitaltwins-core/src/test/java/com/azure/digitaltwins/core/DigitalTwinsTestBase.java b/...ure-digitaltwins-core/src/test/java/com/azure/digitaltwins/core/DigitalTwinsTestBase.java
@@ -10,8 +10,9 @@
 import com.azure.core.http.policy.HttpLogDetailLevel;
 import com.azure.core.http.policy.HttpLogOptions;
 import com.azure.core.http.policy.HttpPipelinePolicy;
-import com.azure.core.test.TestBase;
 import com.azure.core.test.TestMode;
+import com.azure.core.test.TestProxyTestBase;
+import com.azure.core.test.models.CustomMatcher;
 import com.azure.core.util.Configuration;
 import com.azure.identity.ClientSecretCredentialBuilder;
 import com.fasterxml.jackson.core.JsonProcessingException;
@@ -20,9 +21,10 @@
 import reactor.core.publisher.Mono;
 
 import java.time.OffsetDateTime;
+import java.util.Arrays;
 import java.util.function.Function;
 
-public class DigitalTwinsTestBase extends TestBase {
+public class DigitalTwinsTestBase extends TestProxyTestBase {
     private static final String PLAYBACK_ENDPOINT = "https://playback.api.wus2.digitaltwins.azure.net";
     private static final int DEFAULT_WAIT_TIME_IN_SECONDS = 5;
 
@@ -49,6 +51,8 @@ protected DigitalTwinsClientBuilder getDigitalTwinsClientBuilder(HttpClient http
             builder.credential(new FakeCredentials());
             // Connect to a special host when running tests in playback mode.
             builder.endpoint(PLAYBACK_ENDPOINT);
+            interceptorManager.addMatchers(Arrays.asList(new CustomMatcher().setHeadersKeyOnlyMatch(Arrays.asList("Telemetry-Source-Time"))));
+
             return builder;
         }
 

diff --git a/...re/src/test/resources/session-records/ComponentsAsyncTests.componentLifecycleTest[1].json b/...re/src/test/resources/session-records/ComponentsAsyncTests.componentLifecycleTest[1].json