test: basic NAYM reward compounding

nayms · Dec 3, 2024 · a7dcd65 · a7dcd65
1 parent a410f8d
commit a7dcd65
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/src/libs/LibTokenizedVaultStaking.sol b/src/libs/LibTokenizedVaultStaking.sol
@@ -325,10 +325,13 @@ library LibTokenizedVaultStaking {
 
         uint256 rewardAmount;
         uint256 rewardCount = rewards.currencies.length;
-        for (uint64 i = 0; i <= rewardCount; i++) {
-            if (rewards.currencies[i] == tokenId) {
-                rewardAmount = rewards.amounts[i];
-                break;
+
+        if (rewardCount > 0) {
+            for (uint64 i = 0; i <= rewardCount; i++) {
+                if (rewards.currencies[i] == tokenId) {
+                    rewardAmount = rewards.amounts[i];
+                    break;
+                }
             }
         }
 

diff --git a/test/T06Staking.t.sol b/test/T06Staking.t.sol
@@ -776,6 +776,9 @@ contract T06Staking is D03ProtocolDefaults {
         assertStakedAmount(bob.entityId, bobStakeAmount, "Bob's stake should increase");
         c.log(" ~ [%s] Bob staked".blue(), currentInterval());
 
+        vm.expectRevert("No reward to compound");
+        nayms.compoundRewards(nlf.entityId);
+
         vm.warp(startStaking + 61 days);
 
         assertEq(nayms.lastPaidInterval(nlf.entityId), 0, "Last interval paid should be 0");
@@ -788,12 +791,10 @@ contract T06Staking is D03ProtocolDefaults {
 
         vm.warp(startStaking + 181 days);
 
-        uint256 balanceBeforeClaim = nayms.internalBalanceOf(bob.entityId, NAYM_ID);
-
         startPrank(bob);
-        // (bytes32[] memory currencies, uint256[] memory amounts) = nayms.getRewardsBalance(bob.entityId, nlf.entityId);
-        nayms.collectRewards(nlf.entityId);
-        assertEq(nayms.internalBalanceOf(bob.entityId, NAYM_ID), balanceBeforeClaim + 10_000, "Bob should have NAYM in his balance");
+        nayms.compoundRewards(nlf.entityId);
+        assertStakedAmount(bob.entityId, bobStakeAmount + 10_000, "Bob's stake should increase");
+        c.log(" ~ [%s] Reward1 compound".blue(), currentInterval());
     }
 
     function test_twoStakingRewardCurrencies() public {