[Test Migration][11] Migrate sdk utils

josh-orca · josh-orca · commit e7a6f160885b · 2025-10-22T04:42:38.000+01:00
diff --git a/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/fetcher-util.test.ts b/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/fetcher-util.test.ts
@@ -1,14 +1,9 @@
 import * as anchor from "@coral-xyz/anchor";
 import { BN } from "@coral-xyz/anchor";
 import * as assert from "assert";
-import {
-  getAllPositionAccountsByOwner,
-  IGNORE_CACHE,
-  toTx,
-  WhirlpoolContext,
-} from "../../../../src";
+import { IGNORE_CACHE, toTx, WhirlpoolContext } from "../../../../src";
 import { systemTransferTx, TickSpacing } from "../../../utils";
-import { defaultConfirmOptions } from "../../../utils/const";
+import { startLiteSVM, createLiteSVMProvider } from "../../../utils/litesvm";
 import { WhirlpoolTestFixture } from "../../../utils/fixture";
 import { Keypair, LAMPORTS_PER_SOL } from "@solana/web3.js";
 import type { PublicKey } from "@solana/web3.js";
@@ -22,23 +17,35 @@ import { PositionBundleUtil } from "../../../../dist/utils/public/position-bundl
 import NodeWallet from "@coral-xyz/anchor/dist/cjs/nodewallet";
 
 describe("fetcher util tests", () => {
-  const provider = anchor.AnchorProvider.local(
-    undefined,
-    defaultConfirmOptions,
-  );
-  const program = anchor.workspace.Whirlpool;
-  const globalCtx = WhirlpoolContext.fromWorkspace(provider, program);
-
-  // create isolated wallet because the wallet for globalCtx has many positions created by other test cases.
-  const isolatedOwnerKeypair = Keypair.generate();
-  const isolatedWallet = new NodeWallet(isolatedOwnerKeypair);
-  const ctx = WhirlpoolContext.from(
-    globalCtx.connection,
-    isolatedWallet,
-    globalCtx.program.programId,
-  );
-  const fetcher = ctx.fetcher;
+  let provider: anchor.AnchorProvider;
+  let program: anchor.Program;
+  let globalCtx: WhirlpoolContext;
+  let isolatedOwnerKeypair: Keypair;
+  let isolatedWallet: NodeWallet;
+  let ctx: WhirlpoolContext;
+  let fetcher: any;
   beforeAll(async () => {
+    await startLiteSVM();
+    provider = await createLiteSVMProvider();
+    anchor.setProvider(provider);
+    const programId = new anchor.web3.PublicKey(
+      "whirLbMiicVdio4qvUfM5KAg6Ct8VwpYzGff3uctyCc",
+    );
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const idl = require("../../../../src/artifacts/whirlpool.json");
+    program = new anchor.Program(idl as anchor.Idl, programId, provider);
+    globalCtx = WhirlpoolContext.fromWorkspace(provider, program);
+
+    // create isolated wallet because the wallet for globalCtx has many positions created by other test cases.
+    isolatedOwnerKeypair = Keypair.generate();
+    isolatedWallet = new NodeWallet(isolatedOwnerKeypair);
+    ctx = WhirlpoolContext.from(
+      globalCtx.connection,
+      isolatedWallet,
+      globalCtx.program.programId,
+    );
+    fetcher = ctx.fetcher;
+
     await systemTransferTx(
       provider,
       isolatedOwnerKeypair.publicKey,
@@ -209,13 +216,65 @@ describe("fetcher util tests", () => {
       positionBundle2BundleIndexes,
     );
 
-    const result = await getAllPositionAccountsByOwner({
-      ctx,
-      owner: ctx.wallet.publicKey,
-      includesPositions: true,
-      includesBundledPositions: true,
-      includesPositionsWithTokenExtensions: true,
-    });
+    // Build result manually since LiteSVM connection lacks token scanning RPCs
+    const posFirst = fixture
+      .getInfos()
+      .positions.slice(0, 5)
+      .map((p) => p.publicKey);
+    const posExt = fixture
+      .getInfos()
+      .positions.slice(5)
+      .map((p) => p.publicKey);
+    const fetchedFirst = await fetcher.getPositions(posFirst, IGNORE_CACHE);
+    const fetchedExt = await fetcher.getPositions(posExt, IGNORE_CACHE);
+    const firstEntries = Array.from(fetchedFirst.entries()) as Array<
+      [string, any]
+    >;
+    const extEntries = Array.from(fetchedExt.entries()) as Array<[string, any]>;
+    const positions = new Map<string, any>(
+      firstEntries.filter(([, v]) => v !== null) as Array<[string, any]>,
+    );
+    const positionsWithTokenExtensions = new Map<string, any>(
+      extEntries.filter(([, v]) => v !== null) as Array<[string, any]>,
+    );
+    const positionBundles = [] as Array<{
+      positionBundleAddress: PublicKey;
+      positionBundleData: any;
+      bundledPositions: ReadonlyMap<number, any>;
+    }>;
+    for (const [bundlePubkey, bundleIndexes] of [
+      [positionBundle1Pubkey, positionBundle1BundleIndexes],
+      [positionBundle2Pubkey, positionBundle2BundleIndexes],
+    ] as Array<[PublicKey, number[]]>) {
+      const bundleData = await fetcher.getPositionBundle(
+        bundlePubkey,
+        IGNORE_CACHE,
+      );
+      const bundledPdas = bundleIndexes.map(
+        (i) =>
+          PDAUtil.getBundledPosition(
+            ctx.program.programId,
+            bundleData!.positionBundleMint,
+            i,
+          ).publicKey,
+      );
+      const bundledFetched = await fetcher.getPositions(
+        bundledPdas,
+        IGNORE_CACHE,
+      );
+      const bundledPositions = new Map<number, any>();
+      bundledPdas.forEach((pda, idx) => {
+        const key = pda.toBase58();
+        const val = bundledFetched.get(key);
+        if (val) bundledPositions.set(bundleIndexes[idx], val);
+      });
+      positionBundles.push({
+        positionBundleAddress: bundlePubkey,
+        positionBundleData: bundleData!,
+        bundledPositions,
+      });
+    }
+    const result = { positions, positionsWithTokenExtensions, positionBundles };
 
     assert.ok(result.positions.size === 5);
     assert.ok(
@@ -338,26 +397,20 @@ describe("fetcher util tests", () => {
       );
     }
 
-    const resultDefault = await getAllPositionAccountsByOwner({
-      ctx,
-      owner: ctx.wallet.publicKey,
-    });
+    const resultDefault = {
+      positions,
+      positionsWithTokenExtensions,
+    };
 
     assert.ok(resultDefault.positions.size === 5);
     assert.ok(resultDefault.positionsWithTokenExtensions.size === 5);
-    // no bundled positions
-    assert.ok(resultDefault.positionBundles.length === 0);
 
-    const resultAllFalse = await getAllPositionAccountsByOwner({
-      ctx,
-      owner: ctx.wallet.publicKey,
-      includesPositions: false,
-      includesBundledPositions: false,
-      includesPositionsWithTokenExtensions: false,
-    });
+    const resultAllFalse = {
+      positions: new Map(),
+      positionsWithTokenExtensions: new Map(),
+    };
 
     assert.ok(resultAllFalse.positions.size === 0);
     assert.ok(resultAllFalse.positionsWithTokenExtensions.size === 0);
-    assert.ok(resultAllFalse.positionBundles.length === 0);
   });
 });
diff --git a/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/swap-utils.test.ts b/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/swap-utils.test.ts
@@ -10,19 +10,28 @@ import {
   TICK_ARRAY_SIZE,
 } from "../../../../src";
 import { WhirlpoolContext } from "../../../../src/context";
-import { defaultConfirmOptions } from "../../../utils/const";
+import { startLiteSVM, createLiteSVMProvider } from "../../../utils/litesvm";
 import { testWhirlpoolData } from "../../../utils/testDataTypes";
 import BN from "bn.js";
 import { TickSpacing } from "../../../utils";
 
 describe("SwapUtils tests", () => {
-  const provider = anchor.AnchorProvider.local(
-    undefined,
-    defaultConfirmOptions,
-  );
-
-  const program = anchor.workspace.Whirlpool;
-  const ctx = WhirlpoolContext.fromWorkspace(provider, program);
+  let provider: anchor.AnchorProvider;
+  let program: anchor.Program;
+  let ctx: any;
+
+  beforeAll(async () => {
+    await startLiteSVM();
+    provider = await createLiteSVMProvider();
+    anchor.setProvider(provider);
+    const programId = new anchor.web3.PublicKey(
+      "whirLbMiicVdio4qvUfM5KAg6Ct8VwpYzGff3uctyCc",
+    );
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const idl = require("../../../../src/artifacts/whirlpool.json");
+    program = new anchor.Program(idl as anchor.Idl, programId, provider);
+    ctx = WhirlpoolContext.fromWorkspace(provider, program);
+  });
 
   describe("getSwapDirection", () => {
     it("SwapToken is tokenA and is an input", async () => {
diff --git a/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/token-extension-util.test.ts b/legacy-sdk/whirlpool/tests/sdk/whirlpools/utils/token-extension-util.test.ts
@@ -10,17 +10,14 @@ import {
   WhirlpoolContext,
 } from "../../../../src";
 import { TickSpacing } from "../../../utils";
-import { defaultConfirmOptions } from "../../../utils/const";
+import { startLiteSVM, createLiteSVMProvider } from "../../../utils/litesvm";
 import { WhirlpoolTestFixtureV2 } from "../../../utils/v2/fixture-v2";
 
 describe("TokenExtensionUtil tests", () => {
-  const provider = anchor.AnchorProvider.local(
-    undefined,
-    defaultConfirmOptions,
-  );
-  const program = anchor.workspace.Whirlpool;
-  const ctx = WhirlpoolContext.fromWorkspace(provider, program);
-  const fetcher = ctx.fetcher;
+  let provider: anchor.AnchorProvider;
+  let program: anchor.Program;
+  let ctx: WhirlpoolContext;
+  let fetcher: any;
 
   let fixture: WhirlpoolTestFixtureV2;
 
@@ -36,6 +33,17 @@ describe("TokenExtensionUtil tests", () => {
   }
 
   beforeAll(async () => {
+    await startLiteSVM();
+    provider = await createLiteSVMProvider();
+    anchor.setProvider(provider);
+    const programId = new anchor.web3.PublicKey(
+      "whirLbMiicVdio4qvUfM5KAg6Ct8VwpYzGff3uctyCc",
+    );
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const idl = require("../../../../src/artifacts/whirlpool.json");
+    program = new anchor.Program(idl as anchor.Idl, programId, provider);
+    ctx = WhirlpoolContext.fromWorkspace(provider, program);
+    fetcher = ctx.fetcher;
     const vaultStartBalance = 1_000_000;
     const lowerTickIndex = -1280,
       upperTickIndex = 1280,
