test: do not revert when nothing to tip

morpho-org · Mar 14, 2024 · 5126bdb · 5126bdb
1 parent 03d629c
commit 5126bdb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/test/prod/aave-v2/TestLifecycle.t.sol b/test/prod/aave-v2/TestLifecycle.t.sol
@@ -235,7 +235,10 @@ contract TestLifecycle is TestSetup {
         _tip(
             borrow.market.underlying,
             address(user),
-            borrow.position.total - ERC20(borrow.market.underlying).balanceOf(address(user))
+            Math.zeroFloorSub(
+                borrow.position.total,
+                ERC20(borrow.market.underlying).balanceOf(address(user))
+            )
         );
         user.approve(borrow.market.underlying, borrow.position.total);
         user.repay(borrow.market.poolToken, address(user), type(uint256).max);

diff --git a/test/prod/compound/TestLifecycle.t.sol b/test/prod/compound/TestLifecycle.t.sol
@@ -286,7 +286,10 @@ contract TestLifecycle is TestSetup {
         _tip(
             borrow.market.underlying,
             address(user),
-            borrow.position.total - ERC20(borrow.market.underlying).balanceOf(address(user))
+            Math.zeroFloorSub(
+                borrow.position.total,
+                ERC20(borrow.market.underlying).balanceOf(address(user))
+            )
         );
         user.approve(borrow.market.underlying, borrow.position.total);
         user.repay(borrow.market.poolToken, address(user), type(uint256).max);