From 133731f288120879b6afdef3db53f183124e5567 Mon Sep 17 00:00:00 2001 From: Artem Bilan Date: Tue, 26 Jun 2018 13:34:01 -0400 Subject: [PATCH] Fix race condition in the KinesisMDChATests --- .../inbound/KinesisMessageDrivenChannelAdapterTests.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/test/java/org/springframework/integration/aws/inbound/KinesisMessageDrivenChannelAdapterTests.java b/src/test/java/org/springframework/integration/aws/inbound/KinesisMessageDrivenChannelAdapterTests.java index 84e13c4..c7604dc 100644 --- a/src/test/java/org/springframework/integration/aws/inbound/KinesisMessageDrivenChannelAdapterTests.java +++ b/src/test/java/org/springframework/integration/aws/inbound/KinesisMessageDrivenChannelAdapterTests.java @@ -160,6 +160,10 @@ public class KinesisMessageDrivenChannelAdapterTests { Assert.assertThat(0, eventually(equalsResult(forLocking::size))); + List consumerInvoker = + TestUtils.getPropertyValue(this.kinesisMessageDrivenChannelAdapter, "consumerInvokers", List.class); + Assert.assertThat(0, eventually(equalsResult(consumerInvoker::size))); + this.kinesisMessageDrivenChannelAdapter.setListenerMode(ListenerMode.batch); this.kinesisMessageDrivenChannelAdapter.setCheckpointMode(CheckpointMode.record); this.checkpointStore.put("SpringIntegration" + ":" + STREAM1 + ":" + "1", "1"); @@ -186,7 +190,7 @@ public class KinesisMessageDrivenChannelAdapterTests { eventually(equalsResult(() -> this.checkpointStore.get("SpringIntegration" + ":" + STREAM1 + ":" + "1")))); - List consumerInvoker = + consumerInvoker = TestUtils.getPropertyValue(this.kinesisMessageDrivenChannelAdapter, "consumerInvokers", List.class); assertThat(consumerInvoker.size()).isEqualTo(2);