This is an automated email from the ASF dual-hosted git repository. Caideyipi pushed a commit to branch fix/multilevel-priority-queue-test-flake in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit b29ccab05f905232d663c984876a12e650d355d0 Author: Caideyipi <[email protected]> AuthorDate: Wed Jun 24 10:19:31 2026 +0800 Fix flaky multilevel priority queue test --- .../schedule/queue/MultilevelPriorityQueueTest.java | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/schedule/queue/MultilevelPriorityQueueTest.java b/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/schedule/queue/MultilevelPriorityQueueTest.java index 0ee68774b12..42bb75a2dc5 100644 --- a/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/schedule/queue/MultilevelPriorityQueueTest.java +++ b/iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/execution/schedule/queue/MultilevelPriorityQueueTest.java @@ -38,6 +38,9 @@ import org.mockito.Mockito; import java.util.ArrayList; import java.util.List; import java.util.OptionalInt; +import java.util.concurrent.TimeUnit; + +import static org.awaitility.Awaitility.await; public class MultilevelPriorityQueueTest { @Test @@ -58,12 +61,14 @@ public class MultilevelPriorityQueueTest { } }); t1.start(); - Thread.sleep(100); - Assert.assertEquals(Thread.State.WAITING, t1.getState()); + await() + .atMost(1, TimeUnit.MINUTES) + .untilAsserted(() -> Assert.assertEquals(Thread.State.WAITING, t1.getState())); DriverTask e2 = mockDriverTask(mockDriverTaskId(), false); queue.push(e2); - Thread.sleep(100); - Assert.assertEquals(Thread.State.TERMINATED, t1.getState()); + await() + .atMost(1, TimeUnit.MINUTES) + .untilAsserted(() -> Assert.assertEquals(Thread.State.TERMINATED, t1.getState())); Assert.assertEquals(1, res.size()); Assert.assertEquals(e2.getDriverTaskId().toString(), res.get(0).getDriverTaskId().toString()); } catch (Exception e) {
