Fixing Flaky Test testLeaderDoAssignmentForNewlyElectedLeaderFailureP…

…athVariation (#953) Co-authored-by: Shrinand Thakkar <[email protected]>
linkedin · Aug 15, 2023 · 2678147 · 2678147
1 parent d19a2c2
commit 2678147
Showing 1 changed file with 2 additions 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
@@ -4062,6 +4062,8 @@ protected synchronized void handleEvent(CoordinatorEvent event) {
     Datastream testDatastream =
         DatastreamTestUtils.createAndStoreDatastreams(zkClient, testCluster, connectorType, streamName)[0];
 
+    // Blocking until we have performed atleast 5 retries of LeaderDoAssignmentEvent with a newly elected leader.
+    PollUtils.poll(() -> shadowListWithPreviousAndNewHeadPairsAtNewLeaderDoAssignmentEvent.size() > 5, 50, 2000);
     coordinator.stop();
     zkClient.close();
     coordinator.getDatastreamCache().getZkclient().close();