[Dataflow] Fix flaky BoundedQueueExecutor test (#34878)

scwhittle · web-flow · commit 23dc44705920 · 2025-05-12T11:30:15.000+02:00
diff --git a/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/util/BoundedQueueExecutorTest.java b/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/util/BoundedQueueExecutorTest.java
@@ -172,9 +172,10 @@ public void testScheduleWorkWhenExceedMaximumBytesOutstanding() throws Exception
 
     // Stop m1 so there are available bytes for m2 to run.
     processStop1.countDown();
+    // m2 should be able to start execution
     processStart2.await();
-    // m2 started.
-    assertEquals(Thread.State.TERMINATED, m2Runner.getState());
+    // ensure that the execute() call scheduling m2 returns even if the completion of m2 is blocked.
+    m2Runner.join();
     processStop2.countDown();
     executor.shutdown();
   }
