rusk: unit-test: Pass base_root to rusk.revert

dusk-network · Jan 19, 2024 · c8444d3 · c8444d3
1 parent 4e79c7c
commit c8444d3
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/rusk/benches/block_ingestion.rs b/rusk/benches/block_ingestion.rs
@@ -99,7 +99,9 @@ pub fn accept_benchmark(c: &mut Criterion) {
                         )
                         .expect("Accepting transactions should succeed");
 
-                        rusk.revert().expect("Reverting should succeed");
+                        let base_root = rusk.base_root();
+                        rusk.revert(base_root)
+                            .expect("Reverting should succeed");
                     })
                 },
             );

diff --git a/rusk/src/lib/lib.rs b/rusk/src/lib/lib.rs
@@ -337,7 +337,7 @@ impl Rusk {
     pub fn revert(&self, state_hash: [u8; 32]) -> Result<[u8; 32]> {
         let mut inner = self.inner.lock();
 
-        let mut commits = &inner.vm.commits();
+        let commits = &inner.vm.commits();
         if !commits.contains(&state_hash) {
             return Err(Error::CommitNotFound(state_hash));
         }

diff --git a/rusk/tests/services/transfer.rs b/rusk/tests/services/transfer.rs
@@ -185,7 +185,8 @@ pub async fn wallet() -> Result<()> {
     assert_ne!(original_root, new_root, "Root should have changed");
 
     // Revert the state
-    rusk.revert().expect("Reverting should succeed");
+    let base_root = rusk.base_root();
+    rusk.revert(base_root).expect("Reverting should succeed");
     cache.write().unwrap().clear();
 
     // Check the state's root is back to the original one