@indigo-labs/indigo-sdk 0.3.10 → 0.3.12

package/tests/queries/treasury-queries.ts

@@ -51,6 +51,7 @@ export async function findRandomTreasuryUtxoWithAsset(
   lucid: LucidEvolution,
   sysParams: SystemParams,
   asset: AssetClass,
+  max_number_of_assets: bigint = 2n,
 ): Promise<UTxO> {
   // We need to consider both with staking credential and without.
   const treasuryUtxos = await lucid.utxosAtWithUnit(
@@ -58,8 +59,12 @@ export async function findRandomTreasuryUtxoWithAsset(
     assetClassToUnit(asset),
   );
 
+  const validTreasuryUtxos = treasuryUtxos.filter(
+    (utxo) => Object.keys(utxo.assets).length <= max_number_of_assets,
+  );
+
   return F.pipe(
-    O.fromNullable(getRandomElement(treasuryUtxos)),
+    O.fromNullable(getRandomElement(validTreasuryUtxos)),
     O.match(() => {
       throw new Error('Expected some treasury UTXOs.');
     }, F.identity),

package/tests/stability-pool.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, expect, test } from 'vitest';
+import { assert, beforeEach, expect, test } from 'vitest';
 import {
   LucidContext,
   repeat,
@@ -757,7 +757,8 @@ describe('Stability pool', () => {
       // Send funds to user4 so liquidation is performed from a clean address.
       await sendValueTo(
         context.users.user4.address,
-        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL),
+        // The extra collateral asset is to be sent to the treasury.
+        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL + 1_000_000n),
         context.lucid,
       );
 
@@ -1130,7 +1131,8 @@ describe('Stability pool', () => {
       // Send funds to user4 so liquidation is performed from a clean address.
       await sendValueTo(
         context.users.user4.address,
-        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL),
+        // The extra collateral asset is to be sent to the treasury.
+        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL + 1_000_000n),
         context.lucid,
       );
 
@@ -1784,7 +1786,8 @@ describe('Stability pool', () => {
       // Send funds to user4 so liquidation is performed from a clean address.
       await sendValueTo(
         context.users.user4.address,
-        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL),
+        // The extra collateral asset is to be sent to the treasury.
+        mkAssetsOf(collateralAssetA, LIQUIDATED_COLLATERAL + 1_000_000n),
         context.lucid,
       );
 
@@ -1920,8 +1923,9 @@ describe('Stability pool', () => {
     });
   });
 
+  // These are the most significant tests for benchmarking the maximum number of collateral assets supported.
   describe('Liquidate', () => {
-    test<MyContext>('Liquidate adding 9th reward asset', async (context: MyContext) => {
+    test<MyContext>('Liquidate with 10 reward assets in the pool - collecting with 2 assets in treasury', async (context: MyContext) => {
       context.lucid.selectWallet.fromSeed(context.users.admin.seedPhrase);
 
       const initialAssets = [
@@ -1934,9 +1938,12 @@ describe('Stability pool', () => {
         collateralAssetF,
         collateralAssetG,
         collateralAssetH,
+        collateralAssetI,
       ];
 
-      expect(initialAssets.length, 'Unexpected reward assets count').toEqual(9);
+      expect(initialAssets.length, 'Unexpected reward assets count').toEqual(
+        10,
+      );
 
       const [sysParams, [iusdAssetInfo]] = await init(
         context.lucid,
@@ -1991,7 +1998,204 @@ describe('Stability pool', () => {
           context.lucid,
           requestSpAccountCreation(
             iusdAssetInfo.iassetTokenNameAscii,
-            BigInt(initialAssets.length + 1) * LIQUIDATED_DEBT,
+            BigInt(initialAssets.length + 2) * LIQUIDATED_DEBT,
+            sysParams,
+            context.lucid,
+          ),
+        );
+
+        context.lucid.selectWallet.fromSeed(context.users.admin.seedPhrase);
+
+        await runAndAwaitTx(
+          context.lucid,
+          runProcessSpRequest(
+            context,
+            sysParams,
+            iusdAssetInfo.iassetTokenNameAscii,
+            paymentCredentialOf(context.users.user2.address).hash,
+          ),
+        );
+      }
+
+      /**********
+       * Add each collateral asset as reward to stability pool at epoch 0 and scale 0
+       ***********/
+      for (let idx = 0; idx < initialAssets.length; idx++) {
+        const collateral = initialAssets[idx];
+
+        context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
+
+        // Send funds to user4 so liquidation is performed from a clean address.
+        await sendValueTo(
+          context.users.user4.address,
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
+          context.lucid,
+        );
+
+        context.emulator.awaitSlot(1000);
+
+        context.lucid.selectWallet.fromSeed(context.users.user4.seedPhrase);
+
+        await refreshPriceOracle(context, sysParams, iusdAssetInfo, collateral);
+
+        await executeLiquidation(
+          context,
+          sysParams,
+          LIQUIDATED_DEBT,
+          LIQUIDATED_COLLATERAL,
+          collateral,
+          iusdAssetInfo,
+        );
+      }
+
+      /**********
+       * Execute last liquidation
+       ***********/
+
+      {
+        {
+          const sp = await findStabilityPool(
+            context.lucid,
+            sysParams,
+            iusdAssetInfo.iassetTokenNameAscii,
+          );
+
+          expect(sp.datum.state.epoch, 'Unexpected epoch').toEqual(0n);
+          expect(sp.datum.state.scale, 'Unexpected scale').toEqual(0n);
+          expect(
+            sp.datum.assetStates.length,
+            'Unexpected asset states',
+          ).toEqual(initialAssets.length);
+        }
+
+        const collateral = initialAssets[initialAssets.length - 1];
+
+        // Make sure that the liquidation will be performed using a treasury UTxO with 2 assets.
+        assert(collateral !== adaAssetClass);
+
+        context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
+
+        // Send funds to user4 so liquidation is performed from a clean address.
+        await sendValueTo(
+          context.users.user4.address,
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
+          context.lucid,
+        );
+
+        context.emulator.awaitSlot(1000);
+
+        context.lucid.selectWallet.fromSeed(context.users.user4.seedPhrase);
+
+        await refreshPriceOracle(context, sysParams, iusdAssetInfo, collateral);
+
+        await executeLiquidation(
+          context,
+          sysParams,
+          LIQUIDATED_DEBT,
+          LIQUIDATED_COLLATERAL,
+          collateral,
+          iusdAssetInfo,
+          (tx) =>
+            benchmarkAndAwaitTx(
+              'Stability Pool - Liquidate with 10 reward assets in the pool - collecting with 2 assets in treasury',
+              tx,
+              context.lucid,
+              context.emulator,
+            ),
+        );
+      }
+
+      {
+        const sp = await findStabilityPool(
+          context.lucid,
+          sysParams,
+          iusdAssetInfo.iassetTokenNameAscii,
+        );
+
+        expect(sp.datum.state.epoch, 'Unexpected epoch').toEqual(0n);
+        expect(sp.datum.state.scale, 'Unexpected scale').toEqual(0n);
+        expect(sp.datum.assetStates.length, 'Unexpected asset states').toEqual(
+          initialAssets.length,
+        );
+      }
+    });
+
+    test<MyContext>('Liquidate with 10 reward assets in the pool - direct treasury payment', async (context: MyContext) => {
+      context.lucid.selectWallet.fromSeed(context.users.admin.seedPhrase);
+
+      const initialAssets = [
+        adaAssetClass,
+        collateralAssetA,
+        collateralAssetB,
+        collateralAssetC,
+        collateralAssetD,
+        collateralAssetE,
+        collateralAssetF,
+        collateralAssetG,
+        collateralAssetH,
+        collateralAssetI,
+      ];
+
+      expect(initialAssets.length, 'Unexpected reward assets count').toEqual(
+        10,
+      );
+
+      const [sysParams, [iusdAssetInfo]] = await init(
+        context.lucid,
+        [
+          {
+            ...iusdInitialAssetCfg(),
+            collateralAssets: mkCollateralAssetsChain(
+              initialAssets.map((a) => mkBaseCollateralAsset(a, 0n)),
+            ),
+          },
+        ],
+        context.emulator.slot,
+      );
+
+      await createMultipleUtxosAtTreasury(
+        mkLovelacesOf(2n),
+        16n,
+        sysParams,
+        context,
+      );
+
+      // After price doubles the collateral ratio will be 110%
+      const LIQUIDATED_DEBT = 5_000_000n;
+      const LIQUIDATED_COLLATERAL = 11_000_000n;
+
+      /**********
+       * Create account for user2
+       ***********/
+      {
+        context.lucid.selectWallet.fromSeed(context.users.user2.seedPhrase);
+
+        await refreshPriceOracle(
+          context,
+          sysParams,
+          iusdAssetInfo,
+          collateralAssetA,
+        );
+
+        await runAndAwaitTx(
+          context.lucid,
+          runOpenCdp(
+            context,
+            sysParams,
+            iusdAssetInfo.iassetTokenNameAscii,
+            collateralAssetA,
+            10_000_000_000n,
+            3_000_000_000n,
+          ),
+        );
+
+        await runAndAwaitTx(
+          context.lucid,
+          requestSpAccountCreation(
+            iusdAssetInfo.iassetTokenNameAscii,
+            BigInt(initialAssets.length + 2) * LIQUIDATED_DEBT,
             sysParams,
             context.lucid,
           ),
@@ -2013,7 +2217,7 @@ describe('Stability pool', () => {
       /**********
        * Add each collateral asset as reward to stability pool at epoch 0 and scale 0
        ***********/
-      for (let idx = 0; idx < initialAssets.length - 1; idx++) {
+      for (let idx = 0; idx < initialAssets.length; idx++) {
         const collateral = initialAssets[idx];
 
         context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
@@ -2038,6 +2242,8 @@ describe('Stability pool', () => {
           LIQUIDATED_COLLATERAL,
           collateral,
           iusdAssetInfo,
+          undefined,
+          true,
         );
       }
 
@@ -2058,7 +2264,7 @@ describe('Stability pool', () => {
           expect(
             sp.datum.assetStates.length,
             'Unexpected asset states',
-          ).toEqual(initialAssets.length - 1);
+          ).toEqual(initialAssets.length);
         }
 
         const collateral = initialAssets[initialAssets.length - 1];
@@ -2087,11 +2293,12 @@ describe('Stability pool', () => {
           iusdAssetInfo,
           (tx) =>
             benchmarkAndAwaitTx(
-              'Stability Pool - Liquidate adding 9th reward asset',
+              'Stability Pool - Liquidate with 10 reward assets in the pool - direct treasury payment',
               tx,
               context.lucid,
               context.emulator,
             ),
+          true,
         );
       }
 
@@ -2110,7 +2317,7 @@ describe('Stability pool', () => {
       }
     });
 
-    test<MyContext>('Liquidate adding 9th reward asset and increasing epoch', async (context: MyContext) => {
+    test<MyContext>('Liquidate with 9 reward assets in the pool and increasing epoch', async (context: MyContext) => {
       context.lucid.selectWallet.fromSeed(context.users.admin.seedPhrase);
 
       const initialAssets = [
@@ -2180,7 +2387,7 @@ describe('Stability pool', () => {
           context.lucid,
           requestSpAccountCreation(
             iusdAssetInfo.iassetTokenNameAscii,
-            BigInt(initialAssets.length) * LIQUIDATED_DEBT,
+            BigInt(initialAssets.length + 1) * LIQUIDATED_DEBT,
             sysParams,
             context.lucid,
           ),
@@ -2202,7 +2409,7 @@ describe('Stability pool', () => {
       /**********
        * Add each collateral asset as reward to stability pool at epoch 0 and scale 0
        ***********/
-      for (let idx = 0; idx < initialAssets.length - 1; idx++) {
+      for (let idx = 0; idx < initialAssets.length; idx++) {
         const collateral = initialAssets[idx];
 
         context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
@@ -2210,7 +2417,8 @@ describe('Stability pool', () => {
         // Send funds to user4 so liquidation is performed from a clean address.
         await sendValueTo(
           context.users.user4.address,
-          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL),
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
           context.lucid,
         );
 
@@ -2245,17 +2453,21 @@ describe('Stability pool', () => {
           expect(
             sp.datum.assetStates.length,
             'Unexpected asset states',
-          ).toEqual(initialAssets.length - 1);
+          ).toEqual(initialAssets.length);
         }
 
         const collateral = initialAssets[initialAssets.length - 1];
 
+        // Make sure that the liquidation will be performed using a treasury UTxO with 2 assets.
+        assert(collateral !== adaAssetClass);
+
         context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
 
         // Send funds to user4 so liquidation is performed from a clean address.
         await sendValueTo(
           context.users.user4.address,
-          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL),
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
           context.lucid,
         );
 
@@ -2272,7 +2484,7 @@ describe('Stability pool', () => {
           iusdAssetInfo,
           async (tx) =>
             benchmarkAndAwaitTx(
-              'Stability Pool - Liquidate adding 9th reward asset and increasing epoch',
+              'Stability Pool - Liquidate with 9 reward assets in the pool and increasing epoch',
               tx,
               context.lucid,
               context.emulator,
@@ -2294,7 +2506,9 @@ describe('Stability pool', () => {
       }
     });
 
-    test<MyContext>('Liquidate adding 9th reward asset and increasing scale', async (context: MyContext) => {
+    // TODO: This test shows abnormally low memory and CPU use compared with the other similar tests
+    // (e.g. the test above increasing epoch) or this very test performed with one less asset.
+    test<MyContext>('Liquidate with 10 reward assets in the pool and increasing scale', async (context: MyContext) => {
       context.lucid.selectWallet.fromSeed(context.users.admin.seedPhrase);
 
       const initialAssets = [
@@ -2307,9 +2521,12 @@ describe('Stability pool', () => {
         collateralAssetF,
         collateralAssetG,
         collateralAssetH,
+        collateralAssetI,
       ];
 
-      expect(initialAssets.length, 'Unexpected reward assets count').toEqual(9);
+      expect(initialAssets.length, 'Unexpected reward assets count').toEqual(
+        10,
+      );
 
       const [sysParams, [iusdAssetInfo]] = await init(
         context.lucid,
@@ -2364,7 +2581,7 @@ describe('Stability pool', () => {
           context.lucid,
           requestSpAccountCreation(
             iusdAssetInfo.iassetTokenNameAscii,
-            BigInt(initialAssets.length - 1) * LIQUIDATED_DEBT +
+            BigInt(initialAssets.length) * LIQUIDATED_DEBT +
               LIQUIDATED_DEBT +
               10n,
             sysParams,
@@ -2388,7 +2605,7 @@ describe('Stability pool', () => {
       /**********
        * Add each collateral asset as reward to stability pool at epoch 0 and scale 0
        ***********/
-      for (let idx = 0; idx < initialAssets.length - 1; idx++) {
+      for (let idx = 0; idx < initialAssets.length; idx++) {
         const collateral = initialAssets[idx];
 
         context.lucid.selectWallet.fromSeed(context.users.user3.seedPhrase);
@@ -2396,7 +2613,8 @@ describe('Stability pool', () => {
         // Send funds to user4 so liquidation is performed from a clean address.
         await sendValueTo(
           context.users.user4.address,
-          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL),
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
           context.lucid,
         );
 
@@ -2421,6 +2639,13 @@ describe('Stability pool', () => {
       await repeat(1, async () => {
         context.lucid.selectWallet.fromSeed(context.users.user4.seedPhrase);
 
+        await refreshPriceOracle(
+          context,
+          sysParams,
+          iusdAssetInfo,
+          adaAssetClass,
+        );
+
         await executeLiquidation(
           context,
           sysParams,
@@ -2472,7 +2697,7 @@ describe('Stability pool', () => {
           expect(
             sp.datum.assetStates.length,
             'Unexpected asset states',
-          ).toEqual(initialAssets.length - 1);
+          ).toEqual(initialAssets.length);
         }
 
         const collateral = initialAssets[initialAssets.length - 1];
@@ -2482,7 +2707,8 @@ describe('Stability pool', () => {
         // Send funds to user4 so liquidation is performed from a clean address.
         await sendValueTo(
           context.users.user4.address,
-          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL),
+          // The extra collateral asset is to be sent to the treasury.
+          mkAssetsOf(collateral, LIQUIDATED_COLLATERAL + 1_000_000n),
           context.lucid,
         );
 
@@ -2499,7 +2725,7 @@ describe('Stability pool', () => {
           iusdAssetInfo,
           (tx) =>
             benchmarkAndAwaitTx(
-              'Stability Pool - Liquidate adding 9th reward asset and increasing scale',
+              'Stability Pool - Liquidate with 10 reward assets in the pool and increasing scale',
               tx,
               context.lucid,
               context.emulator,