Пример #1
0
  @Test
  public void testRetryFileSize() throws IOException {
    GcsOptions pipelineOptions = gcsOptionsWithTestCredential();
    GcsUtil gcsUtil = pipelineOptions.getGcsUtil();

    Storage mockStorage = Mockito.mock(Storage.class);
    gcsUtil.setStorageClient(mockStorage);

    Storage.Objects mockStorageObjects = Mockito.mock(Storage.Objects.class);
    Storage.Objects.Get mockStorageGet = Mockito.mock(Storage.Objects.Get.class);

    BackOff mockBackOff = new AttemptBoundedExponentialBackOff(3, 200);

    when(mockStorage.objects()).thenReturn(mockStorageObjects);
    when(mockStorageObjects.get("testbucket", "testobject")).thenReturn(mockStorageGet);
    when(mockStorageGet.execute())
        .thenThrow(new SocketTimeoutException("SocketException"))
        .thenThrow(new SocketTimeoutException("SocketException"))
        .thenReturn(new StorageObject().setSize(BigInteger.valueOf(1000)));

    assertEquals(
        1000,
        gcsUtil.fileSize(
            GcsPath.fromComponents("testbucket", "testobject"),
            mockBackOff,
            new FastNanoClockAndSleeper()));
    assertEquals(mockBackOff.nextBackOffMillis(), BackOff.STOP);
  }
 /**
  * Sleep and/or determine if the backoff has timed out.
  *
  * @param backoff
  * @return RetryState indicating the current state of the retry logic.
  * @throws IOException in some cases from {@link BackOff#nextBackOffMillis()}
  */
 protected RetryState getRetryState(BackOff backoff) throws IOException {
   long nextBackOffMillis = backoff.nextBackOffMillis();
   if (nextBackOffMillis == BackOff.STOP) {
     logger.warn(
         "Exhausted the number of retries for credentials refresh after "
             + this.retryOptions.getMaxElaspedBackoffMillis()
             + " milliseconds.");
     return RetryState.RetriesExhausted;
   }
   try {
     sleeper.sleep(nextBackOffMillis);
     // Try to perform another call.
     return RetryState.PerformRetry;
   } catch (InterruptedException e) {
     logger.warn("Interrupted while trying to refresh credentials.");
     Thread.interrupted();
     // If the thread is interrupted, terminate immediately.
     return RetryState.Interrupted;
   }
 }