mirror of https://github.com/logos-co/staking.git
chore(StakeManager.t): move migration related tests to migration scope
This commit is contained in:
parent
acf5f41bc7
commit
6c4f5d16f2
|
@ -625,6 +625,20 @@ contract MigrationStakeManagerTest is StakeManagerTest {
|
||||||
assertEq(address(newStakeManager.oldManager()), address(stakeManager));
|
assertEq(address(newStakeManager.oldManager()), address(stakeManager));
|
||||||
assertEq(newStakeManager.totalSupply(), 0);
|
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 {
|
contract ExecuteEpochTest is MigrationStakeManagerTest {
|
||||||
|
@ -636,22 +650,8 @@ contract ExecuteEpochTest is MigrationStakeManagerTest {
|
||||||
stakeManager.executeEpoch();
|
stakeManager.executeEpoch();
|
||||||
assertEq(stakeManager.currentEpoch(), 0);
|
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.
|
//currentEpoch can only increase.
|
||||||
|
|
||||||
function test_ExecuteEpochShouldIncreaseEpoch() public {
|
function test_ExecuteEpochShouldIncreaseEpoch() public {
|
||||||
assertEq(stakeManager.currentEpoch(), 0);
|
assertEq(stakeManager.currentEpoch(), 0);
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue