feat(cheatcode): UintToHex cheatcode (#7767)

uintToHex cheatcode
foundry-rs · Apr 23, 2024 · 21dfde4 · 21dfde4
1 parent 6ff4385
commit 21dfde4
Show file tree

Hide file tree

Showing 6 changed files with 36 additions and 0 deletions.
diff --git a/crates/cheatcodes/assets/cheatcodes.json b/crates/cheatcodes/assets/cheatcodes.json
diff --git a/crates/cheatcodes/spec/src/vm.rs b/crates/cheatcodes/spec/src/vm.rs
@@ -1865,6 +1865,11 @@ interface Vm {
         returns (string memory json);
     /// See `serializeJson`.
     #[cheatcode(group = Json)]
+    function serializeUintToHex(string calldata objectKey, string calldata valueKey, uint256 value)
+        external
+        returns (string memory json);
+    /// See `serializeJson`.
+    #[cheatcode(group = Json)]
     function serializeInt(string calldata objectKey, string calldata valueKey, int256 value)
         external
         returns (string memory json);

diff --git a/crates/cheatcodes/src/json.rs b/crates/cheatcodes/src/json.rs
@@ -248,6 +248,14 @@ impl Cheatcode for serializeBytes_1Call {
     }
 }
 
+impl Cheatcode for serializeUintToHexCall {
+    fn apply(&self, state: &mut Cheatcodes) -> Result {
+        let Self { objectKey, valueKey, value } = self;
+        let hex = format!("0x{:x}", value);
+        serialize_json(state, objectKey, Some(valueKey), &hex)
+    }
+}
+
 impl Cheatcode for writeJson_0Call {
     fn apply(&self, state: &mut Cheatcodes) -> Result {
         let Self { json, path } = self;

diff --git a/crates/evm/traces/src/decoder/mod.rs b/crates/evm/traces/src/decoder/mod.rs
@@ -441,6 +441,7 @@ impl CallTraceDecoder {
             "keyExistsJson" |
             "serializeBool" |
             "serializeUint" |
+            "serializeUintToHex" |
             "serializeInt" |
             "serializeAddress" |
             "serializeBytes32" |

diff --git a/testdata/cheats/Vm.sol b/testdata/cheats/Vm.sol
diff --git a/testdata/default/cheats/Json.t.sol b/testdata/default/cheats/Json.t.sol
@@ -224,6 +224,7 @@ contract WriteJsonTest is DSTest {
         vm.serializeBool(json1, "boolean", true);
         vm.serializeInt(json2, "key2", -234);
         vm.serializeUint(json2, "deploy", uint256(254));
+        vm.serializeUintToHex(json2, "hexUint", uint256(255));
         string memory data = vm.serializeBool(json2, "boolean", true);
         vm.serializeString(json2, "json1", data);
         emit log(data);