Merge pull request #247 from euler-xyz/fix-invariants-tests

fix invariant_totalAssetsShouldBeEqualToBalanceAfterSMEAR
euler-xyz · May 27, 2024 · ba6e4fb · ba6e4fb
2 parents 6215d49 + 663b9f5
commit ba6e4fb
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/test/unit/esr/ESR.Fuzz.t.sol b/test/unit/esr/ESR.Fuzz.t.sol
@@ -12,13 +12,14 @@ contract ESRFuzzTest is ESRTest {
         assertGe(esrSlot.interestLeft, accruedInterest);
     }
 
-    // totalAssets should be equal to the balance after SMEAR has passed
-    // function invariant_totalAssetsShouldBeEqualToBalanceAfterSMEAR() public {
-    //     vm.assume(asset.balanceOf(address(esr)) <= type(uint168).max);
-    //     esr.gulp();
-    //     skip(esr.INTEREST_SMEAR()); // make sure smear has passed
-    //     assertEq(esr.totalAssets(), asset.balanceOf(address(esr)));
-    // }
+    //totalAssets should be equal to the balance after SMEAR has passed
+    function invariant_totalAssetsShouldBeEqualToBalanceAfterSMEAR() public {
+        vm.assume(asset.balanceOf(address(esr)) <= type(uint168).max);
+        if (asset.balanceOf(address(esr)) == 0) return;
+        esr.gulp();
+        skip(esr.INTEREST_SMEAR()); // make sure smear has passed
+        assertEq(esr.totalAssets(), asset.balanceOf(address(esr)));
+    }
 
     function testFuzz_interestAccrued_under_uint168(uint256 interestAmount, uint256 depositAmount, uint256 timePassed)
         public