fix(tests): update test cases for postprocessing cleanup

Jon-Becker · Jan 27, 2024 · d167b00 · d167b00
1 parent ca5c994
commit d167b00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/src/ether/lexers/cleanup.rs b/common/src/ether/lexers/cleanup.rs
@@ -330,7 +330,7 @@ mod tests {
     fn test_simplify_casts_simple() {
         let line = String::from("uint256(uint256(arg0));");
 
-        assert_eq!(simplify_casts(&line), String::from("uint256(arg0);"));
+        assert_eq!(simplify_casts(&line), String::from("uint256((arg0));"));
     }
 
     #[test]
@@ -346,7 +346,7 @@ mod tests {
 
         assert_eq!(
             simplify_casts(&line),
-            String::from("ecrecover(uint256(arg0), uint256(arg0), uint256((arg0)));")
+            String::from("ecrecover(uint256((arg0)), uint256((arg0)), uint256(((arg0))));")
         ); // double parens are expected because we dont simplify_parentheses here
     }