fix(core): always close the queue after receive

May improve test flakyness
kestra-io · Jan 14, 2025 · f217d33 · f217d33
1 parent 0c0ff37
commit f217d33
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/core/src/main/java/io/kestra/core/runners/RunnerUtils.java b/core/src/main/java/io/kestra/core/runners/RunnerUtils.java
@@ -152,15 +152,18 @@ public Execution awaitExecution(Predicate<Execution> predicate, Runnable executi
             }
         }, false);
 
-        executionEmitter.run();
+        try {
+            executionEmitter.run();
 
-        if (duration == null) {
-            Await.until(() -> receive.get() != null, Duration.ofMillis(10));
-        } else {
-            Await.until(() -> receive.get() != null, Duration.ofMillis(10), duration);
+            if (duration == null) {
+                Await.until(() -> receive.get() != null, Duration.ofMillis(10));
+            } else {
+                Await.until(() -> receive.get() != null, Duration.ofMillis(10), duration);
+            }
+        } finally {
+            cancel.run();
         }
 
-        cancel.run();
 
         return receive.get();
     }