diff --git a/src/test/java/org/springframework/integration/aws/kinesis/KinesisIntegrationTests.java b/src/test/java/org/springframework/integration/aws/kinesis/KinesisIntegrationTests.java index a7e2a16..abb1c30 100644 --- a/src/test/java/org/springframework/integration/aws/kinesis/KinesisIntegrationTests.java +++ b/src/test/java/org/springframework/integration/aws/kinesis/KinesisIntegrationTests.java @@ -109,13 +109,13 @@ public class KinesisIntegrationTests { this.kinesisSendChannel.send(MessageBuilder.withPayload(now).setHeader(AwsHeaders.STREAM, TEST_STREAM) .setHeader("foo", "BAR").build()); - Message receive = this.kinesisReceiveChannel.receive(10_000); + Message receive = this.kinesisReceiveChannel.receive(20_000); assertThat(receive).isNotNull(); assertThat(receive.getPayload()).isEqualTo(now); assertThat(receive.getHeaders()).contains(entry("foo", "BAR")); assertThat(receive.getHeaders()).containsKey(IntegrationMessageHeaderAccessor.SOURCE_DATA); - Message errorMessage = this.errorChannel.receive(10_000); + Message errorMessage = this.errorChannel.receive(20_000); assertThat(errorMessage).isNotNull(); assertThat(errorMessage.getHeaders().get(AwsHeaders.RAW_RECORD)).isNotNull(); assertThat(((Exception) errorMessage.getPayload()).getMessage()) diff --git a/src/test/java/org/springframework/integration/aws/kinesis/KplKclIntegrationTests.java b/src/test/java/org/springframework/integration/aws/kinesis/KplKclIntegrationTests.java index c1cfdc2..9b090ea 100644 --- a/src/test/java/org/springframework/integration/aws/kinesis/KplKclIntegrationTests.java +++ b/src/test/java/org/springframework/integration/aws/kinesis/KplKclIntegrationTests.java @@ -123,7 +123,7 @@ public class KplKclIntegrationTests { this.kinesisSendChannel.send(MessageBuilder.withPayload(now).setHeader(AwsHeaders.STREAM, TEST_STREAM) .setHeader("foo", "BAR").build()); - Message receive = this.kinesisReceiveChannel.receive(20_000); + Message receive = this.kinesisReceiveChannel.receive(30_000); assertThat(receive).isNotNull(); assertThat(receive.getPayload()).isEqualTo(now); assertThat(receive.getHeaders()).contains(entry("foo", "BAR"));