logos-co
/
staking
mirror of https://github.com/logos-co/staking.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore(StakeManager.t): move migration related tests to migration scope

This commit is contained in:
Ricardo Guilherme Schmidt 2024-03-04 11:53:44 -03:00 committed by r4bbit
parent acf5f41bc7
commit 6c4f5d16f2
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
test/StakeManager.t.sol
View File

@ -625,6 +625,20 @@ contract MigrationStakeManagerTest is StakeManagerTest {
assertEq(address(newStakeManager.oldManager()), address(stakeManager));
assertEq(newStakeManager.totalSupply(), 0);
}
function test_ExecuteEpochShouldNotIncreaseEpochInMigration() public {
assertEq(stakeManager.currentEpoch(), 0);
assertEq(address(stakeManager.migration()), address(0));
vm.prank(deployer);
stakeManager.startMigration(newStakeManager);
assertEq(address(stakeManager.migration()), address(newStakeManager));
vm.warp(stakeManager.epochEnd());
vm.expectRevert(StakeManager.StakeManager__PendingMigration.selector);
stakeManager.executeEpoch();
assertEq(stakeManager.currentEpoch(), 0);
}
}
contract ExecuteEpochTest is MigrationStakeManagerTest {
@ -636,22 +650,8 @@ contract ExecuteEpochTest is MigrationStakeManagerTest {
stakeManager.executeEpoch();
assertEq(stakeManager.currentEpoch(), 0);
}
function test_ExecuteEpochShouldNotIncreaseEpochInMigration() public {
assertEq(stakeManager.currentEpoch(), 0);
assertEq(address(stakeManager.migration()), address(0));
vm.prank(deployer);
stakeManager.startMigration(newStakeManager);
assertEq(address(stakeManager.migration()), address(newStakeManager));
vm.warp(stakeManager.epochEnd());
vm.expectRevert(StakeManager.StakeManager__PendingMigration.selector);
stakeManager.executeEpoch();
assertEq(stakeManager.currentEpoch(), 0);
}
//currentEpoch can only increase.
function test_ExecuteEpochShouldIncreaseEpoch() public {
assertEq(stakeManager.currentEpoch(), 0);