From 9a96db5e9e1cbbc108ce3373060a3913a7f9ca7b Mon Sep 17 00:00:00 2001
From: Eval EXEC <execvy@gmail.com>
Date: Fri, 2 Aug 2024 11:57:56 +0800
Subject: [PATCH] Improve test_accept_not_a_better_block wait chain_service
 finished before exit

---
 sync/src/relayer/tests/compact_block_process.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/sync/src/relayer/tests/compact_block_process.rs b/sync/src/relayer/tests/compact_block_process.rs
index df4e7491d4..3473b8e405 100644
--- a/sync/src/relayer/tests/compact_block_process.rs
+++ b/sync/src/relayer/tests/compact_block_process.rs
@@ -200,6 +200,21 @@ fn test_accept_not_a_better_block() {
         peer_index,
     );
     assert_eq!(compact_block_process.execute(), Status::ok());
+
+    // wait chain_service processed the compact block, check block hash in snapshot
+    {
+        let now = std::time::Instant::now();
+        loop {
+            std::thread::sleep(std::time::Duration::from_millis(100));
+            if now.elapsed().as_secs() > 5 {
+                panic!("wait chain_service processed the compact block timeout");
+            }
+            let snapshot = relayer.shared.shared().snapshot();
+            if snapshot.get_block(&uncle_block.header().hash()).is_some() {
+                break;
+            }
+        }
+    }
 }
 
 #[test]