Improve test_accept_not_a_better_block wait chain_service finished be…

…fore exit
nervosnetwork · Aug 2, 2024 · 9a96db5 · 9a96db5
1 parent b31c07d
commit 9a96db5
Showing 1 changed file with 15 additions and 0 deletions.
diff --git 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]