add --ir-minimum to forge coverage

blockful-io · Jun 24, 2024 · ef57bed · ef57bed
1 parent 820fa49
commit ef57bed
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -59,7 +59,7 @@ jobs:
                   )" >> $GITHUB_ENV
 
             - name: Run coverage
-              run: forge coverage --report summary --report lcov
+              run: forge coverage --report summary --report lcov --ir-minimum
 
             # To ignore coverage for certain directories modify the paths in this step as needed. The
             # below default ignores coverage results for the test and script directories. Alternatively,

diff --git a/test/Staking.t.sol b/test/Staking.t.sol
@@ -1073,4 +1073,38 @@ contract Unstake is StakingTest {
         vm.expectRevert(Staking.StakeDoesNotBelongToKeyper.selector);
         staking.unstake(_depositor2, stakeId, 0);
     }
+
+    function testFuzz_RevertIf_AmountGreaterThanStakeAmount(
+        address _depositor,
+        uint256 _amount
+    ) public {
+        _amount = _boundToRealisticStake(_amount);
+
+        _mintGovToken(_depositor, _amount);
+        _setKeyper(_depositor, true);
+
+        uint256 stakeIndex = _stake(_depositor, _amount);
+
+        vm.prank(_depositor);
+        vm.expectRevert(Staking.WithdrawAmountTooHigh.selector);
+        staking.unstake(_depositor, stakeIndex, _amount + 1);
+    }
+
+    function testFuzz_RevertIf_NonKeyperTryToUnstake(
+        address _depositor,
+        address _anyone,
+        uint256 _amount
+    ) public {
+        _amount = _boundToRealisticStake(_amount);
+
+        _mintGovToken(_depositor, _amount);
+
+        _setKeyper(_depositor, true);
+
+        uint256 stakeId = _stake(_depositor, _amount);
+
+        vm.prank(_anyone);
+        vm.expectRevert(Staking.OnlyKeyper.selector);
+        staking.unstake(_depositor, stakeId, 0);
+    }
 }