Fix alluxio.util.executor.UniqueBlockingQueueTest.concurrentTest fail

Alluxio · Aug 2, 2024 · bef5f0a · bef5f0a
1 parent f3cf054
commit bef5f0a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/core/common/src/test/java/alluxio/util/executor/UniqueBlockingQueueTest.java b/core/common/src/test/java/alluxio/util/executor/UniqueBlockingQueueTest.java
@@ -17,6 +17,7 @@
 
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
 
 /**
  * Tests the {@link UniqueBlockingQueue}.
@@ -51,6 +52,14 @@ public void concurrentTest() throws Exception {
       });
     }
     executor.shutdown();
+    try {
+      if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {
+        executor.shutdownNow();
+      }
+    } catch (InterruptedException e) {
+      executor.shutdownNow();
+      Thread.currentThread().interrupt();
+    }
     assertEquals(20, test.size());
   }
 }