Skip to content

Commit 86346ce

Browse files
committed
Inject STS endpoint to KPL in the KplKclIntegrationTests
* Disable metrics for KCL in the `KplKclIntegrationTests`
1 parent f298324 commit 86346ce

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

spring-cloud-aws-kinesis/src/test/java/io/awspring/cloud/kinesis/integration/KplKclIntegrationTests.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
import software.amazon.awssdk.services.kinesis.KinesisAsyncClient;
4848
import software.amazon.kinesis.common.InitialPositionInStream;
4949
import software.amazon.kinesis.common.InitialPositionInStreamExtended;
50+
import software.amazon.kinesis.metrics.MetricsLevel;
5051
import software.amazon.kinesis.producer.KinesisProducer;
5152
import software.amazon.kinesis.producer.KinesisProducerConfiguration;
5253

@@ -136,12 +137,15 @@ public KinesisProducerConfiguration kinesisProducerConfiguration() {
136137
.getEndpointOverride(LocalStackContainer.Service.KINESIS);
137138
URI cloudWatchUri = LocalstackContainerTest.LOCAL_STACK_CONTAINER
138139
.getEndpointOverride(LocalStackContainer.Service.CLOUDWATCH);
140+
URI stsUri = LocalstackContainerTest.LOCAL_STACK_CONTAINER
141+
.getEndpointOverride(LocalStackContainer.Service.STS);
139142

140143
return new KinesisProducerConfiguration()
141144
.setCredentialsProvider(LocalstackContainerTest.credentialsProvider())
142145
.setRegion(LocalstackContainerTest.LOCAL_STACK_CONTAINER.getRegion())
143146
.setKinesisEndpoint(kinesisUri.getHost()).setKinesisPort(kinesisUri.getPort())
144147
.setCloudwatchEndpoint(cloudWatchUri.getHost()).setCloudwatchPort(cloudWatchUri.getPort())
148+
.setStsEndpoint(stsUri.getHost()).setStsPort(stsUri.getPort())
145149
.setVerifyCertificate(false);
146150
}
147151

@@ -172,6 +176,9 @@ public KclMessageDrivenChannelAdapter kclMessageDrivenChannelAdapter(PollableCha
172176
adapter.setStreamInitialSequence(
173177
InitialPositionInStreamExtended.newInitialPosition(InitialPositionInStream.TRIM_HORIZON));
174178
adapter.setBindSourceRecord(true);
179+
adapter.setMetricsLevel(MetricsLevel.NONE);
180+
adapter.setLeaseManagementConfigCustomizer(leaseManagementConfig -> leaseManagementConfig
181+
.workerUtilizationAwareAssignmentConfig().disableWorkerMetrics(true));
175182
return adapter;
176183
}
177184

0 commit comments

Comments
 (0)