✨ staking: make harvest and claim pausable

Author: itofarina

contracts/StakedEXA.sol

@@ -275,7 +275,7 @@ contract StakedEXA is
     return rawClaimable_ > claimedAmountProportion ? rawClaimable_ - claimedAmountProportion : 0;
   }
 
-  function harvest() public {
+  function harvest() public whenNotPaused {
     Market memMarket = market;
     address memProvider = provider;
     uint256 assets = Math.min(
@@ -298,7 +298,7 @@ contract StakedEXA is
     return rewardsTokens;
   }
 
-  function claim_(IERC20 reward) internal {
+  function claim_(IERC20 reward) internal whenNotPaused {
     uint256 time = block.timestamp * 1e18 - avgStart[msg.sender];
     if (time <= minTime * 1e18) return;
 

test/StakedEXA.t.sol

@@ -1634,6 +1634,46 @@ contract StakedEXATest is Test {
     stEXA.permitAndDeposit(assets, BOB, Permit(assets, block.timestamp, v, r, s));
   }
 
+  function testPausableClaim() external {
+    uint256 assets = 1_000e18;
+    exa.mint(address(this), assets);
+    stEXA.deposit(assets, address(this));
+
+    address pauser = address(0x1);
+    stEXA.grantRole(stEXA.PAUSER_ROLE(), pauser);
+
+    vm.prank(pauser);
+    stEXA.pause();
+
+    vm.expectRevert(Pausable.EnforcedPause.selector);
+    stEXA.claimAll();
+
+    vm.expectRevert(Pausable.EnforcedPause.selector);
+    stEXA.claim(providerAsset);
+
+    vm.prank(pauser);
+    stEXA.unpause();
+
+    stEXA.claimAll();
+    stEXA.claim(providerAsset);
+  }
+
+  function testPausableHarvest() external {
+    address pauser = address(0x1);
+    stEXA.grantRole(stEXA.PAUSER_ROLE(), pauser);
+
+    vm.prank(pauser);
+    stEXA.pause();
+
+    vm.expectRevert(Pausable.EnforcedPause.selector);
+    stEXA.harvest();
+
+    vm.prank(pauser);
+    stEXA.unpause();
+
+    stEXA.harvest();
+  }
+
   function minMaxWithdrawAllowance() internal view returns (uint256) {
     return Math.min(market.convertToAssets(market.allowance(PROVIDER, address(stEXA))), market.maxWithdraw(PROVIDER));
   }