feat: update PTokensClaimed event

sense-finance · Jul 22, 2024 · f3d90e6 · f3d90e6
1 parent 90a84e4
commit f3d90e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/contracts/PointTokenVault.sol b/contracts/PointTokenVault.sol
@@ -67,7 +67,7 @@ contract PointTokenVault is UUPSUpgradeable, AccessControlUpgradeable, Multicall
     event Deposit(address indexed depositor, address indexed receiver, address indexed token, uint256 amount);
     event Withdraw(address indexed withdrawer, address indexed receiver, address indexed token, uint256 amount);
     event RootUpdated(bytes32 prevRoot, bytes32 newRoot);
-    event PTokensClaimed(address indexed account, bytes32 indexed pointsId, uint256 amount);
+    event PTokensClaimed(address indexed account, address indexed receiver, bytes32 indexed pointsId, uint256 amount, uint256 fee);
     event RewardsClaimed(
         address indexed owner, address indexed receiver, bytes32 indexed pointsId, uint256 amount, uint256 fee
     );
@@ -151,7 +151,7 @@ contract PointTokenVault is UUPSUpgradeable, AccessControlUpgradeable, Multicall
 
         pTokens[pointsId].mint(_receiver, _claim.amountToClaim - pTokenFee); // Subtract mint fee.
 
-        emit PTokensClaimed(_account, pointsId, _claim.amountToClaim);
+        emit PTokensClaimed(_account, _receiver, pointsId, _claim.amountToClaim, pTokenFee);
     }
 
     function trustClaimer(address _account, bool _isTrusted) public {

diff --git a/contracts/test/PointTokenVault.t.sol b/contracts/test/PointTokenVault.t.sol
@@ -242,6 +242,8 @@ contract PointTokenVaultTest is Test {
         );
     }
 
+    event PTokensClaimed(address indexed account, address indexed receiver, bytes32 indexed pointsId, uint256 amount, uint256 fee);
+
     function test_Distribution() public {
         // Merkle tree created from leaves [keccack(vitalik, pointsId, 1e18), keccack(toly, pointsId, 0.5e18)].
         bytes32[] memory goodProof = new bytes32[](1);
@@ -271,6 +273,8 @@ contract PointTokenVaultTest is Test {
 
         // Can claim with the right proof
         vm.prank(vitalik);
+        vm.expectEmit(true, true, true, true);
+        emit PTokensClaimed(vitalik, vitalik, eigenPointsId, 1e18, 0);
         pointTokenVault.claimPTokens(PointTokenVault.Claim(eigenPointsId, 1e18, 1e18, goodProof), vitalik, vitalik);
 
         assertEq(pointTokenVault.pTokens(eigenPointsId).balanceOf(vitalik), 1e18);