Fix flaky test GracefulExecutorServicesShutdownTest.shouldTerminateWh…

…enFutureIsCancelled (apache#10592) * fix flaky test GracefulExecutorServicesShutdownTest.shouldTerminateWhenFutureIsCancelled * avoid import *
xche · May 14, 2021 · 55376af · 55376af
1 parent c4098d6
commit 55376af
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.../src/test/java/org/apache/pulsar/broker/service/GracefulExecutorServicesShutdownTest.java b/.../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();
     }