From 05d52c7c41d395240f2aae13998ecbe17599f51d Mon Sep 17 00:00:00 2001
From: Eval EXEC <execvy@gmail.com>
Date: Fri, 16 Aug 2024 15:57:41 +0800
Subject: [PATCH] Integration Test: wait find unverified blocks finish in
 BlockSyncFromOne and IBDProcess

---
 test/src/specs/sync/block_sync.rs  | 3 +++
 test/src/specs/sync/ibd_process.rs | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/test/src/specs/sync/block_sync.rs b/test/src/specs/sync/block_sync.rs
index abdb05607b..10f2ef9ffe 100644
--- a/test/src/specs/sync/block_sync.rs
+++ b/test/src/specs/sync/block_sync.rs
@@ -21,6 +21,9 @@ impl Spec for BlockSyncFromOne {
 
     // NOTE: ENSURE node0 and nodes1 is in genesis state.
     fn run(&self, nodes: &mut Vec<Node>) {
+        nodes
+            .iter()
+            .for_each(|node| node.wait_find_unverified_blocks_finished());
         let node0 = &nodes[0];
         let node1 = &nodes[1];
         let (rpc_client0, rpc_client1) = (node0.rpc_client(), node1.rpc_client());
diff --git a/test/src/specs/sync/ibd_process.rs b/test/src/specs/sync/ibd_process.rs
index 344cd1869a..dcc6e5f33a 100644
--- a/test/src/specs/sync/ibd_process.rs
+++ b/test/src/specs/sync/ibd_process.rs
@@ -8,6 +8,9 @@ impl Spec for IBDProcess {
     crate::setup!(num_nodes: 3);
 
     fn run(&self, nodes: &mut Vec<Node>) {
+        nodes
+            .iter()
+            .for_each(|node| node.wait_find_unverified_blocks_finished());
         info!("Running IBD process");
 
         let node0 = &nodes[0];