fix test case for forking main

starcoinorg · Jan 19, 2024 · a720253 · a720253
1 parent 6f5d331
commit a720253
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/sync/src/block_connector/test_write_dag_block_chain.rs b/sync/src/block_connector/test_write_dag_block_chain.rs
@@ -104,7 +104,7 @@ fn gen_fork_dag_block_chain(
     writeable_block_chain_service: &mut WriteBlockChainService<MockTxPoolService>,
 ) -> Option<HashValue> {
     let miner_account = AccountInfo::random();
-    let dag = BlockDAG::create_for_testing().unwrap();
+    let dag = writeable_block_chain_service.get_dag();
     if let Some(block_header) = writeable_block_chain_service
         .get_main()
         .get_header_by_number(fork_number)

diff --git a/sync/src/block_connector/write_block_chain.rs b/sync/src/block_connector/write_block_chain.rs
@@ -181,6 +181,10 @@ where
         &self.main
     }
 
+    pub fn get_dag(&self) -> BlockDAG {
+        self.dag.clone()
+    }
+
     #[cfg(test)]
     pub fn create_block(
         &self,