From d643d657dfe4a11716e31cbda184b43917a83a32 Mon Sep 17 00:00:00 2001
From: Sara Reynolds <snreynolds2506@gmail.com>
Date: Sun, 20 Oct 2024 17:58:20 -0400
Subject: [PATCH] add decode fuzz test

---
 test/libraries/CalldataDecoder.t.sol | 7 +++++++
 test/mocks/MockCalldataDecoder.sol   | 4 ++++
 2 files changed, 11 insertions(+)

diff --git a/test/libraries/CalldataDecoder.t.sol b/test/libraries/CalldataDecoder.t.sol
index c41122eb..dd47a949 100644
--- 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
index e25d8ec7..c61db4af 100644
--- 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();
+    }
 }