Enabling integration/live tests for Event Hubs V2 stack (Azure#41812)

mssfang · Oct 21, 2024 · 4483e63 · 4483e63
1 parent 3de8b02
commit 4483e63
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...-messaging-eventhubs/src/test/java/com/azure/messaging/eventhubs/IntegrationTestBase.java b/...-messaging-eventhubs/src/test/java/com/azure/messaging/eventhubs/IntegrationTestBase.java
@@ -12,6 +12,7 @@
 import com.azure.core.test.TestContextManager;
 import com.azure.core.test.TestMode;
 import com.azure.core.util.ClientOptions;
+import com.azure.core.util.ConfigurationBuilder;
 import com.azure.core.util.CoreUtils;
 import com.azure.core.util.logging.ClientLogger;
 import com.azure.identity.DefaultAzureCredentialBuilder;
@@ -134,7 +135,10 @@ protected EventHubClientBuilder createBuilder(boolean shareConnection) {
             .retryOptions(RETRY_OPTIONS)
             .clientOptions(OPTIONS_WITH_TRACING)
             .transportType(AmqpTransportType.AMQP)
-            .scheduler(scheduler);
+            .scheduler(scheduler)
+            .configuration(new ConfigurationBuilder()
+                .putProperty("com.azure.messaging.eventhubs.v2", "true")
+                .build());
 
         final String fullyQualifiedDomainName = TestUtils.getFullyQualifiedDomainName();
         final String eventHubName = TestUtils.getEventHubName();