(iotdb) branch master updated: Fix flaky multilevel priority queue test (#18013)

[jiangtian]

This is an automated email from the ASF dual-hosted git repository.

jt2594838 pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/master by this push:
     new 5467513eaa2 Fix flaky multilevel priority queue test (#18013)
5467513eaa2 is described below

commit 5467513eaa25cbe73758392502075c4713545318
Author: Caideyipi <87789683+caidey...@users.noreply.github.com>
AuthorDate: Wed Jun 24 14:51:19 2026 +0800

    Fix flaky multilevel priority queue test (#18013)
---
 .../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) {