Use the state of the database before the benchmarks

FuelLabs · Feb 20, 2025 · a2d5410 · a2d5410
1 parent 2471df6
commit a2d5410
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benches/src/lib.rs b/benches/src/lib.rs
@@ -519,6 +519,7 @@ impl TryFrom<VmBench> for VmBenchPrepared {
         let interpreter_params = InterpreterParams::new(gas_price, &params);
         let memory = memory.unwrap_or_else(MemoryInstance::new);
 
+        let original_db = db.clone();
         let mut txtor = Transactor::new(memory, db, interpreter_params);
 
         txtor.transact(tx);
@@ -556,6 +557,7 @@ impl TryFrom<VmBench> for VmBenchPrepared {
         diff += storage_diff;
         let diff: diff::Diff<diff::InitialVmState> = diff.into();
         vm.reset_vm_state(&diff);
+        *vm.as_mut() = original_db;
         assert_eq!(vm_before_first_instruction, vm);
 
         Ok(Self {