add back the trasnsfer call

PufferFinance · Jun 26, 2024 · 84397e2 · 84397e2
1 parent 0087cb3
commit 84397e2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/mainnet-contracts/src/PufToken.sol b/mainnet-contracts/src/PufToken.sol
@@ -109,6 +109,8 @@ contract PufToken is IPufStakingPool, ERC20, ERC20Permit {
     function withdraw(address recipient, uint256 amount) external validateAddressAndAmount(recipient, amount) {
         _burn(msg.sender, amount);
 
+        TOKEN.safeTransfer(recipient, amount);
+
         emit Withdrawn(msg.sender, recipient, amount);
     }
 

diff --git a/mainnet-contracts/test/unit/PufferL2Staking.t.sol b/mainnet-contracts/test/unit/PufferL2Staking.t.sol
@@ -178,6 +178,8 @@ contract PufferL2Staking is UnitTestHelper {
         vm.expectEmit(true, true, true, true);
         emit IPufStakingPool.Withdrawn(bob, bob, amount); // original deposit amount
         pufToken.withdraw(bob, amount);
+
+        assertEq(sixDecimal.balanceOf(bob), amount, "bob got same amount");
     }
 
     // Deposit & withdraw 22 decimal token
@@ -205,6 +207,8 @@ contract PufferL2Staking is UnitTestHelper {
         vm.expectEmit(true, true, true, true);
         emit IPufStakingPool.Withdrawn(bob, bob, amount); // original deposit amount
         pufToken.withdraw(bob, amount);
+
+        assertEq(twentyTwoDecimal.balanceOf(bob), amount, "bob got same amount");
     }
 
     // Good Permit signature signature