Fix tests with cache

apache · Feb 9, 2025 · aa34805 · aa34805
1 parent 600cce2
commit aa34805
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...ing-service/src/test/java/org/apache/druid/indexing/common/actions/TaskActionTestKit.java b/...ing-service/src/test/java/org/apache/druid/indexing/common/actions/TaskActionTestKit.java
@@ -101,7 +101,7 @@ public void setUseSegmentMetadataCache(boolean useSegmentMetadataCache)
 
   public void syncSegmentMetadataCache()
   {
-    metadataCachePollExec.finishNextPendingTask();
+    metadataCachePollExec.finishNextPendingTasks(4);
   }
 
   @Override

diff --git a/server/src/test/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinatorTest.java b/server/src/test/java/org/apache/druid/metadata/IndexerSQLMetadataStorageCoordinatorTest.java
@@ -158,7 +158,7 @@ public void setUp()
     if (useSegmentCache) {
       segmentMetadataCache.start();
       segmentMetadataCache.becomeLeader();
-      cachePollExecutor.finishNextPendingTask();
+      cachePollExecutor.finishNextPendingTasks(4);
     }
 
     transactionFactory = new SqlSegmentMetadataTransactionFactory(
@@ -210,7 +210,7 @@ public void tearDown()
   void refreshCache()
   {
     if (useSegmentCache) {
-      cachePollExecutor.finishNextPendingTask();
+      cachePollExecutor.finishNextPendingTasks(2);
     }
   }