Use CDL to block threads to avoid flaky tests.

apache · Jan 8, 2025 · ec8cafb · ec8cafb
1 parent d4f0a32
commit ec8cafb
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/lucene/core/src/test/org/apache/lucene/index/TestConcurrentMergeScheduler.java b/lucene/core/src/test/org/apache/lucene/index/TestConcurrentMergeScheduler.java
@@ -418,7 +418,6 @@ protected void doMerge(MergeSource mergeSource, MergePolicy.OneMerge merge)
     dir.close();
   }
 
-  @SuppressForbidden(reason = "Thread sleep")
   public void testIntraMergeThreadPoolIsLimitedByMaxThreads() throws IOException {
     ConcurrentMergeScheduler mergeScheduler = new ConcurrentMergeScheduler();
     MergeScheduler.MergeSource mergeSource =
@@ -475,11 +474,12 @@ public void merge(MergePolicy.OneMerge merge) throws IOException {
       Executor executor = mergeScheduler.intraMergeExecutor;
       AtomicInteger threadsExecutedOnPool = new AtomicInteger();
       AtomicInteger threadsExecutedOnSelf = new AtomicInteger();
-      for (int i = 0; i < 4; i++) {
+      CountDownLatch latch = new CountDownLatch(1);
+      final int totalThreads = 4;
+      for (int i = 0; i < totalThreads; i++) {
         mergeScheduler.mergeThreads.add(
             mergeScheduler.new MergeThread(mergeSource, merge) {
               @Override
-              @SuppressForbidden(reason = "Thread sleep")
               public void run() {
                 executor.execute(
                     () -> {
@@ -489,7 +489,7 @@ public void run() {
                         threadsExecutedOnPool.incrementAndGet();
                       }
                       try {
-                        Thread.sleep(100);
+                        latch.await();
                       } catch (InterruptedException e) {
                         throw new RuntimeException(e);
                       }
@@ -500,6 +500,10 @@ public void run() {
       for (ConcurrentMergeScheduler.MergeThread thread : mergeScheduler.mergeThreads) {
         thread.start();
       }
+      while (threadsExecutedOnSelf.get() + threadsExecutedOnPool.get() < totalThreads) {
+        Thread.yield();
+      }
+      latch.countDown();
       mergeScheduler.sync();
       assertEquals(3, threadsExecutedOnSelf.get());
       assertEquals(1, threadsExecutedOnPool.get());