Examples with ExecutionInterceptor software.amazon.awssdk.core.interceptor.ExecutionInterceptor used on opensource projects

Example 21 with ExecutionInterceptor

use of software.amazon.awssdk.core.interceptor.ExecutionInterceptor in project aws-sdk-java-v2 by aws.

the class ExecutionInterceptorTest method sync_serviceException_failureInterceptorMethodsCalled.

@Test
public void sync_serviceException_failureInterceptorMethodsCalled() {
    // Given
    ExecutionInterceptor interceptor = mock(MessageUpdatingInterceptor.class, CALLS_REAL_METHODS);
    ProtocolRestJsonClient client = client(interceptor);
    MembersInHeadersRequest request = MembersInHeadersRequest.builder().build();
    // When
    assertThatExceptionOfType(SdkServiceException.class).isThrownBy(() -> client.membersInHeaders(request));
    // Expect
    expectServiceCallErrorMethodsCalled(interceptor, false);
}

Example 22 with ExecutionInterceptor

use of software.amazon.awssdk.core.interceptor.ExecutionInterceptor in project aws-sdk-java-v2 by aws.

the class ExecutionInterceptorTest method async_streamingOutput_success_allInterceptorMethodsCalled.

@Test
public void async_streamingOutput_success_allInterceptorMethodsCalled() throws Exception {
    // Given
    ExecutionInterceptor interceptor = mock(NoOpInterceptor.class, CALLS_REAL_METHODS);
    ProtocolRestJsonAsyncClient client = asyncClient(interceptor);
    StreamingOutputOperationRequest request = StreamingOutputOperationRequest.builder().build();
    stubFor(post(urlPathEqualTo(STREAMING_OUTPUT_PATH)).willReturn(aResponse().withStatus(200).withBody("\0")));
    // When
    client.streamingOutputOperation(request, new NoOpAsyncResponseTransformer()).get(10, TimeUnit.SECONDS);
    // Expect
    Context.AfterTransmission afterTransmissionArg = captureAfterTransmissionArgAsync(interceptor);
    assertThat(afterTransmissionArg.responseBody()).isNotNull();
}

Example 23 with ExecutionInterceptor

use of software.amazon.awssdk.core.interceptor.ExecutionInterceptor in project aws-sdk-java-v2 by aws.

the class DefaultClientBuilderTest method buildIncludesClientOverrides.

@Test
public void buildIncludesClientOverrides() {
    List<ExecutionInterceptor> interceptors = new ArrayList<>();
    ExecutionInterceptor interceptor = new ExecutionInterceptor() {
    };
    interceptors.add(interceptor);
    RetryPolicy retryPolicy = RetryPolicy.builder().build();
    Map<String, List<String>> headers = new HashMap<>();
    List<String> headerValues = new ArrayList<>();
    headerValues.add("value");
    headers.put("client-override-test", headerValues);
    List<MetricPublisher> metricPublishers = new ArrayList<>();
    MetricPublisher metricPublisher = new MetricPublisher() {

        @Override
        public void publish(MetricCollection metricCollection) {
        }

        @Override
        public void close() {
        }
    };
    metricPublishers.add(metricPublisher);
    ExecutionAttribute<String> execAttribute = new ExecutionAttribute<>("test");
    ExecutionAttributes executionAttributes = ExecutionAttributes.builder().put(execAttribute, "value").build();
    Signer signer = (request, execAttributes) -> request;
    String suffix = "suffix";
    String prefix = "prefix";
    Duration apiCallTimeout = Duration.ofMillis(42);
    Duration apiCallAttemptTimeout = Duration.ofMillis(43);
    ProfileFile profileFile = ProfileFile.builder().content(new StringInputStream("")).type(ProfileFile.Type.CONFIGURATION).build();
    String profileName = "name";
    ClientOverrideConfiguration overrideConfig = ClientOverrideConfiguration.builder().executionInterceptors(interceptors).retryPolicy(retryPolicy).headers(headers).putAdvancedOption(SIGNER, signer).putAdvancedOption(USER_AGENT_SUFFIX, suffix).putAdvancedOption(USER_AGENT_PREFIX, prefix).apiCallTimeout(apiCallTimeout).apiCallAttemptTimeout(apiCallAttemptTimeout).putAdvancedOption(DISABLE_HOST_PREFIX_INJECTION, Boolean.TRUE).defaultProfileFile(profileFile).defaultProfileName(profileName).metricPublishers(metricPublishers).executionAttributes(executionAttributes).putAdvancedOption(ENDPOINT_OVERRIDDEN_OVERRIDE, Boolean.TRUE).build();
    SdkClientConfiguration config = testClientBuilder().overrideConfiguration(overrideConfig).build().clientConfiguration;
    assertThat(config.option(EXECUTION_INTERCEPTORS)).contains(interceptor);
    assertThat(config.option(RETRY_POLICY)).isEqualTo(retryPolicy);
    assertThat(config.option(ADDITIONAL_HTTP_HEADERS).get("client-override-test")).isEqualTo(headerValues);
    assertThat(config.option(SIGNER)).isEqualTo(signer);
    assertThat(config.option(USER_AGENT_SUFFIX)).isEqualTo(suffix);
    assertThat(config.option(USER_AGENT_PREFIX)).isEqualTo(prefix);
    assertThat(config.option(API_CALL_TIMEOUT)).isEqualTo(apiCallTimeout);
    assertThat(config.option(API_CALL_ATTEMPT_TIMEOUT)).isEqualTo(apiCallAttemptTimeout);
    assertThat(config.option(DISABLE_HOST_PREFIX_INJECTION)).isEqualTo(Boolean.TRUE);
    assertThat(config.option(PROFILE_FILE)).isEqualTo(profileFile);
    assertThat(config.option(PROFILE_NAME)).isEqualTo(profileName);
    assertThat(config.option(METRIC_PUBLISHERS)).contains(metricPublisher);
    assertThat(config.option(EXECUTION_ATTRIBUTES).getAttribute(execAttribute)).isEqualTo("value");
    assertThat(config.option(ENDPOINT_OVERRIDDEN)).isEqualTo(Boolean.TRUE);
}

Example 24 with ExecutionInterceptor

use of software.amazon.awssdk.core.interceptor.ExecutionInterceptor in project aws-sdk-java-v2 by aws.

the class DefaultClientBuilderTest method overrideConfigurationReturnsSetValues.

@Test
public void overrideConfigurationReturnsSetValues() {
    List<ExecutionInterceptor> interceptors = new ArrayList<>();
    RetryPolicy retryPolicy = RetryPolicy.builder().build();
    Map<String, List<String>> headers = new HashMap<>();
    List<MetricPublisher> metricPublishers = new ArrayList<>();
    ExecutionAttributes executionAttributes = new ExecutionAttributes();
    Signer signer = (request, execAttributes) -> request;
    String suffix = "suffix";
    String prefix = "prefix";
    Duration apiCallTimeout = Duration.ofMillis(42);
    Duration apiCallAttemptTimeout = Duration.ofMillis(43);
    ProfileFile profileFile = ProfileFile.builder().content(new StringInputStream("")).type(ProfileFile.Type.CONFIGURATION).build();
    String profileName = "name";
    ClientOverrideConfiguration overrideConfig = ClientOverrideConfiguration.builder().executionInterceptors(interceptors).retryPolicy(retryPolicy).headers(headers).putAdvancedOption(SIGNER, signer).putAdvancedOption(USER_AGENT_SUFFIX, suffix).putAdvancedOption(USER_AGENT_PREFIX, prefix).apiCallTimeout(apiCallTimeout).apiCallAttemptTimeout(apiCallAttemptTimeout).putAdvancedOption(DISABLE_HOST_PREFIX_INJECTION, Boolean.TRUE).defaultProfileFile(profileFile).defaultProfileName(profileName).metricPublishers(metricPublishers).executionAttributes(executionAttributes).putAdvancedOption(ENDPOINT_OVERRIDDEN_OVERRIDE, Boolean.TRUE).build();
    TestClientBuilder builder = testClientBuilder().overrideConfiguration(overrideConfig);
    ClientOverrideConfiguration builderOverrideConfig = builder.overrideConfiguration();
    assertThat(builderOverrideConfig.executionInterceptors()).isEqualTo(interceptors);
    assertThat(builderOverrideConfig.retryPolicy()).isEqualTo(Optional.of(retryPolicy));
    assertThat(builderOverrideConfig.headers()).isEqualTo(headers);
    assertThat(builderOverrideConfig.advancedOption(SIGNER)).isEqualTo(Optional.of(signer));
    assertThat(builderOverrideConfig.advancedOption(USER_AGENT_SUFFIX)).isEqualTo(Optional.of(suffix));
    assertThat(builderOverrideConfig.apiCallTimeout()).isEqualTo(Optional.of(apiCallTimeout));
    assertThat(builderOverrideConfig.apiCallAttemptTimeout()).isEqualTo(Optional.of(apiCallAttemptTimeout));
    assertThat(builderOverrideConfig.advancedOption(DISABLE_HOST_PREFIX_INJECTION)).isEqualTo(Optional.of(Boolean.TRUE));
    assertThat(builderOverrideConfig.defaultProfileFile()).isEqualTo(Optional.of(profileFile));
    assertThat(builderOverrideConfig.defaultProfileName()).isEqualTo(Optional.of(profileName));
    assertThat(builderOverrideConfig.metricPublishers()).isEqualTo(metricPublishers);
    assertThat(builderOverrideConfig.executionAttributes().getAttributes()).isEqualTo(executionAttributes.getAttributes());
    assertThat(builderOverrideConfig.advancedOption(ENDPOINT_OVERRIDDEN_OVERRIDE)).isEqualTo(Optional.of(Boolean.TRUE));
}

Example 25 with ExecutionInterceptor

use of software.amazon.awssdk.core.interceptor.ExecutionInterceptor in project aws-sdk-java-v2 by aws.

the class HttpClientApiCallTimeoutTest method successfulResponse_SlowAfterResponseExecutionInterceptor_ThrowsApiCallTimeoutException.

@Test
public void successfulResponse_SlowAfterResponseExecutionInterceptor_ThrowsApiCallTimeoutException() {
    stubFor(get(anyUrl()).willReturn(aResponse().withStatus(200).withBody("{}")));
    ExecutionInterceptor interceptor = new SlowExecutionInterceptor().afterTransmissionWaitInSeconds(SLOW_REQUEST_HANDLER_TIMEOUT);
    assertThatThrownBy(() -> requestBuilder().executionContext(withInterceptors(interceptor)).execute(noOpSyncResponseHandler())).isInstanceOf(ApiCallTimeoutException.class);
}