diff --git a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/GracefulExecutorServicesShutdownTest.java b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/GracefulExecutorServicesShutdownTest.java index b9785071a0e74..726f37d12cee4 100644 --- a/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/GracefulExecutorServicesShutdownTest.java +++ b/pulsar-broker/src/test/java/org/apache/pulsar/broker/service/GracefulExecutorServicesShutdownTest.java @@ -35,6 +35,7 @@ import java.util.concurrent.TimeoutException; import java.util.concurrent.atomic.AtomicBoolean; import org.testng.annotations.Test; +import org.awaitility.Awaitility; public class GracefulExecutorServicesShutdownTest { @@ -151,8 +152,8 @@ public void shouldTerminateWhenFutureIsCancelled() throws InterruptedException { future.cancel(false); // then - assertTrue(awaitTerminationInterrupted.get(), - "awaitTermination should have been interrupted"); + Awaitility.await().untilAsserted(() -> assertTrue(awaitTerminationInterrupted.get(), + "awaitTermination should have been interrupted")); verify(executorService, times(1)).awaitTermination(anyLong(), any()); verify(executorService, times(1)).shutdownNow(); }