fixup! intro AWAIT_WITH_FLUSH

apache · Dec 2, 2024 · ba7fdf9 · ba7fdf9
1 parent 494b51a
commit ba7fdf9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...tion-test/src/test/java/org/apache/iotdb/subscription/it/IoTDBSubscriptionITConstant.java b/...tion-test/src/test/java/org/apache/iotdb/subscription/it/IoTDBSubscriptionITConstant.java
@@ -54,7 +54,7 @@ public interface WrappedVoidSupplier {
   public static void AWAIT_WITH_FLUSH(final Session session, final WrappedVoidSupplier assertions) {
     AWAIT.untilAsserted(
         () -> {
-          session.executeQueryStatement("flush");
+          session.executeNonQueryStatement("flush");
           assertions.get();
         });
   }

diff --git a/...ubscription/it/triple/regression/pushconsumer/multi/IoTDBOneConsumerMultiTopicsMixIT.java b/...ubscription/it/triple/regression/pushconsumer/multi/IoTDBOneConsumerMultiTopicsMixIT.java
@@ -54,7 +54,7 @@
 import java.util.List;
 import java.util.concurrent.atomic.AtomicInteger;
 
-import static org.apache.iotdb.subscription.it.IoTDBSubscriptionITConstant.AWAIT;
+import static org.apache.iotdb.subscription.it.IoTDBSubscriptionITConstant.AWAIT_WITH_FLUSH;
 
 /***
  * 1 consumer subscribes to 2 topics: historical data
@@ -229,7 +229,8 @@ public void do_test()
     thread.start();
     thread.join();
 
-    AWAIT.untilAsserted(
+    AWAIT_WITH_FLUSH(
+        session_src,
         () -> {
           assertEquals(rowCount1.get(), 0, "pattern1");
           check_count(12, "select count(s_0) from " + device, "dataset pattern1");