fix: cannot borrow reserves

morpho-org · Jan 29, 2024 · de77d30 · de77d30
1 parent e46bbba
commit de77d30
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/prod/compound/setup/TestSetup.sol b/test/prod/compound/setup/TestSetup.sol
@@ -191,6 +191,7 @@ contract TestSetup is Config, ProdTest {
         uint96 _amount,
         uint256 _price
     ) internal view returns (uint256) {
+        ICToken poolToken = ICToken(_market.poolToken);
         return
             bound(
                 _amount,
@@ -199,9 +200,7 @@ contract TestSetup is Config, ProdTest {
                     Math.min(
                         Math.min(
                             (_market.maxBorrows - _market.totalBorrows) / 2,
-                            _market.underlying == wEth
-                                ? cEth.balance
-                                : ERC20(_market.underlying).balanceOf(_market.poolToken)
+                            poolToken.getCash() - poolToken.totalReserves()
                         ),
                         MAX_USD_AMOUNT.div(_price)
                     ),