From 8b473cd3cf6889044f3d3c465959934c01ae7f07 Mon Sep 17 00:00:00 2001 From: Soby Chacko Date: Fri, 26 Aug 2022 12:25:17 -0400 Subject: [PATCH] Trying to increase latency in concurrency test --- .../ConcurrentPulsarMessageListenerContainerTests.java | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/spring-pulsar/src/test/java/org/springframework/pulsar/listener/ConcurrentPulsarMessageListenerContainerTests.java b/spring-pulsar/src/test/java/org/springframework/pulsar/listener/ConcurrentPulsarMessageListenerContainerTests.java index 304fdc63..563332f0 100644 --- a/spring-pulsar/src/test/java/org/springframework/pulsar/listener/ConcurrentPulsarMessageListenerContainerTests.java +++ b/spring-pulsar/src/test/java/org/springframework/pulsar/listener/ConcurrentPulsarMessageListenerContainerTests.java @@ -17,12 +17,14 @@ package org.springframework.pulsar.listener; import static org.assertj.core.api.Assertions.assertThatThrownBy; +import static org.awaitility.Awaitility.await; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; +import java.time.Duration; import java.util.Map; import org.apache.pulsar.client.api.BatchReceivePolicy; @@ -63,9 +65,9 @@ public class ConcurrentPulsarMessageListenerContainerTests { container.start(); - verify(pulsarConsumerFactory, times(3)).createConsumer(any(Schema.class), any(BatchReceivePolicy.class), - any(Map.class)); - verify(consumer, times(3)).batchReceive(); + await().atMost(Duration.ofSeconds(10)).untilAsserted(() -> verify(pulsarConsumerFactory, times(3)) + .createConsumer(any(Schema.class), any(BatchReceivePolicy.class), any(Map.class))); + await().atMost(Duration.ofSeconds(10)).untilAsserted(() -> verify(consumer, times(3)).batchReceive()); } @Test