chg

maticnetwork · Jan 17, 2022 · 2dd5fe5 · 2dd5fe5
1 parent fc58f80
commit 2dd5fe5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/contracts/staking/stakeManager/StakeManager.sol b/contracts/staking/stakeManager/StakeManager.sol
@@ -535,9 +535,14 @@ contract StakeManager is
             require(delegationEnabled, "Delegation is disabled");
         }
 
-        if (validators[validatorId].deactivationEpoch == 0) { // modify timeline only if validator didn't unstake
+        uint256 deactivationEpoch = validators[validatorId].deactivationEpoch;
+
+        if (deactivationEpoch == 0) { // modify timeline only if validator didn't unstake
             updateTimeline(amount, 0, 0);
+        } else if (deactivationEpoch > currentEpoch) { // validator just unstaked, need to wait till next checkpoint
+            revert("unstaking");
         }
+
 
         if (amount >= 0) {
             increaseValidatorDelegatedAmount(validatorId, uint256(amount));

diff --git a/test/units/staking/stakeManager/StakeManager.test.js b/test/units/staking/stakeManager/StakeManager.test.js
@@ -2482,7 +2482,9 @@ contract('StakeManager', async function(accounts) {
           totalStakeAfterUnstake = await this.stakeManager.currentValidatorSetTotalStake()
         })
 
-        testMigration()
+        it('reverts', async function() {
+          await expectRevert(this.stakeManager.migrateDelegation(aliceId, bobId, delegationAmount, { from: delegator }), 'unstaking');
+        })
       })
     })