From 575874705f8138e002ec9ade3400f93e4d59fcfb Mon Sep 17 00:00:00 2001
From: Abhishek Agarwal <1477457+abhishekagarwal87@users.noreply.github.com>
Date: Thu, 17 Feb 2022 12:08:46 +0530
Subject: [PATCH] Fix the flakiness in getLockedIntervals test (#12172)

Fix the flakiness in getLockedIntervals test
---
 .../java/org/apache/druid/tests/indexer/ITIndexerTest.java   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/integration-tests/src/test/java/org/apache/druid/tests/indexer/ITIndexerTest.java b/integration-tests/src/test/java/org/apache/druid/tests/indexer/ITIndexerTest.java
index 93c8d462165..04222abad80 100644
--- a/integration-tests/src/test/java/org/apache/druid/tests/indexer/ITIndexerTest.java
+++ b/integration-tests/src/test/java/org/apache/druid/tests/indexer/ITIndexerTest.java
@@ -360,7 +360,10 @@ public class ITIndexerTest extends AbstractITBatchIndexTest
           Collections.singletonList(Intervals.of("2013-08-31/2013-09-02"))
       );
 
-      waitForAllTasksToCompleteForDataSource(datasourceName);
+      ITRetryUtil.retryUntilTrue(
+          () -> coordinator.areSegmentsLoaded(datasourceName),
+          "Segment Load"
+      );
     }
   }