Speed up BinPackingNodeAllocator test

trinodb · May 18, 2022 · 320250e · 320250e
1 parent c92ea3b
commit 320250e
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/core/trino-main/src/test/java/io/trino/execution/scheduler/TestBinPackingNodeAllocator.java b/core/trino-main/src/test/java/io/trino/execution/scheduler/TestBinPackingNodeAllocator.java
@@ -692,14 +692,17 @@ private void assertNotAcquired(NodeAllocator.NodeLease lease)
 
     private static void assertEventually(ThrowingRunnable assertion)
     {
-        Assert.assertEventually(() -> {
-            try {
-                assertion.run();
-            }
-            catch (Exception e) {
-                throw new RuntimeException(e);
-            }
-        });
+        Assert.assertEventually(
+                new io.airlift.units.Duration(TEST_TIMEOUT, TimeUnit.MILLISECONDS),
+                new io.airlift.units.Duration(10, TimeUnit.MILLISECONDS),
+                () -> {
+                    try {
+                        assertion.run();
+                    }
+                    catch (Exception e) {
+                        throw new RuntimeException(e);
+                    }
+                });
     }
 
     interface ThrowingRunnable