Record playback integration

batch/data-plane/pom.xml

@@ -68,6 +68,19 @@
       <scope>test</scope>
     </dependency>
 
+    <dependency>
+      <groupId>com.microsoft.azure</groupId>
+      <artifactId>azure-mgmt-resources</artifactId>
+      <type>test-jar</type>
+      <scope>test</scope>
+    </dependency>
+
+    <dependency>
+      <groupId>com.microsoft.azure</groupId>
+      <artifactId>azure-mgmt-resources</artifactId>
+      <scope>test</scope>
+    </dependency>
+
   </dependencies>
   <build>
     <plugins>
@@ -83,6 +96,19 @@
         </configuration>
       </plugin>
 
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-resources-plugin</artifactId>
+        <configuration>
+          <outputDirectory>${basedir}/target/test-classes/session-records</outputDirectory>
+          <resources>
+            <resource>
+              <directory>${basedir}/src/test/resources/test-recordings</directory>
+            </resource>
+          </resources>
+        </configuration>
+      </plugin>
+
     </plugins>
 
   </build>

batch/data-plane/src/main/java/com/microsoft/azure/batch/BatchClient.java

@@ -45,17 +45,18 @@ public BatchServiceClient protocolLayer() {
 
     private BatchClient(BatchCredentials credentials) {
         RestClient restClient = new RestClient.Builder()
-            .withBaseUrl(credentials.baseUrl())
-            .withCredentials(credentials)
-            .withSerializerAdapter(new AzureJacksonAdapter())
-            .withResponseBuilderFactory(new ResponseBuilder.Factory() {
-                private final AzureResponseBuilder.Factory baseFactory = new AzureResponseBuilder.Factory();
-                @Override
-                public <T, E extends RestException> ResponseBuilder<T, E> newInstance(SerializerAdapter<?> serializerAdapter) {
-                    return baseFactory.<T, E>newInstance(serializerAdapter).withThrowOnGet404(true);
-                }
-            })
-            .build();
+                .withBaseUrl(credentials.baseUrl())
+                .withCredentials(credentials)
+                .withSerializerAdapter(new AzureJacksonAdapter())
+                .withResponseBuilderFactory(new ResponseBuilder.Factory() {
+                    private final AzureResponseBuilder.Factory baseFactory = new AzureResponseBuilder.Factory();
+                    @Override
+                    public <T, E extends RestException> ResponseBuilder<T, E> newInstance(SerializerAdapter<?> serializerAdapter) {
+                        return baseFactory.<T, E>newInstance(serializerAdapter).withThrowOnGet404(true);
+                    }
+                })
+                .build();
+
         this.protocolLayer = new BatchServiceClientImpl(restClient).withBatchUrl(credentials.baseUrl());
         this.customBehaviors = new LinkedList<>();
         this.customBehaviors.add(new ClientRequestIdInterceptor());
@@ -70,6 +71,21 @@ public <T, E extends RestException> ResponseBuilder<T, E> newInstance(Serializer
         this.computeNodeOperations = new ComputeNodeOperations(this, customBehaviors());
     }
 
+    private BatchClient(RestClient restClient, String baseUrl) {
+        this.protocolLayer = new BatchServiceClientImpl(restClient).withBatchUrl(baseUrl);
+        this.customBehaviors = new LinkedList<>();
+        this.customBehaviors.add(new ClientRequestIdInterceptor());
+        this.certificateOperations = new CertificateOperations(this, customBehaviors());
+        this.jobOperations = new JobOperations(this, customBehaviors());
+        this.taskOperations = new TaskOperations(this, customBehaviors());
+        this.jobScheduleOperations = new JobScheduleOperations(this, customBehaviors());
+        this.poolOperations = new PoolOperations(this, customBehaviors());
+        this.fileOperations = new FileOperations(this, customBehaviors());
+        this.applicationOperations = new ApplicationOperations(this, customBehaviors());
+        this.accountOperations = new AccountOperations(this, customBehaviors());
+        this.computeNodeOperations = new ComputeNodeOperations(this, customBehaviors());
+    }
+
     /**
      * Creates an instance of {@link BatchClient} associated with the specified credentials.
      *
@@ -80,6 +96,18 @@ public static BatchClient open(BatchCredentials credentials) {
         return new BatchClient(credentials);
     }
 
+    /**
+     * Creates an instance of {@link BatchClient} associated with the specified restClient and baseurl.
+     *
+     * @param restClient A {@link RestClient} object specifying the definition of rest client used to
+     *                   interact with batch endpoint.
+     * @param baseUrl A  String object specifying the the batch end point.
+     * @return The new {@link BatchClient} instance.
+     */
+    static BatchClient open(RestClient restClient, String baseUrl) {
+        return new BatchClient(restClient, baseUrl);
+    }
+
     /**
      * Gets a {@link CertificateOperations} object for performing certificate-related operations on the associated account.
      *