@indigo-labs/indigo-sdk 0.3.10 → 0.3.11

package/tests/cdp/actions.ts

@@ -6,6 +6,9 @@ import {
   UTxO,
   toText,
   fromText,
+  Assets,
+  addAssets,
+  fromHex,
 } from '@lucid-evolution/lucid';
 import {
   addrDetails,
@@ -16,6 +19,7 @@ import {
   fromSystemParamsAsset,
   liquidateCdp,
   mintCdp,
+  mkTreasuryAddr,
   openCdp,
   redeemCdp,
   SystemParams,
@@ -36,13 +40,16 @@ import {
   runAndAwaitTxBuilder,
 } from '../test-helpers';
 import { AssetInfo } from '../endpoints/initialize';
-import { expect } from 'vitest';
+import { assert, expect } from 'vitest';
 import { feedPriceOracleTx } from '../../src/contracts/price-oracle/transactions';
 import {
+  adaAssetClass,
   AssetClass,
   assetClassToUnit,
+  assetClassValueOf,
   getInlineDatumOrThrow,
   matchSingle,
+  mkAssetsOf,
 } from '@3rd-eye-labs/cardano-offchain-common';
 import { parseCdpDatumOrThrow } from '../../src/contracts/cdp/types-new';
 import { match, P } from 'ts-pattern';
@@ -50,6 +57,9 @@ import { findPriceOracle } from '../price-oracle/price-oracle-queries';
 import { findCollateralAsset } from '../queries/iasset-queries';
 import { runFeedPriceToOracle } from '../price-oracle/actions';
 import { rationalFromInt, rationalMul } from '../../src/types/rational';
+import { array as A } from 'fp-ts';
+import { sendValueTo } from '../utils';
+import { findRandomTreasuryUtxoWithAsset } from '../queries/treasury-queries';
 
 // Selects users wallet and opens a CDP with the given initial collateral and mint amount
 export async function runOpenCdp(
@@ -60,6 +70,7 @@ export async function runOpenCdp(
   initialCollateral: bigint,
   initialMint: bigint,
   pythMessage?: string,
+  directTreasuryPayment: boolean = false,
 ): Promise<TxBuilder> {
   const orefs = await findAllNecessaryOrefs(
     context.lucid,
@@ -90,7 +101,7 @@ export async function runOpenCdp(
     orefs.collateralAsset.utxo,
     priceOracleUtxo,
     orefs.interestOracleUtxo,
-    orefs.treasuryUtxo,
+    directTreasuryPayment ? undefined : orefs.treasuryUtxo,
     context.lucid,
     context.emulator.slot,
     pythMessage,
@@ -196,6 +207,7 @@ export async function runMintCdp(
   collateralAsset: AssetClass,
   amount: bigint = 100_000n,
   pythMessage?: string,
+  directTreasuryPayment: boolean = false,
 ): Promise<TxBuilder> {
   const [pkh, skh] = await addrDetails(context.lucid);
 
@@ -233,7 +245,7 @@ export async function runMintCdp(
     orefs.collateralAsset.utxo,
     priceOracleUtxo,
     orefs.interestOracleUtxo,
-    orefs.treasuryUtxo,
+    directTreasuryPayment ? undefined : orefs.treasuryUtxo,
     orefs.interestCollectorUtxo,
     sysParams,
     context.lucid,
@@ -320,6 +332,7 @@ export async function runRedeemCdp(
   collateralAsset: AssetClass,
   pkh: string,
   skh: Credential | undefined,
+  directTreasuryPayment: boolean = false,
 ): Promise<TxBuilder> {
   const cdp = await findCdp(
     context.lucid,
@@ -350,7 +363,7 @@ export async function runRedeemCdp(
     priceOracleUtxo,
     orefs.interestOracleUtxo,
     orefs.interestCollectorUtxo,
-    orefs.treasuryUtxo,
+    directTreasuryPayment ? undefined : orefs.treasuryUtxo,
     orefs.govUtxo,
     sysParams,
     context.lucid,
@@ -513,6 +526,8 @@ export async function runLiquidateCdp(
   context: LucidContext,
   sysParams: SystemParams,
   frozenCdpUtxo: UTxO,
+  treasuryUtxo?: UTxO,
+  directTreasuryPayment: boolean = false,
 ): Promise<TxBuilder> {
   const cdpDatum = parseCdpDatumOrThrow(getInlineDatumOrThrow(frozenCdpUtxo));
   const orefs = await findAllNecessaryOrefs(
@@ -525,7 +540,11 @@ export async function runLiquidateCdp(
     frozenCdpUtxo,
     orefs.stabilityPoolUtxo,
     orefs.interestCollectorUtxo,
-    orefs.treasuryUtxo,
+    directTreasuryPayment
+      ? undefined
+      : treasuryUtxo
+        ? treasuryUtxo
+        : orefs.treasuryUtxo,
     sysParams,
     context.lucid,
   );
@@ -544,6 +563,7 @@ export async function executeLiquidation(
   liquidateWrapper: (liquidateTx: TxBuilder) => Promise<void> = async (tx) => {
     await runAndAwaitTxBuilder(context.lucid, tx);
   },
+  directTreasuryPayment: boolean = false,
 ) {
   const price = await findPrice(
     context.lucid,
@@ -566,6 +586,46 @@ export async function executeLiquidation(
     ),
   );
 
+  const user4Value = A.reduce<UTxO, Assets>({}, (acc, utxo) =>
+    addAssets(acc, utxo.assets),
+  )(await context.lucid.utxosAt(context.users['user4'].address));
+
+  const iassetAc: AssetClass = {
+    currencySymbol: fromHex(
+      sysParams.cdpParams.cdpAssetSymbol.unCurrencySymbol,
+    ),
+    tokenName: fromHex(fromText(assetInfo.iassetTokenNameAscii)),
+  };
+
+  // Send iAssets to treasury so liquidation is performed from a clean address.
+  await sendValueTo(
+    mkTreasuryAddr(context.lucid, sysParams),
+    mkAssetsOf(iassetAc, assetClassValueOf(user4Value, iassetAc)),
+    context.lucid,
+  );
+
+  const user4collateralAssetAmount = assetClassValueOf(
+    user4Value,
+    collateralAsset,
+  );
+
+  // Send collateral asset to treasury so liquidation is performed from a clean address.
+  if (collateralAsset !== adaAssetClass && user4collateralAssetAmount > 0n) {
+    await sendValueTo(
+      mkTreasuryAddr(context.lucid, sysParams),
+      mkAssetsOf(collateralAsset, user4collateralAssetAmount),
+      context.lucid,
+    );
+  }
+
+  const user4ValueBeforeLiquidation = A.reduce<UTxO, Assets>({}, (acc, utxo) =>
+    addAssets(acc, utxo.assets),
+  )(await context.lucid.utxosAt(context.users['user4'].address));
+
+  assert(Object.keys(user4ValueBeforeLiquidation).length === 1);
+
+  context.emulator.awaitSlot(1000);
+
   await runFeedPriceToOracle(
     context,
     sysParams,
@@ -596,8 +656,29 @@ export async function executeLiquidation(
     (_) => new Error('Expected only single frozen CDP'),
   );
 
+  let treasuryUtxo = undefined;
+
+  if (!directTreasuryPayment) {
+    treasuryUtxo = await findRandomTreasuryUtxoWithAsset(
+      context.lucid,
+      sysParams,
+      collateralAsset,
+    );
+
+    // This treasury UTxO should contain ADA and collateral asset.
+    if (collateralAsset !== adaAssetClass) {
+      assert(Object.keys(treasuryUtxo.assets).length === 2);
+    }
+  }
+
   await liquidateWrapper(
-    await runLiquidateCdp(context, sysParams, frozenCdp.utxo),
+    await runLiquidateCdp(
+      context,
+      sysParams,
+      frozenCdp.utxo,
+      treasuryUtxo,
+      directTreasuryPayment,
+    ),
   );
 
   await runFeedPriceToOracle(