change from uint64 to uint256

PufferFinance · Jun 17, 2024 · 1f17a0c · 1f17a0c
1 parent 95f14c2
commit 1f17a0c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 11 deletions.
diff --git a/mainnet-contracts/src/PufferL2Depositor.sol b/mainnet-contracts/src/PufferL2Depositor.sol
@@ -44,7 +44,7 @@ contract PufferL2Depositor is IPufferL2Depositor, AccessManaged {
      * @inheritdoc IPufferL2Depositor
      * @dev Restricted in this context is like `whenNotPaused` modifier from Pausable.sol
      */
-    function deposit(address token, address account, Permit calldata permitData, uint64 referralCode)
+    function deposit(address token, address account, Permit calldata permitData, uint256 referralCode)
         external
         onlySupportedTokens(token)
         restricted
@@ -75,7 +75,7 @@ contract PufferL2Depositor is IPufferL2Depositor, AccessManaged {
      * @inheritdoc IPufferL2Depositor
      * @dev Restricted in this context is like `whenNotPaused` modifier from Pausable.sol
      */
-    function depositETH(address account, uint64 referralCode) external payable restricted {
+    function depositETH(address account, uint256 referralCode) external payable restricted {
         IWETH(WETH).deposit{ value: msg.value }();
 
         _deposit({ token: WETH, depositor: msg.sender, account: account, amount: msg.value, referralCode: referralCode });
@@ -117,7 +117,7 @@ contract PufferL2Depositor is IPufferL2Depositor, AccessManaged {
      */
     function revertIfPaused() external restricted { }
 
-    function _deposit(address token, address depositor, address account, uint256 amount, uint64 referralCode)
+    function _deposit(address token, address depositor, address account, uint256 amount, uint256 referralCode)
         internal
     {
         PufToken pufToken = PufToken(tokens[token]);

diff --git a/mainnet-contracts/src/interface/IPufferL2Depositor.sol b/mainnet-contracts/src/interface/IPufferL2Depositor.sol
@@ -43,7 +43,7 @@ interface IPufferL2Depositor {
         address indexed depositor,
         address indexed account,
         uint256 tokenAmount,
-        uint64 referralCode
+        uint256 referralCode
     );
 
     event DepositCapUpdated(address indexed token, uint256 oldDepositCap, uint256 newDepositCap);
@@ -54,14 +54,14 @@ interface IPufferL2Depositor {
      *
      * @dev Restricted in this context is like `whenNotPaused` modifier from Pausable.sol
      */
-    function deposit(address token, address account, Permit calldata permitData, uint64 referralCode) external;
+    function deposit(address token, address account, Permit calldata permitData, uint256 referralCode) external;
 
     /**
      * @notice Deposits naative ETH by wrapping it into WETH and then depositing to corresponding token contract
      *
      * @dev Restricted in this context is like `whenNotPaused` modifier from Pausable.sol
      */
-    function depositETH(address account, uint64 referralCode) external payable;
+    function depositETH(address account, uint256 referralCode) external payable;
 
     /**
      * @notice Called by the Token contracts to check if the system is paused

diff --git a/mainnet-contracts/test/unit/PufferL2Staking.t.sol b/mainnet-contracts/test/unit/PufferL2Staking.t.sol
@@ -63,7 +63,7 @@ contract PufferL2Staking is UnitTestHelper {
 
     address bob = makeAddr("bob");
 
-    uint64 referralCode = 0;
+    uint256 referralCode = 0;
 
     function setUp() public override {
         super.setUp();
@@ -132,7 +132,7 @@ contract PufferL2Staking is UnitTestHelper {
     }
 
     // Bad permit signature + approve
-    function test_depositFor_dai_approve(uint32 amount, uint64 refCode) public {
+    function test_depositFor_dai_approve(uint32 amount, uint256 refCode) public {
         vm.assume(amount > 0);
 
         // This is a bad permit signature
@@ -208,7 +208,7 @@ contract PufferL2Staking is UnitTestHelper {
     }
 
     // Good Permit signature signature
-    function test_depositFor_dai_permit(uint32 amount, uint64 refCode) public {
+    function test_depositFor_dai_permit(uint32 amount, uint256 refCode) public {
         vm.assume(amount > 0);
 
         // Good permit signature
@@ -229,7 +229,7 @@ contract PufferL2Staking is UnitTestHelper {
     }
 
     // Weth doesn't have `permit` at all
-    function test_deposiFor_WETH(uint32 amount, uint64 refCode) public {
+    function test_deposiFor_WETH(uint32 amount, uint256 refCode) public {
         vm.assume(amount > 0);
 
         // WETH Doesn't have permit
@@ -253,7 +253,7 @@ contract PufferL2Staking is UnitTestHelper {
     }
 
     // ETH deposit & weth withdrawal
-    function test_depositFor_ETH_withdraw_weth(uint16 amount, uint64 refCode) public {
+    function test_depositFor_ETH_withdraw_weth(uint16 amount, uint256 refCode) public {
         vm.assume(amount > 0);
 
         vm.deal(bob, amount);