use of com.palantir.dialogue.Response in project dialogue by palantir.
the class RetryingChannelTest method doesNotRetryRuntimeException.
@Test
public void doesNotRetryRuntimeException() {
when(channel.execute(any())).thenReturn(Futures.immediateFailedFuture(new SafeRuntimeException("bug"))).thenReturn(SUCCESS);
EndpointChannel retryer = new RetryingChannel(channel, TestEndpoint.POST, "my-channel", 1, Duration.ZERO, ClientConfiguration.ServerQoS.AUTOMATIC_RETRY, ClientConfiguration.RetryOnTimeout.DISABLED);
ListenableFuture<Response> response = retryer.execute(REQUEST);
assertThatThrownBy(response::get).hasRootCauseExactlyInstanceOf(SafeRuntimeException.class).hasRootCauseMessage("bug");
}
use of com.palantir.dialogue.Response in project dialogue by palantir.
the class RetryingChannelTest method retries_308s_when_429_and_503_are_propagated.
@Test
public void retries_308s_when_429_and_503_are_propagated() throws Exception {
Response mockResponse = mock(Response.class);
when(mockResponse.code()).thenReturn(308);
when(mockResponse.getFirstHeader(eq("Location"))).thenReturn(Optional.of("https://localhost"));
when(channel.execute(any())).thenReturn(Futures.immediateFuture(mockResponse));
EndpointChannel retryer = new RetryingChannel(channel, TestEndpoint.POST, "my-channel", 3, Duration.ZERO, // This does not apply to 308 responses
ClientConfiguration.ServerQoS.PROPAGATE_429_and_503_TO_CALLER, ClientConfiguration.RetryOnTimeout.DISABLED);
ListenableFuture<Response> response = retryer.execute(REQUEST);
assertThat(response).isDone();
assertThat(response.get()).as("After retries are exhausted the 308 response should be returned").isSameAs(mockResponse);
verify(channel, times(4)).execute(REQUEST);
}
use of com.palantir.dialogue.Response in project dialogue by palantir.
the class RetryingChannelTest method doesnt_retry_500s_for_post.
@Test
public void doesnt_retry_500s_for_post() throws Exception {
when(channel.execute(any())).thenReturn(Futures.immediateFuture(new TestResponse().code(500)));
EndpointChannel retryer = new RetryingChannel(channel, TestEndpoint.POST, "my-channel", 3, Duration.ZERO, ClientConfiguration.ServerQoS.AUTOMATIC_RETRY, ClientConfiguration.RetryOnTimeout.DISABLED);
ListenableFuture<Response> response = retryer.execute(REQUEST);
assertThat(response).isDone();
assertThat(response.get().code()).isEqualTo(500);
verify(channel, times(1)).execute(REQUEST);
}
use of com.palantir.dialogue.Response in project dialogue by palantir.
the class RetryingChannelTest method testNoFailures.
@Test
public void testNoFailures() throws ExecutionException, InterruptedException {
when(channel.execute(any())).thenReturn(SUCCESS);
EndpointChannel retryer = new RetryingChannel(channel, TestEndpoint.POST, "my-channel", 3, Duration.ZERO, ClientConfiguration.ServerQoS.AUTOMATIC_RETRY, ClientConfiguration.RetryOnTimeout.DISABLED);
ListenableFuture<Response> response = retryer.execute(REQUEST);
assertThat(response.get()).isEqualTo(EXPECTED_RESPONSE);
}
use of com.palantir.dialogue.Response in project dialogue by palantir.
the class RetryingChannelTest method retries_500s_for_delete.
@Test
public void retries_500s_for_delete() throws Exception {
when(channel.execute(any())).thenReturn(Futures.immediateFuture(new TestResponse().code(500))).thenReturn(Futures.immediateFuture(new TestResponse().code(200)));
EndpointChannel retryer = new RetryingChannel(channel, TestEndpoint.DELETE, "my-channel", 3, Duration.ZERO, ClientConfiguration.ServerQoS.AUTOMATIC_RETRY, ClientConfiguration.RetryOnTimeout.DISABLED);
ListenableFuture<Response> response = retryer.execute(REQUEST);
assertThat(response).isDone();
assertThat(response.get().code()).isEqualTo(200);
verify(channel, times(2)).execute(REQUEST);
}
Aggregations