use of software.amazon.awssdk.metrics.MetricPublisher in project aws-sdk-java-v2 by aws.
the class AsyncClientMetricPublisherResolutionTest method testApiCall_publishersSetOnRequest_requestPublishersInvoked.
@Test
public void testApiCall_publishersSetOnRequest_requestPublishersInvoked() throws IOException {
MetricPublisher publisher1 = mock(MetricPublisher.class);
MetricPublisher publisher2 = mock(MetricPublisher.class);
client = clientWithPublishers();
try {
client.allTypes(r -> r.overrideConfiguration(o -> o.addMetricPublisher(publisher1).addMetricPublisher(publisher2))).join();
} catch (Throwable t) {
// ignored, call fails because our mock HTTP client isn't set up
} finally {
verify(publisher1).publish(any(MetricCollection.class));
verify(publisher2).publish(any(MetricCollection.class));
}
}
use of software.amazon.awssdk.metrics.MetricPublisher in project aws-sdk-java-v2 by aws.
the class AsyncCoreMetricsTest method testPaginatingApiCall_publisherOverriddenOnRequest_requestPublisherTakesPrecedence.
@Test
public void testPaginatingApiCall_publisherOverriddenOnRequest_requestPublisherTakesPrecedence() throws Exception {
stubSuccessfulResponse();
MetricPublisher requestMetricPublisher = mock(MetricPublisher.class);
PaginatedOperationWithResultKeyPublisher paginatedPublisher = client.paginatedOperationWithResultKeyPaginator(r -> r.overrideConfiguration(o -> o.addMetricPublisher(requestMetricPublisher)));
CompletableFuture<Void> future = paginatedPublisher.subscribe(PaginatedOperationWithResultKeyResponse::items);
future.get();
verify(requestMetricPublisher).publish(any(MetricCollection.class));
verifyNoMoreInteractions(mockPublisher);
}
use of software.amazon.awssdk.metrics.MetricPublisher 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);
}
use of software.amazon.awssdk.metrics.MetricPublisher 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));
}
use of software.amazon.awssdk.metrics.MetricPublisher in project aws-sdk-java-v2 by aws.
the class RequestOverrideConfigurationTest method addMetricPublisher_listPreviouslyAdded_appendedToList.
@Test
public void addMetricPublisher_listPreviouslyAdded_appendedToList() {
List<MetricPublisher> publishers = new ArrayList<>();
publishers.add(mock(MetricPublisher.class));
publishers.add(mock(MetricPublisher.class));
MetricPublisher thirdAdded = mock(MetricPublisher.class);
SdkRequestOverrideConfiguration.Builder builder = SdkRequestOverrideConfiguration.builder();
builder.metricPublishers(publishers);
builder.addMetricPublisher(thirdAdded);
SdkRequestOverrideConfiguration overrideConfig = builder.build();
assertThat(overrideConfig.metricPublishers()).containsExactly(publishers.get(0), publishers.get(1), thirdAdded);
}
Aggregations