Fix mock consensus.get_block_roots_for_period()

commonprefix · Jan 29, 2024 · 08b7154 · 08b7154
1 parent 0c0b430
commit 08b7154
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/prover/src/prover/proof_generator.rs b/prover/src/prover/proof_generator.rs
@@ -338,6 +338,15 @@ mod tests {
             Ok(tree)
         });
 
+        consensus
+            .expect_get_block_roots_for_period()
+            .returning(|_| {
+                let file = File::open("./src/prover/testdata/block_roots.json").unwrap();
+                let tree: Vector<_, SLOTS_PER_HISTORICAL_ROOT> =
+                    serde_json::from_reader(file).unwrap();
+                Ok(tree)
+            });
+
         let mut state_prover = MockStateProver::new();
 
         state_prover