Fixing Flaky Test

linkedin · Aug 8, 2023 · b8d7192 · b8d7192
1 parent e7f3e2d
commit b8d7192
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/datastream-server-restli/src/test/java/com/linkedin/datastream/server/TestCoordinator.java b/datastream-server-restli/src/test/java/com/linkedin/datastream/server/TestCoordinator.java
@@ -4042,6 +4042,7 @@ protected void performPreAssignmentCleanup(List<DatastreamGroup> datastreamGroup
               protected synchronized void handleEvent(CoordinatorEvent event) {
                 CoordinatorEvent previousHead = peekCoordinatorEventBlockingQueue();
                 super.handleEvent(event);
+                PollUtils.poll(() -> peekCoordinatorEventBlockingQueue() != null, 50, 1000);
                 CoordinatorEvent nextHead = peekCoordinatorEventBlockingQueue();
 
                 // recording previous and new heads of the CoordinatorEventBlockingQueue