assertGt > assertGe and improve fuzzing

blockful-io · Jun 23, 2024 · 5773044 · 5773044
1 parent 7e703c4
commit 5773044
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/Staking.t.sol b/test/Staking.t.sol
@@ -351,7 +351,7 @@ contract Stake is StakingTest {
 
         _stake(_depositor2, _amount);
 
-        assertGt(
+        assertGe(
             staking.balanceOf(_depositor1),
             staking.balanceOf(_depositor2),
             "Wrong balance"
@@ -918,6 +918,11 @@ contract Unstake is StakingTest {
         _amount = _boundToRealisticStake(_amount);
         _jump = _boundRealisticTimeAhead(_jump);
 
+        vm.assume(
+            _anyone != address(0) &&
+                _anyone != ProxyUtils.getAdminAddress(address(staking))
+        );
+
         _mintGovToken(_depositor, _amount);
         _setKeyper(_depositor, true);