BatchRequestTest.java

// Copyright 2012 Google Inc. All Rights Reserved.

package com.google.api.client.googleapis.batch;

import static java.nio.charset.StandardCharsets.UTF_8;

import com.google.api.client.googleapis.batch.BatchRequest.RequestInfo;
import com.google.api.client.googleapis.json.GoogleJsonError;
import com.google.api.client.googleapis.json.GoogleJsonError.ErrorInfo;
import com.google.api.client.googleapis.json.GoogleJsonErrorContainer;
import com.google.api.client.googleapis.testing.services.MockGoogleClient;
import com.google.api.client.googleapis.testing.services.MockGoogleClientRequest;
import com.google.api.client.http.ByteArrayContent;
import com.google.api.client.http.ExponentialBackOffPolicy;
import com.google.api.client.http.GenericUrl;
import com.google.api.client.http.HttpContent;
import com.google.api.client.http.HttpExecuteInterceptor;
import com.google.api.client.http.HttpHeaders;
import com.google.api.client.http.HttpMethods;
import com.google.api.client.http.HttpRequest;
import com.google.api.client.http.HttpRequestInitializer;
import com.google.api.client.http.HttpResponse;
import com.google.api.client.http.HttpUnsuccessfulResponseHandler;
import com.google.api.client.http.LowLevelHttpRequest;
import com.google.api.client.http.LowLevelHttpResponse;
import com.google.api.client.json.GenericJson;
import com.google.api.client.json.JsonObjectParser;
import com.google.api.client.json.gson.GsonFactory;
import com.google.api.client.protobuf.ProtoObjectParser;
import com.google.api.client.testing.http.HttpTesting;
import com.google.api.client.testing.http.MockHttpTransport;
import com.google.api.client.testing.http.MockLowLevelHttpRequest;
import com.google.api.client.testing.http.MockLowLevelHttpResponse;
import com.google.api.client.util.Charsets;
import com.google.api.client.util.Key;
import com.google.api.client.util.ObjectParser;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.List;
import junit.framework.TestCase;

/**
 * Tests {@link BatchRequest}.
 *
 * @author rmistry@google.com (Ravi Mistry)
 */
public class BatchRequestTest extends TestCase {

  private static final String ROOT_URL = "http://www.test.com/";
  private static final String SERVICE_PATH = "test/";
  private static final String TEST_BATCH_URL = "http://www.testgoogleapis.com/batch";
  private static final String URI_TEMPLATE1 = "uri/template/1";
  private static final String URI_TEMPLATE2 = "uri/template/2";
  private static final String METHOD1 = HttpMethods.GET;
  private static final String METHOD2 = HttpMethods.POST;
  private static final String ERROR_MSG = "Error message";
  private static final String ERROR_REASON = "notFound";
  private static final int ERROR_CODE = 503;
  private static final String ERROR_DOMAIN = "global";
  private static final String RESPONSE_BOUNDARY = "ABC=DE=F";
  private static final String TEST_ID = "Humpty Dumpty";
  private static final String TEST_KIND = "Big\nEgg\n"; // Newlines help test boundary detection
  private static final String TEST_NAME = "James Bond";
  private static final String TEST_NUM = "007";

  private TestCallback1 callback1;
  private TestCallback2 callback2;
  private TestCallback3 callback3;

  private MockTransport transport;

  private MockCredential credential;

  @Override
  protected void setUp() {
    callback1 = new TestCallback1();
    callback2 = new TestCallback2();
    callback3 = new TestCallback3();
  }

  public static class MockDataClass1 extends GenericJson {
    @Key String id;

    @Key String kind;
  }

  public static class MockDataClass2 extends GenericJson {
    @Key String name;

    @Key String number;
  }

  private static class TestCallback1
      implements BatchCallback<MockDataClass1, GoogleJsonErrorContainer> {

    int successCalls;

    TestCallback1() {}

    @Override
    public void onSuccess(MockDataClass1 dataClass, HttpHeaders responseHeaders) {
      successCalls++;
      assertEquals(TEST_ID, dataClass.id);
      assertEquals(TEST_KIND, dataClass.kind);
    }

    @Override
    public void onFailure(GoogleJsonErrorContainer e, HttpHeaders responseHeaders) {
      fail("Should not be invoked in this test");
    }
  }

  private static class TestCallback2
      implements BatchCallback<MockDataClass2, GoogleJsonErrorContainer> {

    int successCalls;
    int failureCalls;

    TestCallback2() {}

    @Override
    public void onSuccess(MockDataClass2 dataClass, HttpHeaders responseHeaders) {
      successCalls++;
      assertEquals(TEST_NAME, dataClass.name);
      assertEquals(TEST_NUM, dataClass.number);
    }

    @Override
    public void onFailure(GoogleJsonErrorContainer e, HttpHeaders responseHeaders) {
      failureCalls++;
      GoogleJsonError error = e.getError();
      ErrorInfo errorInfo = error.getErrors().get(0);
      assertEquals(ERROR_DOMAIN, errorInfo.getDomain());
      assertEquals(ERROR_REASON, errorInfo.getReason());
      assertEquals(ERROR_MSG, errorInfo.getMessage());
      assertEquals(ERROR_CODE, error.getCode());
      assertEquals(ERROR_MSG, error.getMessage());
    }
  }

  private static class TestCallback3 implements BatchCallback<Void, Void> {

    int successCalls;
    int failureCalls;

    TestCallback3() {}

    @Override
    public void onSuccess(Void dataClass, HttpHeaders responseHeaders) {
      successCalls++;
      assertNull(dataClass);
    }

    @Override
    public void onFailure(Void e, HttpHeaders responseHeaders) {
      failureCalls++;
      assertNull(e);
    }
  }

  /**
   * Base class for callback adapters to handle error conversion.
   *
   * @param <InputType> The input type
   * @param <OutputType> The output type
   */
  private abstract static class TestCallbackBaseAdapter<InputType, OutputType>
      implements BatchCallback<InputType, ErrorOutput.ErrorBody> {

    protected final BatchCallback<OutputType, GoogleJsonErrorContainer> callback;

    protected TestCallbackBaseAdapter(
        BatchCallback<OutputType, GoogleJsonErrorContainer> callback) {
      this.callback = callback;
    }

    @Override
    public void onFailure(ErrorOutput.ErrorBody e, HttpHeaders responseHeaders) throws IOException {
      GoogleJsonErrorContainer errorContainer = new GoogleJsonErrorContainer();

      if (e.hasError()) {
        ErrorOutput.ErrorProto errorProto = e.getError();

        GoogleJsonError error = new GoogleJsonError();
        if (errorProto.hasCode()) {
          error.setCode(errorProto.getCode());
        }
        if (errorProto.hasMessage()) {
          error.setMessage(errorProto.getMessage());
        }

        List<ErrorInfo> errorInfos = new ArrayList<ErrorInfo>(errorProto.getErrorsCount());
        for (ErrorOutput.IndividualError individualError : errorProto.getErrorsList()) {
          ErrorInfo errorInfo = new ErrorInfo();
          if (individualError.hasDomain()) {
            errorInfo.setDomain(individualError.getDomain());
          }
          if (individualError.hasMessage()) {
            errorInfo.setMessage(individualError.getMessage());
          }
          if (individualError.hasReason()) {
            errorInfo.setReason(individualError.getReason());
          }
          errorInfos.add(errorInfo);
        }
        error.setErrors(errorInfos);
        errorContainer.setError(error);
      }
      callback.onFailure(errorContainer, responseHeaders);
    }
  }

  private static class TestCallback1Adapter
      extends TestCallbackBaseAdapter<MockData.Class1, MockDataClass1> {

    public TestCallback1Adapter(TestCallback1 callback) {
      super(callback);
    }

    @Override
    public void onSuccess(MockData.Class1 message, HttpHeaders responseHeaders) throws IOException {
      MockDataClass1 dataClass = new MockDataClass1();
      dataClass.id = message.hasId() ? message.getId() : null;
      dataClass.kind = message.hasKind() ? message.getKind() : null;
      callback.onSuccess(dataClass, responseHeaders);
    }
  }

  private static class TestCallback2Adapter
      extends TestCallbackBaseAdapter<MockData.Class2, MockDataClass2> {

    public TestCallback2Adapter(TestCallback2 callback) {
      super(callback);
    }

    @Override
    public void onSuccess(MockData.Class2 message, HttpHeaders responseHeaders) throws IOException {
      MockDataClass2 dataClass = new MockDataClass2();
      dataClass.name = message.hasName() ? message.getName() : null;
      dataClass.number = message.hasNumber() ? message.getNumber() : null;
      callback.onSuccess(dataClass, responseHeaders);
    }
  }

  private static class MockUnsuccessfulResponseHandler implements HttpUnsuccessfulResponseHandler {

    MockTransport transport;
    boolean returnSuccessAuthenticatedContent;

    MockUnsuccessfulResponseHandler(
        MockTransport transport, boolean returnSuccessAuthenticatedContent) {
      this.transport = transport;
      this.returnSuccessAuthenticatedContent = returnSuccessAuthenticatedContent;
    }

    @Override
    public boolean handleResponse(
        HttpRequest request, HttpResponse response, boolean supportsRetry) {
      if (transport.returnErrorAuthenticatedContent) {
        // If transport has already been set to return error content do not handle response.
        return false;
      }
      if (returnSuccessAuthenticatedContent) {
        transport.returnSuccessAuthenticatedContent = true;
      } else {
        transport.returnErrorAuthenticatedContent = true;
      }
      return true;
    }
  }

  private static class MockTransport extends MockHttpTransport {

    final boolean testServerError;
    final boolean testAuthenticationError;
    boolean returnSuccessAuthenticatedContent;
    boolean returnErrorAuthenticatedContent;
    final boolean testRedirect;
    final boolean testBinary;
    final boolean testMissingLength;
    int actualCalls;
    int callsBeforeSuccess;

    MockTransport(
        boolean testServerError,
        boolean testAuthenticationError,
        boolean testRedirect,
        boolean testBinary,
        boolean testMissingLength) {
      this.testServerError = testServerError;
      this.testAuthenticationError = testAuthenticationError;
      this.testRedirect = testRedirect;
      this.testBinary = testBinary;
      this.testMissingLength = testMissingLength;
    }

    @Override
    public LowLevelHttpRequest buildRequest(String name, String url) {
      actualCalls++;
      return new MockLowLevelHttpRequest() {
        @Override
        public LowLevelHttpResponse execute() throws IOException {
          MockLowLevelHttpResponse response = new MockLowLevelHttpResponse();
          response.setStatusCode(200);
          response.addHeader("Content-Type", "multipart/mixed; boundary=" + RESPONSE_BOUNDARY);
          String contentType =
              testBinary ? "application/x-protobuf" : "application/json; charset=UTF-8";
          byte[] content1 =
              testBinary
                  ? MockData.Class1.newBuilder()
                      .setId(TEST_ID)
                      .setKind(TEST_KIND)
                      .build()
                      .toByteArray()
                  : utf8Encode(
                      "{\n \"id\": \""
                          + TEST_ID
                          + "\",\n \"kind\": \""
                          + TEST_KIND.replace("\n", "\\n")
                          + "\"\n}");
          byte[] content2 =
              testBinary
                  ? MockData.Class2.newBuilder()
                      .setName(TEST_NAME)
                      .setNumber(TEST_NUM)
                      .build()
                      .toByteArray()
                  : utf8Encode(
                      "{\"name\": \"" + TEST_NAME + "\", \"number\": \"" + TEST_NUM + "\"}");
          byte[] errorContent =
              testBinary
                  ? ErrorOutput.ErrorBody.newBuilder()
                      .setError(
                          ErrorOutput.ErrorProto.newBuilder()
                              .setCode(ERROR_CODE)
                              .setMessage(ERROR_MSG)
                              .addErrors(
                                  ErrorOutput.IndividualError.newBuilder()
                                      .setDomain(ERROR_DOMAIN)
                                      .setReason(ERROR_REASON)
                                      .setMessage(ERROR_MSG)))
                      .build()
                      .toByteArray()
                  : utf8Encode(
                      "{\"error\": { \"errors\": [{\"domain\": \""
                          + ERROR_DOMAIN
                          + "\","
                          + "\"reason\": \""
                          + ERROR_REASON
                          + "\", \"message\": \""
                          + ERROR_MSG
                          + "\"}],"
                          + "\"code\": "
                          + ERROR_CODE
                          + ", \"message\": \""
                          + ERROR_MSG
                          + "\"}}");
          ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
          Writer responseContent = new OutputStreamWriter(outputStream, "ISO-8859-1");
          if (returnSuccessAuthenticatedContent || (testRedirect && actualCalls > 1)) {
            if (returnSuccessAuthenticatedContent || actualCalls == callsBeforeSuccess) {
              responseContent
                  .append("--" + RESPONSE_BOUNDARY + "\n")
                  .append("Content-Type: application/http\n")
                  .append("Content-Transfer-Encoding: binary\n")
                  .append("Content-ID: response-1\n\n")
                  .append("HTTP/1.1 200 OK\n")
                  .append("Content-Type: " + contentType + "\n");
              if (!testMissingLength) {
                responseContent.append("Content-Length: " + content2.length + "\n");
              }
              responseContent.append("\n");
              responseContent.flush();
              outputStream.write(content2);
              responseContent.append("\n--" + RESPONSE_BOUNDARY + "--\n\n");
            } else {
              responseContent
                  .append("--" + RESPONSE_BOUNDARY + "\n")
                  .append("Content-Type: application/http\n")
                  .append("Content-Transfer-Encoding: binary\n")
                  .append("Content-ID: response-1\n\n")
                  .append("HTTP/1.1 " + ERROR_CODE + " Not Found\n")
                  .append("Content-Type: " + contentType + "\n");
              if (!testMissingLength) {
                responseContent.append("Content-Length: " + errorContent.length + "\n");
              }
              responseContent.append("\n");
              responseContent.flush();
              outputStream.write(errorContent);
              responseContent.append("\n--" + RESPONSE_BOUNDARY + "--\n\n");
            }
          } else if (returnErrorAuthenticatedContent) {
            responseContent
                .append("Content-Type: application/http\n")
                .append("Content-Transfer-Encoding: binary\n")
                .append("Content-ID: response-1\n\n");
            responseContent
                .append("HTTP/1.1 " + ERROR_CODE + " Not Found\n")
                .append("Content-Type: " + contentType + "\n");
            if (!testMissingLength) {
              responseContent.append("Content-Length: " + errorContent.length + "\n");
            }
            responseContent.append("\n");
            responseContent.flush();
            outputStream.write(errorContent);
            responseContent.append("\n--" + RESPONSE_BOUNDARY + "--\n\n");
          } else {
            responseContent
                .append("--" + RESPONSE_BOUNDARY + "\n")
                .append("Content-Type: application/http\n")
                .append("Content-Transfer-Encoding: binary\n")
                .append("Content-ID: response-1\n\n")
                .append("HTTP/1.1 200 OK\n")
                .append("Content-Type: " + contentType + "\n");
            if (!testMissingLength) {
              responseContent.append("Content-Length: " + content1.length + "\n");
            }
            responseContent.append("\n");
            responseContent.flush();
            outputStream.write(content1);
            responseContent
                .append("\n--" + RESPONSE_BOUNDARY + "\n")
                .append("Content-Type: application/http\n")
                .append("Content-Transfer-Encoding: binary\n")
                .append("Content-ID: response-2\n\n");

            if (testServerError) {
              responseContent
                  .append("HTTP/1.1 " + ERROR_CODE + " Not Found\n")
                  .append("Content-Type: " + contentType + "\n");
              if (!testMissingLength) {
                responseContent.append("Content-Length: " + errorContent.length + "\n");
              }
              responseContent.append("\n");
              responseContent.flush();
              outputStream.write(errorContent);
              responseContent.append("\n--" + RESPONSE_BOUNDARY + "--\n\n");
            } else if (testAuthenticationError) {
              responseContent
                  .append("HTTP/1.1 401 Unauthorized\n")
                  .append("Content-Type: application/json; charset=UTF-8\n\n")
                  .append("--" + RESPONSE_BOUNDARY + "--\n\n");
            } else if (testRedirect && actualCalls == 1) {
              responseContent
                  .append("HTTP/1.1 301 MovedPermanently\n")
                  .append("Content-Type: " + contentType + "\n")
                  .append("Location: http://redirect/location\n\n")
                  .append("--" + RESPONSE_BOUNDARY + "--\n\n");
            } else {
              responseContent
                  .append("HTTP/1.1 200 OK\n")
                  .append("Content-Type: " + contentType + "\n");
              if (!testMissingLength) {
                responseContent.append("Content-Length: " + content2.length + "\n");
              }
              responseContent.append("\n");
              responseContent.flush();
              outputStream.write(content2);
              responseContent.append("\n--" + RESPONSE_BOUNDARY + "--\n\n");
            }
          }
          responseContent.flush();
          response.setContent(outputStream.toByteArray());
          return response;
        }

        // Short-hand to encode a String as a UTF-8 byte array
        private byte[] utf8Encode(String string) {
          return Charsets.UTF_8.encode(string).array();
        }
      };
    }
  }

  private static class MockCredential implements HttpRequestInitializer, HttpExecuteInterceptor {

    boolean initializerCalled = false;
    boolean interceptorCalled = false;

    MockCredential() {}

    @Override
    public void initialize(HttpRequest request) {
      request.setInterceptor(this);
      initializerCalled = true;
    }

    @Override
    public void intercept(HttpRequest request) {
      interceptorCalled = true;
    }
  }

  private BatchRequest getBatchPopulatedWithRequests(
      boolean testServerError,
      boolean testAuthenticationError,
      boolean returnSuccessAuthenticatedContent,
      boolean testRedirect,
      boolean testBinary,
      boolean testMissingLength)
      throws IOException {
    transport =
        new MockTransport(
            testServerError, testAuthenticationError, testRedirect, testBinary, testMissingLength);
    MockGoogleClient client =
        new MockGoogleClient.Builder(transport, ROOT_URL, SERVICE_PATH, null, null)
            .setApplicationName("Test Application")
            .build();
    MockGoogleClientRequest<String> jsonHttpRequest1 =
        new MockGoogleClientRequest<String>(client, METHOD1, URI_TEMPLATE1, null, String.class);
    MockGoogleClientRequest<String> jsonHttpRequest2 =
        new MockGoogleClientRequest<String>(client, METHOD2, URI_TEMPLATE2, null, String.class);
    credential = new MockCredential();

    ObjectParser parser =
        testBinary ? new ProtoObjectParser() : new JsonObjectParser(new GsonFactory());
    BatchRequest batchRequest =
        new BatchRequest(transport, credential).setBatchUrl(new GenericUrl(TEST_BATCH_URL));
    HttpRequest request1 = jsonHttpRequest1.buildHttpRequest();
    request1.setParser(parser);
    HttpRequest request2 = jsonHttpRequest2.buildHttpRequest();
    request2.setParser(parser);
    if (testAuthenticationError) {
      request2.setUnsuccessfulResponseHandler(
          new MockUnsuccessfulResponseHandler(transport, returnSuccessAuthenticatedContent));
    }

    if (testBinary) {
      batchRequest.queue(
          request1,
          MockData.Class1.class,
          ErrorOutput.ErrorBody.class,
          new TestCallback1Adapter(callback1));
      batchRequest.queue(
          request2,
          MockData.Class2.class,
          ErrorOutput.ErrorBody.class,
          new TestCallback2Adapter(callback2));
    } else {
      batchRequest.queue(request1, MockDataClass1.class, GoogleJsonErrorContainer.class, callback1);
      batchRequest.queue(request2, MockDataClass2.class, GoogleJsonErrorContainer.class, callback2);
    }
    return batchRequest;
  }

  public void testQueueDatastructures() throws Exception {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, false, false, false, false, false);
    List<RequestInfo<?, ?>> requestInfos = batchRequest.requestInfos;

    // Assert that the expected objects are queued.
    assertEquals(2, requestInfos.size());
    assertEquals(MockDataClass1.class, requestInfos.get(0).dataClass);
    assertEquals(callback1, requestInfos.get(0).callback);
    assertEquals(MockDataClass2.class, requestInfos.get(1).dataClass);
    assertEquals(callback2, requestInfos.get(1).callback);
    // Assert that the requests in the queue are as expected.
    assertEquals(
        ROOT_URL + SERVICE_PATH + URI_TEMPLATE1, requestInfos.get(0).request.getUrl().build());
    assertEquals(
        ROOT_URL + SERVICE_PATH + URI_TEMPLATE2, requestInfos.get(1).request.getUrl().build());
    assertEquals(METHOD1, requestInfos.get(0).request.getRequestMethod());
    assertEquals(METHOD2, requestInfos.get(1).request.getRequestMethod());
  }

  public void testExecute() throws IOException {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, false, false, false, false, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(1, callback2.successCalls);
    assertEquals(0, callback2.failureCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }

  public void testExecuteWithError() throws IOException {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(true, false, false, false, false, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(0, callback2.successCalls);
    assertEquals(1, callback2.failureCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
    // Assert transport called expected number of times.
    assertEquals(1, transport.actualCalls);
  }

  public void testExecuteWithVoidCallback() throws Exception {
    subTestExecuteWithVoidCallback(false);
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(1, callback3.successCalls);
    assertEquals(0, callback3.failureCalls);
  }

  public void testExecuteWithVoidCallbackError() throws Exception {
    subTestExecuteWithVoidCallback(true);
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(0, callback3.successCalls);
    assertEquals(1, callback3.failureCalls);
  }

  public void subTestExecuteWithVoidCallback(boolean testServerError) throws IOException {
    MockTransport transport = new MockTransport(testServerError, false, false, false, false);
    MockGoogleClient client =
        new MockGoogleClient.Builder(transport, ROOT_URL, SERVICE_PATH, null, null)
            .setApplicationName("Test Application")
            .build();
    MockGoogleClientRequest<String> jsonHttpRequest1 =
        new MockGoogleClientRequest<String>(client, METHOD1, URI_TEMPLATE1, null, String.class);
    MockGoogleClientRequest<String> jsonHttpRequest2 =
        new MockGoogleClientRequest<String>(client, METHOD2, URI_TEMPLATE2, null, String.class);
    ObjectParser parser = new JsonObjectParser(new GsonFactory());
    BatchRequest batchRequest =
        new BatchRequest(transport, null).setBatchUrl(new GenericUrl(TEST_BATCH_URL));
    HttpRequest request1 = jsonHttpRequest1.buildHttpRequest();
    request1.setParser(parser);
    HttpRequest request2 = jsonHttpRequest2.buildHttpRequest();
    request2.setParser(parser);
    batchRequest.queue(request1, MockDataClass1.class, GoogleJsonErrorContainer.class, callback1);
    batchRequest.queue(request2, Void.class, Void.class, callback3);
    batchRequest.execute();
    // Assert transport called expected number of times.
    assertEquals(1, transport.actualCalls);
  }

  public void testExecuteWithAuthenticationErrorThenSuccessCallback() throws Exception {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, true, true, false, false, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(1, callback2.successCalls);
    assertEquals(0, callback2.failureCalls);
    // Assert transport called expected number of times.
    assertEquals(2, transport.actualCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }

  public void testExecuteWithAuthenticationErrorThenErrorCallback() throws Exception {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, true, false, false, false, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(0, callback2.successCalls);
    assertEquals(1, callback2.failureCalls);
    // Assert transport called expected number of times.
    assertEquals(2, transport.actualCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }

  public void testInterceptor() throws Exception {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(true, false, false, false, false, false);
    batchRequest.execute();
    // Assert the top-level request initializer is called.
    assertTrue(credential.initializerCalled);
    assertTrue(credential.interceptorCalled);
  }

  public void testRedirect() throws Exception {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, false, false, true, false, false);
    transport.callsBeforeSuccess = 2;
    batchRequest.execute();
    // Assert transport called expected number of times.
    assertEquals(2, transport.actualCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }

  public void testExecute_checkWriteTo() throws Exception {
    String request1Method = HttpMethods.POST;
    String request1Url = "http://test/dummy/url1";
    String request1ContentType = "application/json";
    String request1Content = "{\"data\":{\"foo\":{\"v1\":{}}}}";

    String request2Method = HttpMethods.GET;
    String request2Url = "http://test/dummy/url2";

    // MIME content boundaries are not reproducible.
    StringBuilder part1 = new StringBuilder();
    part1.append("Content-Length: 118\r\n");
    part1.append("Content-Type: application/http\r\n");
    part1.append("content-id: 1\r\n");
    part1.append("content-transfer-encoding: binary\r\n");
    part1.append("\r\n");
    part1.append("POST http://test/dummy/url1 HTTP/1.1\r\n");
    part1.append("Content-Length: 26\r\n");
    part1.append("Content-Type: " + request1ContentType + "\r\n");
    part1.append("\r\n");
    part1.append(request1Content + "\r\n");
    part1.append("--__END_OF_PART__");
    String expected1 = part1.toString();

    StringBuilder part2 = new StringBuilder();
    part2.append("Content-Length: 39\r\n");
    part2.append("Content-Type: application/http\r\n");
    part2.append("content-id: 2\r\n");
    part2.append("content-transfer-encoding: binary\r\n");
    part2.append("\r\n");
    part2.append("GET http://test/dummy/url2 HTTP/1.1\r\n");
    part2.append("\r\n");
    part2.append("\r\n");
    part2.append("--__END_OF_PART__");
    String expected2 = part2.toString();

    MockHttpTransport transport = new MockHttpTransport();
    HttpRequest request1 =
        transport
            .createRequestFactory()
            .buildRequest(
                request1Method,
                new GenericUrl(request1Url),
                new ByteArrayContent(request1ContentType, request1Content.getBytes(UTF_8)));
    HttpRequest request2 =
        transport
            .createRequestFactory()
            .buildRequest(request2Method, new GenericUrl(request2Url), null);
    subtestExecute_checkWriteTo(expected1, expected2, request1, request2);
  }

  private void subtestExecute_checkWriteTo(
      final String part1, final String part2, HttpRequest... requests) throws IOException {

    MockHttpTransport transport =
        new MockHttpTransport() {

          @Override
          public LowLevelHttpRequest buildRequest(String method, String url) {
            return new MockLowLevelHttpRequest(url) {

              @Override
              public LowLevelHttpResponse execute() throws IOException {
                assertTrue(
                    getContentType().startsWith("multipart/mixed; boundary=__END_OF_PART__"));
                ByteArrayOutputStream out = new ByteArrayOutputStream();
                getStreamingContent().writeTo(out);
                String actual = out.toString("UTF-8");
                assertTrue(actual + "\n does not contain \n" + part1, actual.contains(part1));
                assertTrue(actual.contains(part2));
                MockLowLevelHttpResponse response = new MockLowLevelHttpResponse();
                response.setStatusCode(200);
                response.addHeader(
                    "Content-Type", "multipart/mixed; boundary=" + RESPONSE_BOUNDARY);
                String content2 =
                    "{\"name\": \"" + TEST_NAME + "\", \"number\": \"" + TEST_NUM + "\"}";
                StringBuilder responseContent = new StringBuilder();
                responseContent
                    .append("--" + RESPONSE_BOUNDARY + "\n")
                    .append("Content-Type: application/http\n")
                    .append("Content-Transfer-Encoding: binary\n")
                    .append("Content-ID: response-1\n\n")
                    .append("HTTP/1.1 200 OK\n")
                    .append("Content-Type: application/json; charset=UTF-8\n")
                    .append("Content-Length: " + content2.length() + "\n\n")
                    .append(content2 + "\n\n")
                    .append("--" + RESPONSE_BOUNDARY + "--\n\n");
                response.setContent(responseContent.toString());
                return response;
              }
            };
          }
        };

    BatchRequest batchRequest = new BatchRequest(transport, null);
    BatchCallback<Void, Void> callback =
        new BatchCallback<Void, Void>() {

          @Override
          public void onSuccess(Void t, HttpHeaders responseHeaders) {}

          @Override
          public void onFailure(Void e, HttpHeaders responseHeaders) {}
        };
    for (HttpRequest request : requests) {
      batchRequest.queue(request, Void.class, Void.class, callback);
    }
    batchRequest.execute();
  }

  public void testExecute_checkWriteToNoHeaders() throws IOException {
    MockHttpTransport transport = new MockHttpTransport();
    HttpRequest request =
        transport
            .createRequestFactory()
            .buildPostRequest(
                HttpTesting.SIMPLE_GENERIC_URL,
                new HttpContent() {

                  @Override
                  public long getLength() {
                    return -1;
                  }

                  @Override
                  public String getType() {
                    return null;
                  }

                  @Override
                  public void writeTo(OutputStream out) {}

                  @Override
                  public boolean retrySupported() {
                    return true;
                  }
                });
    String expected =
        new StringBuilder()
            .append("Content-Length: 36\r\n")
            .append("Content-Type: application/http\r\n")
            .append("content-id: 1\r\n")
            .append("content-transfer-encoding: binary\r\n")
            .append("\r\n")
            .append("POST http://google.com/ HTTP/1.1\r\n")
            .append("\r\n")
            .append("\r\n")
            .append("--__END_OF_PART__")
            .toString();
    subtestExecute_checkWriteTo(expected, expected, request);
  }

  public void testProtoExecute() throws IOException {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, false, false, false, true, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(1, callback2.successCalls);
    assertEquals(0, callback2.failureCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }

  public void testProtoExecuteWithError() throws IOException {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(true, false, false, false, true, false);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(0, callback2.successCalls);
    assertEquals(1, callback2.failureCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
    // Assert transport called expected number of times.
    assertEquals(1, transport.actualCalls);
  }

  public void testProtoExecuteWithoutLength() throws IOException {
    BatchRequest batchRequest =
        getBatchPopulatedWithRequests(false, false, false, false, true, true);
    batchRequest.execute();
    // Assert callbacks have been invoked.
    assertEquals(1, callback1.successCalls);
    assertEquals(1, callback2.successCalls);
    assertEquals(0, callback2.failureCalls);
    // Assert requestInfos is empty after execute.
    assertTrue(batchRequest.requestInfos.isEmpty());
  }
}