Adds more complex assertions to scenario4 regarding surplus verification

LiskHQ · Jun 4, 2024 · 7910f30 · 7910f30
1 parent 443fd53
commit 7910f30
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/test/L2/L2Reward.t.sol b/test/L2/L2Reward.t.sol
@@ -348,12 +348,17 @@ contract L2RewardTest is Test {
     {
         uint256 rewardsCap = totalAmountLocked / 365;
         uint256 expectedSurplus;
+        uint256 lastDay = deploymentDate + endDay;
 
-        for (uint256 d = deploymentDate + startDay; d <= deploymentDate + endDay; d++) {
+        for (uint256 d = deploymentDate + startDay; d <= lastDay; d++) {
             expectedSurplus += dailyRewards - rewardsCap;
         }
 
+        // console2.logUint(expectedSurplus);
         scenario.rewardsSurplus += expectedSurplus;
+        // console2.log(l2Reward.rewardsSurplus());
+        // console2.log(scenario.rewardsSurplus);
+
         assertEq(scenario.rewardsSurplus, l2Reward.rewardsSurplus());
     }
 
@@ -705,12 +710,23 @@ contract L2RewardTest is Test {
         verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(700), 30, 34, scenario);
         onDay(40);
         allStakersClaim(scenario);
-        verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(710), 35, 39, scenario);
         checkRewardsContractBalance(funds);
         onDay(50);
         allStakersClaim(scenario);
         checkRewardsContractBalance(funds);
-        verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(710), 40, 49, scenario);
+        onDay(90);
+        allStakersClaim(scenario);
+        checkRewardsContractBalance(funds);
+        verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(710), 35, 89, scenario);
+        onDay(130);
+        allStakersClaim(scenario);
+        checkRewardsContractBalance(funds);
+        verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(710), 90, 129, scenario);
+        onDay(150);
+        allStakersClaim(scenario);
+        checkRewardsContractBalance(funds);
+        // stake 3's expDate is on 130th day so total amount locked reduces by LSK(50).
+        verifyRewardsSurplusForLockedAmountBetweenDays(dailyRewards, LSK(660), 130, 149, scenario);
     }
 
     function checkConsistencyPendingUnlockDailyUnlocked(