feat: add prev cap

sense-finance · May 3, 2024 · 7dd072e · 7dd072e
1 parent e757804
commit 7dd072e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/contracts/PointTokenVault.sol b/contracts/PointTokenVault.sol
@@ -64,7 +64,7 @@ contract PointTokenVault is UUPSUpgradeable, AccessControlUpgradeable, Multicall
         bytes32 indexed pointsId, ERC20 rewardToken, uint256 rewardsPerPToken, bool isMerkleBased
     );
     event PTokenDeployed(bytes32 indexed pointsId, address indexed pToken);
-    event CapSet(address indexed token, uint256 cap);
+    event CapSet(address indexed token, uint256 prevCap, uint256 cap);
 
     error ProofInvalidOrExpired();
     error ClaimTooLarge();
@@ -214,8 +214,9 @@ contract PointTokenVault is UUPSUpgradeable, AccessControlUpgradeable, Multicall
     }
 
     function setCap(address _token, uint256 _cap) external onlyRole(OPERATOR_ROLE) {
+        uint256 prevCap = caps[_token];
         caps[_token] = _cap;
-        emit CapSet(_token, _cap);
+        emit CapSet(_token, prevCap, _cap);
     }
 
     function setIsCapped(bool _isCapped) external onlyRole(OPERATOR_ROLE) {

diff --git a/contracts/test/PointTokenVault.t.sol b/contracts/test/PointTokenVault.t.sol
@@ -96,13 +96,17 @@ contract PointTokenVaultTest is Test {
         assertEq(pointTokenVault.balances(vitalik, pointEarningToken), 0);
     }
 
+    event CapSet(address indexed token, uint256 prevCap, uint256 cap);
+
     function test_DepositCaps() public {
         // Deploy a new mock token
         MockERC20 newMockToken = new MockERC20("New Test Token", "NTT", 18);
 
         // Set a cap for the new token
         uint256 capAmount = 1e18; // 1 token cap
         vm.prank(operator);
+        vm.expectEmit(true, true, true, true);
+        emit CapSet(address(newMockToken), 0, capAmount);
         pointTokenVault.setCap(address(newMockToken), capAmount);
 
         // Mint tokens to vitalik