From 3a40cfd472f2d1cc7741bd640fe506e5e40cfc18 Mon Sep 17 00:00:00 2001
From: William Wen <william123.wen@gmail.com>
Date: Wed, 25 Oct 2023 18:16:48 +0800
Subject: [PATCH] fix some test

---
 src/tests/simulation/tests/integration_tests/sink/scale.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/tests/simulation/tests/integration_tests/sink/scale.rs b/src/tests/simulation/tests/integration_tests/sink/scale.rs
index f327468543a37..56a12e62d3661 100644
--- a/src/tests/simulation/tests/integration_tests/sink/scale.rs
+++ b/src/tests/simulation/tests/integration_tests/sink/scale.rs
@@ -36,7 +36,10 @@ async fn scale_and_check(
 ) -> Result<()> {
     for (plan, expected_parallelism) in schedule_plan {
         let prev_count = test_sink.store.id_count();
-        assert!(prev_count < target_count, "sink finish before scale");
+        assert!(prev_count <= target_count);
+        if prev_count == target_count {
+            return Ok(());
+        }
         cluster.reschedule(plan).await?;
         let after_count = test_sink.store.id_count();
         sleep(Duration::from_secs(10)).await;
@@ -50,7 +53,7 @@ async fn scale_and_check(
                 after_count
             );
             cluster.kill_node(&KillOpts::ALL).await;
-            sleep(Duration::from_secs(5)).await;
+            sleep(Duration::from_secs(10)).await;
         }
     }
     Ok(())