@indigo-labs/indigo-sdk 0.2.41 → 0.3.0

package/tests/datums.test.ts

@@ -1,18 +1,5 @@
 import { describe, expect, it } from 'vitest';
-import {
-  CDPContent,
-  IAssetContent,
-  parseIAssetDatumOrThrow,
-  parseInterestOracleDatum,
-  parsePriceOracleDatum,
-  serialiseFeedInterestOracleRedeemer,
-  serialiseIAssetDatum,
-  serialiseInterestOracleDatum,
-  serialisePriceOracleDatum,
-  serialiseCdpDatum,
-  parseCdpDatum,
-} from '../src/index';
-import { fromHex, fromText } from '@lucid-evolution/lucid';
+import { Data, fromHex } from '@lucid-evolution/lucid';
 import {
   parseStakingManagerDatum,
   parseStakingPositionOrThrow,
@@ -21,30 +8,34 @@ import {
   StakingPosition,
 } from '../src/contracts/staking/types-new';
 import {
-  AccountContent,
-  parseAccountDatum,
-  parseSnapshotEpochToScaleToSumDatum,
-  parseStabilityPoolDatum,
-  serialiseStabilityPoolDatum,
-  SnapshotEpochToScaleToSumContent,
-  StabilityPoolContent,
-} from '../src/contracts/stability-pool/types-new';
-import { option as O } from 'fp-ts';
+  parsePriceOracleDatum,
+  serialisePriceOracleDatum,
+} from '../src/contracts/price-oracle/types-new';
+import {
+  parseInterestOracleDatum,
+  serialiseFeedInterestOracleRedeemer,
+  serialiseInterestOracleDatum,
+} from '../src/contracts/interest-oracle/types-new';
+import * as Core from '@evolution-sdk/evolution';
 
 describe('Datum checks', () => {
   it('Price Oracle', () => {
-    expect(
-      parsePriceOracleDatum('d8799fd8799f1a0013c347ff1b00000194d68e13d8ff'),
-    ).toEqual({
-      price: { getOnChainInt: 1295175n },
-      expiration: 1738766423000n,
-    });
     expect(
       serialisePriceOracleDatum({
-        price: { getOnChainInt: 1295175n },
-        expiration: 1738766423000n,
+        price: { numerator: 1_295_175n, denominator: 1_000_000n },
+        expirationTime: 1738766423000n,
+        auxiliaryData: Core.Data.fromCBORHex(Data.void()),
       }),
-    ).toEqual('d8799fd8799f1a0013c347ff1b00000194d68e13d8ff');
+    ).toEqual('d8799fd8799f1a0013c3471a000f4240ff1b00000194d68e13d8d87980ff');
+    expect(
+      parsePriceOracleDatum(
+        'd8799fd8799f1a0013c3471a000f4240ff1b00000194d68e13d8d87980ff',
+      ),
+    ).toEqual({
+      price: { numerator: 1295175n, denominator: 1000000n },
+      expirationTime: 1738766423000n,
+      auxiliaryData: Core.Data.fromCBORHex(Data.void()),
+    });
   });
 
   it('Interest Oracle', () => {
@@ -75,78 +66,82 @@ describe('Datum checks', () => {
     ).toEqual('d8799fd8799f1a000f4240ff1a66cf2400ff');
   });
 
-  it('CDP', () => {
-    // Active CDP
-    const activeCDPDatum =
-      'd8799fd8799fd8799f581c98e30e1c6dbb727dc98bdcb48b99b313c97fabfb537ff4b29a94ed1cff44695553441b00000004d9b0a47ed8799f1b00000194d5ebec201b03022de04fddf5f9ffffff';
-    const activeCDPObject: CDPContent = {
-      cdpOwner: '98e30e1c6dbb727dc98bdcb48b99b313c97fabfb537ff4b29a94ed1c',
-      iasset: fromText('iUSD'),
-      mintedAmt: 20832101502n,
-      cdpFees: {
-        ActiveCDPInterestTracking: {
-          lastSettled: 1738755796000n,
-          unitaryInterestSnapshot: 216786173503075833n,
-        },
-      },
-    };
-    expect(serialiseCdpDatum(activeCDPObject)).toEqual(activeCDPDatum);
-    expect(parseCdpDatum(activeCDPDatum)).toEqual(O.some(activeCDPObject));
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it.todo('CDP', () => {
+  //   // Active CDP
+  //   const activeCDPDatum =
+  //     'd8799fd8799fd8799f581c98e30e1c6dbb727dc98bdcb48b99b313c97fabfb537ff4b29a94ed1cff44695553441b00000004d9b0a47ed8799f1b00000194d5ebec201b03022de04fddf5f9ffffff';
+  //   const activeCDPObject: CDPContent = {
+  //     cdpOwner: fromHex(
+  //       '98e30e1c6dbb727dc98bdcb48b99b313c97fabfb537ff4b29a94ed1c',
+  //     ),
+  //     iasset: fromHex(fromText('iUSD')),
+  //     mintedAmt: 20832101502n,
+  //     cdpFees: {
+  //       ActiveCDPInterestTracking: {
+  //         lastSettled: 1738755796000n,
+  //         unitaryInterestSnapshot: 216786173503075833n,
+  //       },
+  //     },
+  //   };
+  //   expect(serialiseCdpDatum(activeCDPObject)).toEqual(activeCDPDatum);
+  //   expect(parseCdpDatum(activeCDPDatum)).toEqual(O.some(activeCDPObject));
 
-    // Frozen CDP
-    const frozenCDPDatum =
-      'd8799fd8799fd87a8044695553441a0050924ed87a9f1a0002765a1a0003ca56ffffff';
-    const frozenCDPObject: CDPContent = {
-      cdpOwner: null,
-      iasset: fromText('iUSD'),
-      mintedAmt: 5280334n,
-      cdpFees: {
-        FrozenCDPAccumulatedFees: {
-          lovelacesTreasury: 161370n,
-          lovelacesIndyStakers: 248406n,
-        },
-      },
-    };
-    expect(parseCdpDatum(frozenCDPDatum)).toEqual(O.some(frozenCDPObject));
-    expect(serialiseCdpDatum(frozenCDPObject)).toEqual(frozenCDPDatum);
-  });
+  //   // Frozen CDP
+  //   const frozenCDPDatum =
+  //     'd8799fd8799fd87a8044695553441a0050924ed87a9f1a0002765a1a0003ca56ffffff';
+  //   const frozenCDPObject: CDPContent = {
+  //     cdpOwner: null,
+  //     iasset: fromHex(fromText('iUSD')),
+  //     mintedAmt: 5280334n,
+  //     cdpFees: {
+  //       FrozenCDPAccumulatedFees: {
+  //         iassetInterest: 161370n,
+  //         lovelacesIndyStakers: 248406n,
+  //       },
+  //     },
+  //   };
+  //   expect(parseCdpDatum(frozenCDPDatum)).toEqual(O.some(frozenCDPObject));
+  //   expect(serialiseCdpDatum(frozenCDPObject)).toEqual(frozenCDPDatum);
+  // });
 
-  it('iAsset', () => {
-    const assetDatum =
-      'd87a9fd8799f4469455448d87a9fd8799fd8799f581c6c9497ffd7e8baf86c3c0d6fcd43c524daa49ad5fceba26d715468e952694554483230323231323139313931333032ffffffd8799f581c7b75e317505dddce858ae7bf200656a967c7544e55efa5d18ef302494d694554485f494e544552455354ffd8799f1a08f0d180ffd8799f1a06dac2c0ffd8799f1a068e7780ffd8799f1a000186a0ffd8799f1a001e8480ffd8799f19c350ffd8799f1a000f4240ffd8799f1a000f4240ffd8799f1a01c9c380ffd87980d8799f4469534f4cffffff';
-    const assetObject: IAssetContent = {
-      assetName: fromText('iETH'),
-      price: {
-        Oracle: {
-          content: {
-            oracleNft: {
-              currencySymbol:
-                '6c9497ffd7e8baf86c3c0d6fcd43c524daa49ad5fceba26d715468e9',
-              tokenName: fromText('iETH20221219191302'),
-            },
-          },
-        },
-      },
-      interestOracleNft: {
-        currencySymbol:
-          '7b75e317505dddce858ae7bf200656a967c7544e55efa5d18ef30249',
-        tokenName: fromText('iETH_INTEREST'),
-      },
-      redemptionRatio: { getOnChainInt: 150000000n },
-      maintenanceRatio: { getOnChainInt: 115000000n },
-      liquidationRatio: { getOnChainInt: 110000000n },
-      debtMintingFeePercentage: { getOnChainInt: 100000n },
-      liquidationProcessingFeePercentage: { getOnChainInt: 2000000n },
-      stabilityPoolWithdrawalFeePercentage: { getOnChainInt: 50000n },
-      redemptionReimbursementPercentage: { getOnChainInt: 1000000n },
-      redemptionProcessingFeePercentage: { getOnChainInt: 1000000n },
-      interestCollectorPortionPercentage: { getOnChainInt: 30000000n },
-      firstIAsset: false,
-      nextIAsset: fromText('iSOL'),
-    };
-    expect(serialiseIAssetDatum(assetObject)).toEqual(assetDatum);
-    expect(parseIAssetDatumOrThrow(assetDatum)).toEqual(assetObject);
-  });
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it.todo('iAsset', () => {
+  //   const assetDatum =
+  //     'd87a9fd8799f4469455448d87a9fd8799fd8799f581c6c9497ffd7e8baf86c3c0d6fcd43c524daa49ad5fceba26d715468e952694554483230323231323139313931333032ffffffd8799f581c7b75e317505dddce858ae7bf200656a967c7544e55efa5d18ef302494d694554485f494e544552455354ffd8799f1a08f0d180ffd8799f1a06dac2c0ffd8799f1a068e7780ffd8799f1a000186a0ffd8799f1a001e8480ffd8799f19c350ffd8799f1a000f4240ffd8799f1a000f4240ffd8799f1a01c9c380ffd87980d8799f4469534f4cffffff';
+  //   const assetObject: IAssetDatum = {
+  //     assetName: fromHex(fromText('iETH')),
+  //     price: {
+  //       Oracle: {
+  //         oracleNft: {
+  //           currencySymbol: fromHex(
+  //             '6c9497ffd7e8baf86c3c0d6fcd43c524daa49ad5fceba26d715468e9',
+  //           ),
+  //           tokenName: fromHex(fromText('iETH20221219191302')),
+  //         },
+  //       },
+  //     },
+  //     interestOracleNft: {
+  //       currencySymbol: fromHex(
+  //         '7b75e317505dddce858ae7bf200656a967c7544e55efa5d18ef30249',
+  //       ),
+  //       tokenName: fromHex(fromText('iETH_INTEREST')),
+  //     },
+  //     redemptionRatio: { getOnChainInt: 150000000n },
+  //     maintenanceRatio: { getOnChainInt: 115000000n },
+  //     liquidationRatio: { getOnChainInt: 110000000n },
+  //     debtMintingFeePercentage: { getOnChainInt: 100000n },
+  //     liquidationProcessingFeePercentage: { getOnChainInt: 2000000n },
+  //     stabilityPoolWithdrawalFeePercentage: { getOnChainInt: 50000n },
+  //     redemptionReimbursementPercentage: { getOnChainInt: 1000000n },
+  //     redemptionProcessingFeePercentage: { getOnChainInt: 1000000n },
+  //     interestCollectorPortionPercentage: { getOnChainInt: 30000000n },
+  //     firstIAsset: false,
+  //     nextIAsset: fromHex(fromText('iSOL')),
+  //   };
+  //   expect(serialiseIAssetDatum(assetObject)).toEqual(assetDatum);
+  //   expect(parseIAssetDatumOrThrow(assetDatum)).toEqual(assetObject);
+  // });
 
   it('Staking Manager', () => {
     const stakingManagerDatum =
@@ -168,14 +163,14 @@ describe('Datum checks', () => {
 
   it('Staking Position', () => {
     const stakingPositionDatum =
-      'd87a9fd8799f581cd45527a088a92fd31f42b5777fe39c40f810e0f79d13c6d77eeb7f43a11853d8799f1a5c8c1cfb1b0000019616971410ffd8799f1b0000013a7ed5b0fdffffff';
+      'd87a9fd8799f581cd45527a088a92fd31f42b5777fe39c40f810e0f79d13c6d77eeb7f439f9f1853d8799f1a5c8c1cfb1b0000019616971410ffffffd8799f1b0000013a7ed5b0fdffffff';
     const stakingPositionObject: StakingPosition = {
       owner: fromHex(
         'd45527a088a92fd31f42b5777fe39c40f810e0f79d13c6d77eeb7f43',
       ),
-      lockedAmount: new Map([
+      lockedAmount: [
         [83n, { voteAmt: 1552686331n, votingEnd: 1744135722000n }],
-      ]),
+      ],
       positionSnapshot: {
         snapshotAda: 1350747664637n,
       },
@@ -189,115 +184,119 @@ describe('Datum checks', () => {
     );
   });
 
-  it('Stability Pool', () => {
-    const stabilityPoolDatum =
-      'd8799fd8799f4469555344d8799fd8799f1b0a37ad5c452ffb2affd8799fc24d1f94ac680ce6b48ea21bb122baffd8799f1b0fde3bba456cd5deff0100ffa2d8799f0000ffd8799f1b084494e2d23b2b7effd8799f0100ffd8799f1b0fde3bba456cd5deffffff';
-    const stabilityPoolObject: StabilityPoolContent = {
-      asset: fromHex(fromText('iUSD')),
-      poolSnapshot: {
-        productVal: { value: 736247675907734314n },
-        depositVal: { value: 2502085246000826468068228145850n },
-        sumVal: { value: 1143417026613401054n },
-        epoch: 1n,
-        scale: 0n,
-      },
-      epochToScaleToSum: new Map([
-        [{ epoch: 0n, scale: 0n }, { value: 595764752630360958n }],
-        [{ epoch: 1n, scale: 0n }, { value: 1143417026613401054n }],
-      ]),
-    };
-    expect(parseStabilityPoolDatum(stabilityPoolDatum)).toEqual(
-      stabilityPoolObject,
-    );
-    expect(
-      serialiseStabilityPoolDatum({ StabilityPool: stabilityPoolObject }),
-    ).toEqual(stabilityPoolDatum);
-  });
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it('Stability Pool', () => {
+  //   const stabilityPoolDatum =
+  //     'd8799fd8799f4469555344d8799fd8799f1b0a37ad5c452ffb2affd8799fc24d1f94ac680ce6b48ea21bb122baffd8799f1b0fde3bba456cd5deff0100ff9f9fd8799f0000ffd8799f1b084494e2d23b2b7effff9fd8799f0100ffd8799f1b0fde3bba456cd5deffffffffff';
+  //   const stabilityPoolObject: StabilityPoolContent = {
+  //     asset: fromHex(fromText('iUSD')),
+  //     poolSnapshot: {
+  //       productVal: { value: 736247675907734314n },
+  //       depositVal: { value: 2502085246000826468068228145850n },
+  //       sumVal: { value: 1143417026613401054n },
+  //       epoch: 1n,
+  //       scale: 0n,
+  //     },
+  //     epochToScaleToSum: [
+  //       [{ epoch: 0n, scale: 0n }, { value: 595764752630360958n }],
+  //       [{ epoch: 1n, scale: 0n }, { value: 1143417026613401054n }],
+  //     ],
+  //   };
+  //   expect(parseStabilityPoolDatum(stabilityPoolDatum)).toEqual(
+  //     stabilityPoolObject,
+  //   );
+  //   expect(
+  //     serialiseStabilityPoolDatum({ StabilityPool: stabilityPoolObject }),
+  //   ).toEqual(stabilityPoolDatum);
+  // });
 
-  it('Stability Pool Account', () => {
-    const stabilityPoolDatum =
-      'd87a9fd8799f581c12c646d4c6d7a35c14788d15f0f6142f6148975d8932592fbd625f674469555344d8799fd8799f1b0a37ad5c452ffb2affd8799fc24c39fa2838b1f7dd38267f0a6dffd8799f1b0fde3b75c28ab489ff0100ffd87a80ffff';
-    const stabilityPoolObject: AccountContent = {
-      owner: fromHex(
-        '12c646d4c6d7a35c14788d15f0f6142f6148975d8932592fbd625f67',
-      ),
-      asset: fromHex(fromText('iUSD')),
-      accountSnapshot: {
-        productVal: { value: 736247675907734314n },
-        depositVal: { value: 17943066955221270821727046253n },
-        sumVal: { value: 1143416732359767177n },
-        epoch: 1n,
-        scale: 0n,
-      },
-      request: null,
-    };
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it('Stability Pool Account', () => {
+  //   const stabilityPoolDatum =
+  //     'd87a9fd8799f581c12c646d4c6d7a35c14788d15f0f6142f6148975d8932592fbd625f674469555344d8799fd8799f1b0a37ad5c452ffb2affd8799fc24c39fa2838b1f7dd38267f0a6dffd8799f1b0fde3b75c28ab489ff0100ffd87a80ffff';
+  //   const stabilityPoolObject: AccountContent = {
+  //     owner: fromHex(
+  //       '12c646d4c6d7a35c14788d15f0f6142f6148975d8932592fbd625f67',
+  //     ),
+  //     asset: fromHex(fromText('iUSD')),
+  //     accountSnapshot: {
+  //       productVal: { value: 736247675907734314n },
+  //       depositVal: { value: 17943066955221270821727046253n },
+  //       sumVal: { value: 1143416732359767177n },
+  //       epoch: 1n,
+  //       scale: 0n,
+  //     },
+  //     request: null,
+  //   };
 
-    expect(parseAccountDatum(stabilityPoolDatum)).toEqual(stabilityPoolObject);
-    expect(
-      serialiseStabilityPoolDatum({ Account: stabilityPoolObject }),
-    ).toEqual(stabilityPoolDatum);
-  });
+  //   expect(parseAccountDatum(stabilityPoolDatum)).toEqual(stabilityPoolObject);
+  //   expect(
+  //     serialiseStabilityPoolDatum({ Account: stabilityPoolObject }),
+  //   ).toEqual(stabilityPoolDatum);
+  // });
 
-  it('Stability Pool Account w/ Adjust Request', () => {
-    const stabilityPoolDatum =
-      'd87a9fd8799f581c90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c4469555344d8799fd8799f1b0a374472be304a62ffd8799fc24b01aef07f96e5ce00f80000ffd8799f1b0f88aa07a1048079ff0100ffd8799fd87a9f3a0007c359d8799fd8799f581c90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15cffd8799fd8799fd8799f581c75a4f9204b9308a92a09b0e22b94125e56f24b73bb85e2795f176c6affffffffffffffff';
-    const stabilityPoolObject: AccountContent = {
-      owner: fromHex(
-        '90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c',
-      ),
-      asset: fromHex(fromText('iUSD')),
-      accountSnapshot: {
-        productVal: { value: 736132323706161762n },
-        depositVal: { value: 2035054000000000000000000n },
-        sumVal: { value: 1119331457144488057n },
-        epoch: 1n,
-        scale: 0n,
-      },
-      request: {
-        Adjust: {
-          amount: -508762n,
-          outputAddress: {
-            paymentCredential: {
-              PublicKeyCredential: fromHex(
-                '90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c',
-              ),
-            },
-            stakeCredential: {
-              Inline: {
-                PublicKeyCredential: fromHex(
-                  '75a4f9204b9308a92a09b0e22b94125e56f24b73bb85e2795f176c6a',
-                ),
-              },
-            },
-          },
-        },
-      },
-    };
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it('Stability Pool Account w/ Adjust Request', () => {
+  //   const stabilityPoolDatum =
+  //     'd87a9fd8799f581c90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c4469555344d8799fd8799f1b0a374472be304a62ffd8799fc24b01aef07f96e5ce00f80000ffd8799f1b0f88aa07a1048079ff0100ffd8799fd87a9f3a0007c359d8799fd8799f581c90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15cffd8799fd8799fd8799f581c75a4f9204b9308a92a09b0e22b94125e56f24b73bb85e2795f176c6affffffffffffffff';
+  //   const stabilityPoolObject: AccountContent = {
+  //     owner: fromHex(
+  //       '90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c',
+  //     ),
+  //     asset: fromHex(fromText('iUSD')),
+  //     accountSnapshot: {
+  //       productVal: { value: 736132323706161762n },
+  //       depositVal: { value: 2035054000000000000000000n },
+  //       sumVal: { value: 1119331457144488057n },
+  //       epoch: 1n,
+  //       scale: 0n,
+  //     },
+  //     request: {
+  //       Adjust: {
+  //         amount: -508762n,
+  //         outputAddress: {
+  //           paymentCredential: {
+  //             PublicKeyCredential: fromHex(
+  //               '90e40129516ee738fa6aa9183cf57b45c46946496e1590d34ca1b15c',
+  //             ),
+  //           },
+  //           stakeCredential: {
+  //             Inline: {
+  //               PublicKeyCredential: fromHex(
+  //                 '75a4f9204b9308a92a09b0e22b94125e56f24b73bb85e2795f176c6a',
+  //               ),
+  //             },
+  //           },
+  //         },
+  //       },
+  //     },
+  //   };
 
-    expect(parseAccountDatum(stabilityPoolDatum)).toEqual(stabilityPoolObject);
-    expect(
-      serialiseStabilityPoolDatum({ Account: stabilityPoolObject }),
-    ).toEqual(stabilityPoolDatum);
-  });
+  //   expect(parseAccountDatum(stabilityPoolDatum)).toEqual(stabilityPoolObject);
+  //   expect(
+  //     serialiseStabilityPoolDatum({ Account: stabilityPoolObject }),
+  //   ).toEqual(stabilityPoolDatum);
+  // });
 
-  it('Stability Pool SnapshotEpochToScaleToSum', () => {
-    const stabilityPoolDatum =
-      'd87b9fd8799fa2d8799f0000ffd8799f1b084494e2d23b2b7effd8799f0100ffd8799f1b0fde3bba456cd5deff4469555344ffff';
-    const stabilityPoolObject: SnapshotEpochToScaleToSumContent = {
-      asset: fromHex(fromText('iUSD')),
-      snapshot: new Map([
-        [{ epoch: 0n, scale: 0n }, { value: 595764752630360958n }],
-        [{ epoch: 1n, scale: 0n }, { value: 1143417026613401054n }],
-      ]),
-    };
+  // eslint-disable-next-line vitest/no-commented-out-tests
+  // it('Stability Pool SnapshotEpochToScaleToSum', () => {
+  //   const stabilityPoolDatum =
+  //     'd87b9fd8799f9f9fd8799f0000ffd8799f1b084494e2d23b2b7effff9fd8799f0100ffd8799f1b0fde3bba456cd5deffffff4469555344ffff';
+  //   const stabilityPoolObject: SnapshotEpochToScaleToSumContent = {
+  //     asset: fromHex(fromText('iUSD')),
+  //     snapshot: [
+  //       [{ epoch: 0n, scale: 0n }, { value: 595764752630360958n }],
+  //       [{ epoch: 1n, scale: 0n }, { value: 1143417026613401054n }],
+  //     ],
+  //   };
 
-    expect(parseSnapshotEpochToScaleToSumDatum(stabilityPoolDatum)).toEqual(
-      stabilityPoolObject,
-    );
-    expect(
-      serialiseStabilityPoolDatum({
-        SnapshotEpochToScaleToSum: stabilityPoolObject,
-      }),
-    ).toEqual(stabilityPoolDatum);
-  });
+  //   expect(parseSnapshotEpochToScaleToSumDatum(stabilityPoolDatum)).toEqual(
+  //     stabilityPoolObject,
+  //   );
+  //   expect(
+  //     serialiseStabilityPoolDatum({
+  //       SnapshotEpochToScaleToSum: stabilityPoolObject,
+  //     }),
+  //   ).toEqual(stabilityPoolDatum);
+  // });
 });

package/tests/endpoints/initialize.ts

@@ -0,0 +1,68 @@
+/**
+ * Test endpoint for protocol initialization.
+ * Re-exports from the initialize contract and provides a test-friendly init() wrapper
+ * that uses protocolParamsMock and optional initPyth when pythConfig is not provided.
+ */
+
+import type { LucidEvolution } from '@lucid-evolution/lucid';
+import type { ScriptCredential } from '../../src';
+import type { ProtocolParams } from '../../src/contracts/gov/types-new';
+import {
+  init as initContract,
+  initPythConfig,
+} from '../../src/contracts/initialize/actions';
+import {
+  mintOneTimeToken,
+  DEFAULT_INIT_OPTIONS,
+} from '../../src/contracts/initialize/helpers';
+import type {
+  AssetInfo,
+  CollateralAssetInfo,
+  InitialAssetParam,
+  InitializeOptions,
+} from '../../src/contracts/initialize/types';
+import { protocolParamsMock } from '../mock/protocol-params-mock';
+import { AssetClass } from '@3rd-eye-labs/cardano-offchain-common';
+
+export type {
+  AssetInfo,
+  CollateralAssetInfo,
+  InitialAssetParam,
+  InitializeOptions,
+};
+export { initPythConfig, mintOneTimeToken, DEFAULT_INIT_OPTIONS };
+
+/**
+ * Initialize the protocol (test helper).
+ * Uses protocolParamsMock merged with overrideProtocolParams.
+ * If pythConfig is not provided, initializes Pyth via initPyth + initPythConfig.
+ * Uses DEFAULT_INIT_OPTIONS when initOptions is not provided.
+ */
+export async function init(
+  lucid: LucidEvolution,
+  defaultInitialAssets: InitialAssetParam[],
+  currentSlot: number,
+  mkPythBasedAssets: (
+    pythStateNft: AssetClass,
+  ) => InitialAssetParam[] = () => [],
+  overrideProtocolParams: Partial<ProtocolParams> = {},
+  initOptions: InitializeOptions = DEFAULT_INIT_OPTIONS,
+  spStakingCred?: ScriptCredential,
+  treasuryStakingCred?: ScriptCredential,
+): Promise<ReturnType<typeof initContract>> {
+  const protocolParams: ProtocolParams = {
+    ...protocolParamsMock,
+    ...overrideProtocolParams,
+  };
+
+  return initContract(
+    lucid,
+    defaultInitialAssets,
+    currentSlot,
+    mkPythBasedAssets,
+    protocolParams,
+    initOptions,
+    spStakingCred,
+    treasuryStakingCred,
+  );
+}

package/tests/endpoints/interest-collector.ts

@@ -0,0 +1,37 @@
+import { Assets, Data, UTxO } from '@lucid-evolution/lucid';
+import { createScriptAddress, SystemParams } from '../../src';
+import { IndigoTestContext, runAndAwaitTxBuilder } from '../test-helpers';
+import { getNewUtxosAtAddressAfterAction } from '../utils';
+
+export async function createUtxosAtInterestCollector(
+  utxosNo: number,
+  assets: Assets,
+  sysParams: SystemParams,
+  context: IndigoTestContext,
+): Promise<UTxO[]> {
+  const interestCollectorAddr = createScriptAddress(
+    context.lucid.config().network!,
+    sysParams.validatorHashes.interestCollectionHash,
+  );
+
+  const tx = context.lucid.newTx();
+
+  for (let i = 0; i < utxosNo; i++) {
+    tx.pay.ToContract(
+      interestCollectorAddr,
+      { kind: 'inline', value: Data.void() },
+      assets,
+    );
+  }
+
+  const [_, utxos] = await getNewUtxosAtAddressAfterAction(
+    context.lucid,
+    interestCollectorAddr,
+    () => runAndAwaitTxBuilder(context.lucid, tx),
+  );
+
+  if (utxos.length !== utxosNo)
+    throw Error('Expected a different number of interest collectors');
+
+  return utxos;
+}

package/tests/endpoints/treasury.ts

@@ -0,0 +1,70 @@
+import { Assets, Data, UTxO } from '@lucid-evolution/lucid';
+import {
+  fromSystemParamsAsset,
+  SystemParams,
+} from '../../src/types/system-params';
+import { LucidContext, runAndAwaitTxBuilder } from '../test-helpers';
+import { createScriptAddress } from '../../src/utils/lucid-utils';
+import { getNewUtxosAtAddressAfterAction } from '../utils';
+import { matchSingle, mkAssetsOf } from '@3rd-eye-labs/cardano-offchain-common';
+
+export async function createMultipleUtxosAtTreasury(
+  assets: Assets,
+  numberUtxos: bigint,
+  sysParams: SystemParams,
+  context: LucidContext,
+): Promise<UTxO[]> {
+  const treasuryAddr = createScriptAddress(
+    context.lucid.config().network!,
+    sysParams.validatorHashes.treasuryHash,
+  );
+
+  const tx = context.lucid.newTx();
+
+  for (let i = 0; i < numberUtxos; i++) {
+    tx.pay.ToContract(
+      treasuryAddr,
+      { kind: 'inline', value: Data.void() },
+      assets,
+    );
+  }
+
+  const [_, utxos] = await getNewUtxosAtAddressAfterAction(
+    context.lucid,
+    treasuryAddr,
+    () => runAndAwaitTxBuilder(context.lucid, tx),
+  );
+
+  if (utxos.length !== Number(numberUtxos)) {
+    new Error('Expected a different number of new treasury UTxOs');
+  }
+
+  return utxos;
+}
+
+export async function createUtxoAtTreasury(
+  assets: Assets,
+  sysParams: SystemParams,
+  context: LucidContext,
+): Promise<UTxO> {
+  const utxos = await createMultipleUtxosAtTreasury(
+    assets,
+    1n,
+    sysParams,
+    context,
+  );
+
+  return matchSingle(utxos, () => new Error('Expected a single treasury UTXO'));
+}
+
+export async function createIndyUtxoAtTreasury(
+  indyAmt: bigint,
+  sysParams: SystemParams,
+  context: LucidContext,
+): Promise<UTxO> {
+  return createUtxoAtTreasury(
+    mkAssetsOf(fromSystemParamsAsset(sysParams.govParams.indyAsset), indyAmt),
+    sysParams,
+    context,
+  );
+}