Fix two unit tests

LiskHQ · Jun 20, 2024 · 6e5db30 · 6e5db30
1 parent ec00a29
commit 6e5db30
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 12 deletions.
diff --git a/test/L2/L2Staking.t.sol b/test/L2/L2Staking.t.sol
@@ -1683,6 +1683,12 @@ contract L2StakingTest is Test {
         vm.prank(alice);
         l2Staking.lockAmount(alice, 50 * 10 ** 18, 365);
 
+        // create a paused stake to have it for testing resumeCountdown function call inside Stake contract
+        vm.startPrank(alice);
+        l2Staking.lockAmount(alice, 20 * 10 ** 18, 365);
+        l2Staking.pauseRemainingLockingDuration(2);
+        vm.stopPrank();
+
         upgradeLockingPositionContractToPausedVersion();
 
         // try to call lockAmount
@@ -1716,5 +1722,10 @@ contract L2StakingTest is Test {
         vm.expectRevert(L2LockingPositionPaused.LockingPositionIsPaused.selector);
         vm.prank(alice);
         l2Staking.pauseRemainingLockingDuration(1);
+
+        // try to call resumeCountdown
+        vm.expectRevert(L2LockingPositionPaused.LockingPositionIsPaused.selector);
+        vm.prank(alice);
+        l2Staking.resumeCountdown(2);
     }
 }
diff --git a/test/L2/paused/L2RewardPaused.t.sol b/test/L2/paused/L2RewardPaused.t.sol
@@ -55,16 +55,6 @@ contract L2RewardPausedTest is Test {
         uint256 duration;
     }
 
-    function upgradeLockingPositionContractToPausedVersion() private {
-        // deploy L2LockingPositionPaused contract
-        L2LockingPositionPaused l2LockingPositionPaused = new L2LockingPositionPaused();
-
-        // upgrade L2LockingPosition contract to L2LockingPositionPaused contract
-        l2LockingPosition.upgradeToAndCall(
-            address(l2LockingPositionPaused), abi.encodeWithSelector(l2LockingPositionPaused.initializePaused.selector)
-        );
-    }
-
     function convertLiskToSmallestDenomination(uint256 lisk) internal pure returns (uint256) {
         return lisk * 10 ** 18;
     }
@@ -161,8 +151,6 @@ contract L2RewardPausedTest is Test {
         stakerPositions = new uint256[](1);
         stakerPositions[0] = ID;
 
-        upgradeLockingPositionContractToPausedVersion();
-
         // deploy L2RewardPaused contract
         L2RewardPaused l2RewardPaused = new L2RewardPaused();