From 6c4f5d16f2b4a197077a898fd15fb725f2242554 Mon Sep 17 00:00:00 2001
From: Ricardo Guilherme Schmidt <3esmit@gmail.com>
Date: Mon, 4 Mar 2024 11:53:44 -0300
Subject: [PATCH] chore(StakeManager.t): move migration related tests to
 migration scope

---
 test/StakeManager.t.sol | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/test/StakeManager.t.sol b/test/StakeManager.t.sol
index f729902..03f9d84 100644
--- a/test/StakeManager.t.sol
+++ b/test/StakeManager.t.sol
@@ -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);