use IntegrationTest::generate_block to generate blocks

nervosnetwork · Feb 18, 2024 · 8371c48 · 8371c48
1 parent df44a06
commit 8371c48
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/test/src/miner.rs b/test/src/miner.rs
@@ -41,25 +41,18 @@ impl Miner {
     }
 
     pub fn generate_block(&self) -> H256 {
-        let template = self
-            .rpc
-            .lock()
-            .unwrap()
-            .get_block_template(None, None, None)
-            .expect("RPC get_block_template");
-        let work_id = template.work_id.value();
-        let block = Into::<Block>::into(template);
         self.rpc
             .lock()
             .unwrap()
-            .submit_block(work_id.to_string(), block.into())
-            .expect("RPC submit_block")
+            .generate_block()
+            .expect("RPC generate_block")
     }
 
     pub fn generate_blocks(&self, count: u64) {
         log::info!("generating {} blocks...", count);
         (0..count).for_each(|_| {
-            self.generate_block();
+            let block_hash = self.generate_block();
+            log::info!("generated block: {}", block_hash);
             thread::sleep(Duration::from_millis(10));
         })
     }

diff --git a/test/src/setup.rs b/test/src/setup.rs
@@ -168,6 +168,10 @@ impl Setup {
             .as_array_mut()
             .unwrap()
             .push(toml::Value::String("Indexer".to_string()));
+        value["rpc"]["modules"]
+            .as_array_mut()
+            .unwrap()
+            .push(toml::Value::String("IntegrationTest".to_string()));
         // value["indexer_v2"]["index_tx_pool"] = true;
         value
             .as_table_mut()