fix unit test

fixed off-by-one issue producing unexpected results.
OffchainLabs · Nov 2, 2023 · eab0519 · eab0519
1 parent c3ab7ad
commit eab0519
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/arbitrator/stylus/tests/nonreentrant-enter/src/main.rs b/arbitrator/stylus/tests/nonreentrant-enter/src/main.rs
@@ -26,7 +26,5 @@ fn user_main(input: Vec<u8>) -> Result<Vec<u8>, Vec<u8>> {
     }
     console!("counter: {}", input[40]);
 
-    let mut input = input.to_vec();
-    input[40] -= 1;
     RawCall::new().call(dest, input.as_slice())
 }
diff --git a/arbitrator/stylus/tests/reentrant-enter/src/main.rs b/arbitrator/stylus/tests/reentrant-enter/src/main.rs
@@ -26,7 +26,5 @@ fn user_main(input: Vec<u8>) -> Result<Vec<u8>, Vec<u8>> {
     }
     console!("counter: {}", input[40]);
 
-    let mut input = input.to_vec();
-    input[40] -= 1;
     unsafe{RawCall::new().call(dest, input.as_slice())}
 }
diff --git a/arbitrator/stylus/tests/reentrant-reenter/src/main.rs b/arbitrator/stylus/tests/reentrant-reenter/src/main.rs
@@ -23,5 +23,7 @@ fn user_main(input: Vec<u8>) -> Result<Vec<u8>, Vec<u8>> {
     if input[40] == 0 {
         return unsafe{RawCall::new_delegate().call(dest, input.as_slice())}
     }
+    let mut input = input;
+    input[40] -= 1;
     unsafe{RawCall::new().call(dest, input.as_slice())}
 }