fix(market): fix supply to utilization edge cases

morpho-org · Aug 2, 2024 · 7fc42e7 · 7fc42e7
1 parent b11ee66
commit 7fc42e7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/packages/blue-sdk/src/market/MarketUtils.test.ts b/packages/blue-sdk/src/market/MarketUtils.test.ts
@@ -36,6 +36,20 @@ describe("MarketUtils", () => {
         90_0000000000000000n,
       ),
     ).toEqual(0n);
+
+    expect(
+      MarketUtils.getSupplyToUtilization(
+        { totalSupplyAssets: MathLib.WAD, totalBorrowAssets: 0n },
+        0n,
+      ),
+    ).toEqual(0n);
+
+    expect(
+      MarketUtils.getSupplyToUtilization(
+        { totalSupplyAssets: MathLib.WAD, totalBorrowAssets: 1n },
+        0n,
+      ),
+    ).toEqual(MathLib.MAX_UINT_256);
   });
 
   it("should calculate the withdraw volume to reach utilization", () => {

diff --git a/packages/blue-sdk/src/market/MarketUtils.ts b/packages/blue-sdk/src/market/MarketUtils.ts
@@ -137,26 +137,22 @@ export namespace MarketUtils {
    * @param utilization The target utilization rate (scaled by WAD).
    */
   export function getSupplyToUtilization(
-    {
-      totalSupplyAssets,
-      totalBorrowAssets,
-    }: {
+    market: {
       totalSupplyAssets: BigIntish;
       totalBorrowAssets: BigIntish;
     },
     utilization: BigIntish,
   ) {
     utilization = BigInt(utilization);
-    totalBorrowAssets = BigInt(totalBorrowAssets);
     if (utilization === 0n) {
-      if (totalBorrowAssets === 0n) return totalSupplyAssets;
+      if (getUtilization(market) === 0n) return 0n;
 
-      return 0n;
+      return MathLib.MAX_UINT_256;
     }
 
     return MathLib.zeroFloorSub(
-      MathLib.wDivUp(totalBorrowAssets, utilization),
-      totalSupplyAssets,
+      MathLib.wDivUp(market.totalBorrowAssets, utilization),
+      market.totalSupplyAssets,
     );
   }
 
@@ -176,6 +172,7 @@ export namespace MarketUtils {
     utilization: BigIntish,
   ) {
     utilization = BigInt(utilization);
+    totalSupplyAssets = BigInt(totalSupplyAssets);
     totalBorrowAssets = BigInt(totalBorrowAssets);
     if (utilization === 0n) {
       if (totalBorrowAssets === 0n) return totalSupplyAssets;