test(strategy): Update assertion in test_collectInterest() for all st…

…rategy tests
Sperax · Jan 19, 2024 · 64b48d8 · 64b48d8
1 parent d0a6904
commit 64b48d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/test/strategy/CompoundStrategy.t.sol b/test/strategy/CompoundStrategy.t.sol
@@ -257,9 +257,7 @@ contract CollectInterestTest is CompoundStrategyTest {
         strategy.collectInterest(ASSET);
 
         uint256 current_bal = IERC20(ASSET).balanceOf(yieldReceiver);
-        uint256 newInterestEarned = strategy.checkInterestEarned(ASSET);
-
-        assertEq(newInterestEarned, 0);
+        assertApproxEqAbs(strategy.checkInterestEarned(ASSET), 0, 1);
         assertEq(current_bal, (initial_bal + harvestAmount));
     }
 }

diff --git a/test/strategy/StargateStrategy.t.sol b/test/strategy/StargateStrategy.t.sol
@@ -487,6 +487,7 @@ contract Test_CollectInterest is Test_Harvest {
             vm.recordLogs();
 
             strategy.collectInterest(assetData[i].asset);
+            assertApproxEqAbs(strategy.checkInterestEarned(assetData[i].asset), 0, 1);
 
             VmSafe.Log[] memory logs = vm.getRecordedLogs();
 
@@ -505,7 +506,6 @@ contract Test_CollectInterest is Test_Harvest {
 
             /// @note precision Error from stargate
             assertApproxEqAbs(strategy.checkLPTokenBalance(assetData[i].asset), initialLPBal, 1);
-            assertEq(strategy.checkInterestEarned(assetData[i].asset), 0);
         }
     }