You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by ju...@apache.org on 2026/06/24 02:19:53 UTC
(iotdb) 01/01: Fix flaky multilevel priority queue test
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 <87...@users.noreply.github.com>
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) {