import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
/**
 * Builds a HttpResponse with the given string response.
 *
 * @param header header value to provide or null if none.
 * @param uploadId upload id to provide in the url upload id param or null if none.
 * @param uploadType upload type to provide in url upload type param or null if none.
 * @return HttpResponse with the given parameters
 * @throws IOException
 */
private HttpResponse buildHttpResponse(String header, String uploadId, String uploadType)
    throws IOException {
  MockHttpTransport.Builder builder = new MockHttpTransport.Builder();
  MockLowLevelHttpResponse resp = new MockLowLevelHttpResponse();
  builder.setLowLevelHttpResponse(resp);
  resp.setStatusCode(200);
  GenericUrl url = new GenericUrl(HttpTesting.SIMPLE_URL);
  if (header != null) {
    resp.addHeader("X-GUploader-UploadID", header);
  }
  if (uploadId != null) {
    url.put("upload_id", uploadId);
  }
  if (uploadType != null) {
    url.put("uploadType", uploadType);
  }
  return builder.build().createRequestFactory().buildGetRequest(url).execute();
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
/** Returns a valid GoogleJsonResponseException for the given status code and error message.  */
private GoogleJsonResponseException makeResponseException(
    final int statusCode,
    final String message) throws Exception {
  HttpTransport transport = new MockHttpTransport() {
    @Override
    public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
      return new MockLowLevelHttpRequest() {
        @Override
        public LowLevelHttpResponse execute() throws IOException {
          MockLowLevelHttpResponse response = new MockLowLevelHttpResponse();
          response.setStatusCode(statusCode);
          response.setContentType(Json.MEDIA_TYPE);
          response.setContent(String.format(
              "{\"error\":{\"code\":%d,\"message\":\"%s\",\"domain\":\"global\","
              + "\"reason\":\"duplicate\"}}",
              statusCode,
              message));
          return response;
        }};
    }};
  HttpRequest request = transport.createRequestFactory()
      .buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL)
      .setThrowExceptionOnExecuteError(false);
  return GoogleJsonResponseException.from(new JacksonFactory(), request.execute());
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testHandleResponse() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(true);
		labelResponse.setMeta(meta);
		Boolean retry = responseHandler.handleResponse(req, labelResponse, true);
		assertTrue(retry);
		assertEquals(new Long(1000), sleeper.getDelay());
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testExponentialDelay() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(true);
		labelResponse.setMeta(meta);
		Boolean retry = null;
		Long delay = new Long(1000);
		for(int i = 0; i < 5; i++) {
			retry = responseHandler.handleResponse(req, labelResponse, true);
			assertEquals(delay, sleeper.getDelay());
			delay *= 2;
		}
		assertTrue(retry);
		
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testRetryStopOnSixth() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(true);
		labelResponse.setMeta(meta);
		Boolean retry = null;
		for(int i = 0; i < 6; i++) {
			retry = responseHandler.handleResponse(req, labelResponse, true);
		}
		assertFalse(retry);
		
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testNotRetryable() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(false);
		labelResponse.setMeta(meta);
		Boolean retry = null;
		retry = responseHandler.handleResponse(req, labelResponse, true);
		assertFalse(retry);
		
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testRetryableNullInResponse() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(null);
		labelResponse.setMeta(meta);
		Boolean retry = null;
		retry = responseHandler.handleResponse(req, labelResponse, true);
		assertFalse(retry);
		
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testDoNotRetry() {
	HttpRequest req;
	try {
		req = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		req.setNumberOfRetries(5);
		LabelResponse labelResponse = new LabelResponse();
		Meta meta = new Meta();
		meta.setCode(200);
		meta.setRetryable(true);
		labelResponse.setMeta(meta);
		Boolean retry = null;
		retry = responseHandler.handleResponse(req, labelResponse, false);
		assertFalse(retry);
		
		//responseHandler.handleResponse(req, labelResponse, true);
	} catch (IOException e) {
	}	
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testPostIOError() throws InterruptedException, ExecutionException, TimeoutException {

	Jups jups = new Jups(HttpTesting.SIMPLE_URL, "APP_ID", "SECRET");
    jups.mHttpTransport = new MockHttpTransport() {
        @Override
        public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
            return new MockLowLevelHttpRequest() {
                @Override
                public LowLevelHttpResponse execute() throws IOException {
                    throw new IOException("test");
                }
            };
        }
    };

    String result = jups.send(new PushNotification("MASTER_VARIANT_ID", "VARIANT_ID", "ALERT_MESSAGE", "SOUND"), "TEST_UUID");
    assertEquals("Push notification should have failed", result, "failed");
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testSuccessResult() throws InterruptedException, ExecutionException, TimeoutException {

    Afm afm = Afm.get(HttpTesting.SIMPLE_URL);
    JsonMockHttpTransport mockTransport = new JsonMockHttpTransport(succeeded_task_json);
    afm.mHttpTransport = mockTransport;

    FutureCallback result = new FutureCallback();
    afm.postTask(new Task("TASK_UUID", "TEST_UUID", "wait", 1), result);

    Task resultTask = result.task.get(5000, TimeUnit.MILLISECONDS);
    assertNotNull("Task should have been returned", resultTask);
    assertTrue("Task should have succeeded", result.succeeded);
    assertNull("Result should not contain error.", resultTask.task_error);
    assertEquals("Afm should have only made 2 requests.", 2, mockTransport.getRequestsCount());
    checkTaskBasics(resultTask);
    assertNotNull("Task should have result data", resultTask.task_result);
    assertEquals("result value should match", new BigDecimal(2), resultTask.task_result.get("waited"));
    assertEquals("int result value should have zero", 0, ((BigDecimal) resultTask.task_result.get("waited")).scale());
    assertEquals("result value should match", new BigDecimal(2.5), resultTask.task_result.get("waited-double"));
    assertNotEquals("double return value should not have zero scale", 0, ((BigDecimal) resultTask.task_result.get("waited-double")).scale());
    assertEquals("result value should match", "string-value", resultTask.task_result.get("string-value"));
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testFailedResult() throws InterruptedException, ExecutionException, TimeoutException {

    Afm afm = Afm.get(HttpTesting.SIMPLE_URL);
    JsonMockHttpTransport mockTransport = new JsonMockHttpTransport(failed_task_json);
    afm.mHttpTransport = mockTransport;

    FutureCallback result = new FutureCallback();
    afm.postTask(new Task("TASK_UUID", "TEST_UUID", "wait", 1), result);

    Task resultTask = result.task.get(5000, TimeUnit.MILLISECONDS);
    assertFalse("Task should have failed", result.succeeded);
    assertNotNull("Task should have been returned.", resultTask);
    assertNotNull("Error should be set.", resultTask.task_error);
    assertEquals("Error message should match definition.", resultTask.task_error, "error description");
    assertEquals("Afm should have only made 2 requests.", 2, mockTransport.getRequestsCount());
    checkTaskBasics(resultTask);
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test(expected = TimeoutException.class)
public void testResultTimeout() throws InterruptedException, ExecutionException, TimeoutException {

    Afm afm = Afm.get(HttpTesting.SIMPLE_URL);
    JsonMockHttpTransport mockTransport = new JsonMockHttpTransport(waiting_task_json);
    afm.mHttpTransport = mockTransport;

    FutureCallback result = new FutureCallback();
    afm.postTask(new Task("TASK_UUID", "TEST_UUID", "wait", 1), result);

    try {
        result.task.get(5000, TimeUnit.MILLISECONDS);
    } catch (TimeoutException e) {
        assertTrue("Afm should have made multiple requests to poll results.", mockTransport.getRequestsCount() > 2);
        throw e;
    }
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testPostIOError() throws InterruptedException, ExecutionException, TimeoutException {

    Afm afm = Afm.get(HttpTesting.SIMPLE_URL);
    afm.mHttpTransport = new MockHttpTransport() {
        @Override
        public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
            return new MockLowLevelHttpRequest() {
                @Override
                public LowLevelHttpResponse execute() throws IOException {
                    throw new IOException("test");
                }
            };
        }
    };

    FutureCallback result = new FutureCallback();
    afm.postTask(new Task("TASK_UUID", "TEST_UUID", "wait", 1), result);

    Task resultTask = result.task.get(5000, TimeUnit.MILLISECONDS);
    assertFalse("Task should have failed", result.succeeded);
    assertNotNull("Task should have been returned.", resultTask);
    assertNotNull("Error should be set.", resultTask.task_error);
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
HttpResponseException makeHttpException(int status) throws IOException {
  MockHttpTransport.Builder builder = new MockHttpTransport.Builder();
  MockLowLevelHttpResponse resp = new MockLowLevelHttpResponse();
  resp.setStatusCode(status);
  builder.setLowLevelHttpResponse(resp);
  try {
    HttpResponse res =
        builder.build()
            .createRequestFactory()
            .buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL)
            .execute();
    return new HttpResponseException(res);
  } catch (HttpResponseException exception) {
    return exception; // Throws the exception we want anyway, so just return it.
  }
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
private HttpResponse createFakeResponse(final String responseHeader, final String responseValue,
                                        final InputStream content) throws IOException {
  HttpTransport transport = new MockHttpTransport() {
    @Override
    public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
      MockLowLevelHttpRequest req = new MockLowLevelHttpRequest() {
        @Override
        public LowLevelHttpResponse execute() throws IOException {
          return new MockLowLevelHttpResponse()
              .addHeader(responseHeader, responseValue)
              .setContent(content);
        }
      };
      return req;
    }
  };
  HttpRequest request =
      transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
  return request.execute();
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
/**
 * Check that a response logs with the correct log.
 */
@Test
public void testResponseLogs() throws IOException {
  new UploadIdResponseInterceptor().interceptResponse(buildHttpResponse("abc", null, "type"));
  GenericUrl url = new GenericUrl(HttpTesting.SIMPLE_URL);
  url.put("uploadType", "type");
  String worker = System.getProperty("worker_id");
  expectedLogs.verifyDebug("Upload ID for url " + url + " on worker " + worker + " is abc");
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
/**
 * Builds a fake GoogleJsonResponseException for testing API error handling.
 */
private static GoogleJsonResponseException googleJsonResponseException(
    final int status, final String reason, final String message) throws IOException {
  final JsonFactory jsonFactory = new JacksonFactory();
  HttpTransport transport = new MockHttpTransport() {
    @Override
    public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
      ErrorInfo errorInfo = new ErrorInfo();
      errorInfo.setReason(reason);
      errorInfo.setMessage(message);
      errorInfo.setFactory(jsonFactory);
      GenericJson error = new GenericJson();
      error.set("code", status);
      error.set("errors", Arrays.asList(errorInfo));
      error.setFactory(jsonFactory);
      GenericJson errorResponse = new GenericJson();
      errorResponse.set("error", error);
      errorResponse.setFactory(jsonFactory);
      return new MockLowLevelHttpRequest().setResponse(
          new MockLowLevelHttpResponse().setContent(errorResponse.toPrettyString())
          .setContentType(Json.MEDIA_TYPE).setStatusCode(status));
      }
  };
  HttpRequest request =
      transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
  request.setThrowExceptionOnExecuteError(false);
  HttpResponse response = request.execute();
  return GoogleJsonResponseException.from(jsonFactory, response);
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testInterceptWhenRateLimitIsZero() {
	try {
		HttpRequest request = transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
		long time = Calendar.getInstance().getTimeInMillis() + 5000;
		rateLimit.setResetTime(time);
		rateLimit.setRateCount(-1);
		interceptor.intercept(request);
		Double delay = sleeper.getDelay().doubleValue();
		delay = Math.ceil(delay/10) * 10;
		assertTrue(delay.intValue() > 4000 && delay.intValue() <= 5000);
	} catch (IOException e) {
		
	}
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
private static GoogleJsonResponseException googleJsonResponseException(
    final int status, final ErrorInfo errorInfo, final String httpStatusString)
    throws IOException {
  final JsonFactory jsonFactory = new JacksonFactory();
  HttpTransport transport = new MockHttpTransport() {
    @Override
    public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
      errorInfo.setFactory(jsonFactory);
      GoogleJsonError jsonError = new GoogleJsonError();
      jsonError.setCode(status);
      jsonError.setErrors(Arrays.asList(errorInfo));
      jsonError.setMessage(httpStatusString);
      jsonError.setFactory(jsonFactory);
      GenericJson errorResponse = new GenericJson();
      errorResponse.set("error", jsonError);
      errorResponse.setFactory(jsonFactory);
      return new MockLowLevelHttpRequest().setResponse(
          new MockLowLevelHttpResponse().setContent(errorResponse.toPrettyString())
          .setContentType(Json.MEDIA_TYPE).setStatusCode(status));
      }
  };
  HttpRequest request =
      transport.createRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
  request.setThrowExceptionOnExecuteError(false);
  HttpResponse response = request.execute();
  return GoogleJsonResponseException.from(jsonFactory, response);
}
 

import com.google.api.client.testing.http.HttpTesting; //导入依赖的package包/类
@Test
public void testBackendErrorRetries() throws Exception {

  HttpTransport transport = new MockHttpTransport() {
    @Override
    public LowLevelHttpRequest buildRequest(String method, String url) throws IOException {
      return new MockLowLevelHttpRequest() {
        @Override
        public LowLevelHttpResponse execute() throws IOException {
          MockLowLevelHttpResponse response = new MockLowLevelHttpResponse();
          response.setStatusCode(500);
          return response;
        }
      };
    }
  };

  GenomicsFactory genomicsFactory =
      GenomicsFactory.builder("test_client").setHttpTransport(transport).build();
  Genomics genomics = genomicsFactory.fromApiKey("xyz");

  HttpRequest request =
      genomics.getRequestFactory().buildGetRequest(HttpTesting.SIMPLE_GENERIC_URL);
  try {
    request.execute();
    fail("this request should not have succeeded");
  } catch (HttpResponseException e) {
  }

  assertEquals(1, genomicsFactory.initializedRequestsCount());
  assertEquals(6, genomicsFactory.unsuccessfulResponsesCount());
  assertEquals(0, genomicsFactory.ioExceptionsCount());
}