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" + ); } }