@indigo-labs/indigo-sdk 0.3.16 → 0.3.17

package/tests/treasury.test.ts

@@ -1,242 +0,0 @@
-import { addAssets, Assets, TxBuilder } from '@lucid-evolution/lucid';
-import { assert, beforeEach, describe, test } from 'vitest';
-import { IndigoTestContext } from './test-helpers';
-import { treasuryCollect, treasuryMerge, treasurySplit } from '../src';
-import {
-  createIndigoTestContext,
-  EXAMPLE_TOKEN_1,
-  EXAMPLE_TOKEN_2,
-  EXAMPLE_TOKEN_3,
-} from './indigo-test-helpers';
-import { benchmarkAndAwaitTx } from './utils/benchmark-utils';
-import {
-  findAllTreasuryUtxos,
-  findRandomTreasuryUtxoWithOnlyAda,
-} from './queries/treasury-queries';
-import {
-  adaAssetClass,
-  mkLovelacesOf,
-} from '@3rd-eye-labs/cardano-offchain-common';
-import { mkAssetsOf } from '@3rd-eye-labs/cardano-offchain-common';
-import { createUtxoAtTreasury } from './endpoints/treasury';
-import { expectScriptFailure } from './utils/asserts';
-
-// NOTE: Withdrawing and PrepareWithdrawal from Treasury is tested in gov.test.ts
-describe('Treasury', () => {
-  beforeEach<IndigoTestContext>(async (context: IndigoTestContext) => {
-    await createIndigoTestContext(context);
-  });
-
-  async function treasuryMergeTx(
-    context: IndigoTestContext,
-    treasuryOutputs: Assets[],
-  ): Promise<TxBuilder> {
-    for (const output of treasuryOutputs) {
-      await createUtxoAtTreasury(output, context.systemParams, context);
-    }
-
-    const treasuryUtxos = await findAllTreasuryUtxos(
-      context.lucid,
-      context.systemParams,
-    );
-
-    // Simpler approach: for each expected output in treasuryOutputs, find one UTxO with exactly matching assets and take it
-    // Find UTxOs for each expected output, but remove it from the candidate list as we match
-    const availableUtxos = [...treasuryUtxos];
-    const matchedTreasuryUtxos = treasuryOutputs
-      .map((expectedAssets) => {
-        const idx = availableUtxos.findIndex((utxo) => {
-          const utxoAssets = utxo.assets;
-          const keys1 = Object.keys(utxoAssets).sort();
-          const keys2 = Object.keys(expectedAssets).sort();
-          if (keys1.length !== keys2.length) return false;
-          return keys1.every(
-            (k, i) => k === keys2[i] && utxoAssets[k] === expectedAssets[k],
-          );
-        });
-        if (idx !== -1) {
-          return availableUtxos.splice(idx, 1)[0];
-        } else {
-          return undefined;
-        }
-      })
-      .filter((utxo) => utxo !== undefined);
-
-    assert(
-      matchedTreasuryUtxos.length === treasuryOutputs.length,
-      'Expected all treasury outputs to be matched',
-    );
-
-    return await treasuryMerge(
-      matchedTreasuryUtxos,
-      context.lucid,
-      context.systemParams,
-    );
-  }
-
-  async function treasurySplitTx(
-    context: IndigoTestContext,
-    treasuryOutput: Assets,
-  ): Promise<TxBuilder> {
-    await createUtxoAtTreasury(treasuryOutput, context.systemParams, context);
-
-    const treasuryUtxos = await findAllTreasuryUtxos(
-      context.lucid,
-      context.systemParams,
-    );
-
-    const matchedTreasuryUtxo = treasuryUtxos.find((utxo) =>
-      Object.keys(treasuryOutput).every(
-        (key) => utxo.assets[key] === treasuryOutput[key],
-      ),
-    );
-
-    assert(
-      matchedTreasuryUtxo !== undefined,
-      'Expected a single treasury UTXO',
-    );
-
-    return await treasurySplit(
-      matchedTreasuryUtxo,
-      context.lucid,
-      context.systemParams,
-    );
-  }
-
-  test<IndigoTestContext>('Merge (3 lovelace UTxOs)', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    await benchmarkAndAwaitTx(
-      'Treasury - Merge (3 lovelace UTxOs)',
-      await treasuryMergeTx(context, [
-        mkLovelacesOf(50_000_000n),
-        mkLovelacesOf(50_000_000n),
-        mkLovelacesOf(50_000_000n),
-      ]),
-      context.lucid,
-      context.emulator,
-    );
-  });
-
-  test<IndigoTestContext>('Merge (3 INDY UTxOs)', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    await benchmarkAndAwaitTx(
-      'Treasury - Merge (3 INDY UTxOs)',
-      await treasuryMergeTx(context, [
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1_000_000_000n),
-        ),
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1_000_000_000n),
-        ),
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1_000_000_000n),
-        ),
-      ]),
-      context.lucid,
-      context.emulator,
-    );
-  });
-
-  test<IndigoTestContext>('Merge (fail, 1 lovelace, 2 INDY UTxOs)', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    await expectScriptFailure(
-      'All inputs have to contain the same merging assets',
-      treasuryMergeTx(context, [
-        mkLovelacesOf(5_000_000n),
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1_000_000_000n),
-        ),
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1_000_000_000n),
-        ),
-      ]),
-    );
-  });
-
-  test<IndigoTestContext>('Split (3 assets)', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    await benchmarkAndAwaitTx(
-      'Treasury - Split (3 assets)',
-      await treasurySplitTx(
-        context,
-        addAssets(
-          mkLovelacesOf(5_000_000n),
-          mkAssetsOf(EXAMPLE_TOKEN_1, 1n),
-          mkAssetsOf(EXAMPLE_TOKEN_2, 1n),
-          mkAssetsOf(EXAMPLE_TOKEN_3, 1n),
-        ),
-      ),
-      context.lucid,
-      context.emulator,
-    );
-  });
-
-  test<IndigoTestContext>('Split (fail, lovelace only)', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    await expectScriptFailure(
-      'All inputs must have more than 1 non-ADA asset or 1 non-ADA asset and more than 2x ADA buffer',
-      treasurySplitTx(context, mkLovelacesOf(5_000_000n)),
-    );
-  });
-
-  test<IndigoTestContext>('Collect non-positive amount or negative extra lovelaces fails', async (context: IndigoTestContext) => {
-    context.lucid.selectWallet.fromSeed(context.users.user.seedPhrase);
-
-    const treasuryUTxO = await findRandomTreasuryUtxoWithOnlyAda(
-      context.lucid,
-      context.systemParams,
-    );
-
-    // This tests that the amount to collect cannot be negative.
-    await expectScriptFailure(
-      'Collected amount is positive',
-      treasuryCollect(
-        adaAssetClass,
-        -1_000_000n,
-        0n,
-        context.lucid,
-        context.systemParams,
-        treasuryUTxO,
-        treasuryUTxO,
-      ),
-    );
-
-    // This tests that the amount to collect cannot be zero.
-    await expectScriptFailure(
-      'Collected amount is positive',
-      treasuryCollect(
-        adaAssetClass,
-        0n,
-        0n,
-        context.lucid,
-        context.systemParams,
-        treasuryUTxO,
-        treasuryUTxO,
-      ),
-    );
-
-    // This tests that the extra lovelaces cannot be negative.
-    await expectScriptFailure(
-      'Collected amount is positive',
-      treasuryCollect(
-        adaAssetClass,
-        1n,
-        -1_000_000n,
-        context.lucid,
-        context.systemParams,
-        treasuryUTxO,
-        treasuryUTxO,
-      ),
-    );
-  });
-});