add decode fuzz test

Uniswap · Oct 20, 2024 · d643d65 · d643d65
1 parent 0070fcd
commit d643d65
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/test/libraries/CalldataDecoder.t.sol b/test/libraries/CalldataDecoder.t.sol
@@ -232,6 +232,13 @@ contract CalldataDecoderTest is Test {
         assertEq(amount, _amount);
     }
 
+    function test_fuzz_decodeUint256(uint256 _amount) public {
+        bytes memory params = abi.encode(_amount);
+        (uint256 amount) = decoder.decodeUint256(params);
+
+        assertEq(amount, _amount);
+    }
+
     function _assertEq(PathKey[] memory path1, PathKey[] memory path2) internal pure {
         assertEq(path1.length, path2.length);
         for (uint256 i = 0; i < path1.length; i++) {

diff --git a/test/mocks/MockCalldataDecoder.sol b/test/mocks/MockCalldataDecoder.sol
@@ -136,4 +136,8 @@ contract MockCalldataDecoder {
     {
         return params.decodeCurrencyAddressAndUint256();
     }
+
+    function decodeUint256(bytes calldata params) external pure returns (uint256) {
+        return params.decodeUint256();
+    }
 }