@carrot-protocol/clend-rpc 0.0.1-mrgn-fork1-dev-7be6ef2

package/src/mockJupiterUtils.ts

@@ -0,0 +1,109 @@
+import { web3 } from "@coral-xyz/anchor";
+import Decimal from "decimal.js";
+import { QuoteResponse, SwapMode } from "@jup-ag/api";
+import { IJupiterUtils, SwapInputs, SwapQuote, SwapIxs } from "./jupiterUtils";
+import { JupiterUtils } from "./jupiterUtils";
+
+/**
+ * MockJupiterUtils implements the IJupiterUtils interface for testing purposes.
+ * It allows for bypassing actual Jupiter API calls and returning predefined values.
+ */
+export class MockJupiterUtils implements IJupiterUtils {
+  private mockSwaps: Map<string, SwapIxs>;
+  private realJupiter: JupiterUtils;
+
+  constructor() {
+    this.mockSwaps = new Map();
+    this.realJupiter = new JupiterUtils();
+  }
+
+  /**
+   * Set mock swap instructions
+   * @param payer Payer public key
+   * @param inputs Swap inputs
+   * @param swapIxs Swap instructions to return
+   */
+  setMockSwap(
+    payer: web3.PublicKey,
+    inputs: SwapInputs,
+    swapIxs: SwapIxs,
+  ): void {
+    const key = this.getSwapKey(payer, inputs);
+    this.mockSwaps.set(key, swapIxs);
+  }
+
+  /**
+   * Get Jupiter price for a token pair
+   * @param inputMint Input token mint
+   * @param outputMint Output token mint
+   * @returns Mocked price for the token pair
+   */
+  async getJupiterPrice(mint: web3.PublicKey | string): Promise<number> {
+    // Use real Jupiter API for price
+    return this.realJupiter.getJupiterPrice(mint);
+  }
+
+  /**
+   * Get Jupiter quote for a swap
+   * @param inputMint Input token mint
+   * @param outputMint Output token mint
+   * @param inputMintDecimals Input token decimals
+   * @param outputMintDecimals Output token decimals
+   * @param inputAmountLamports Input amount in lamports
+   * @param slippageBps Slippage in basis points
+   * @param swapMode Swap mode
+   * @returns Mocked quote for the swap
+   */
+  async getJupiterQuote(
+    inputMint: web3.PublicKey,
+    outputMint: web3.PublicKey,
+    inputMintDecimals: number,
+    outputMintDecimals: number,
+    inputAmountLamports: Decimal,
+    slippageBps: number,
+    swapMode?: SwapMode,
+  ): Promise<SwapQuote<QuoteResponse>> {
+    // Use real Jupiter API for quote
+    return this.realJupiter.getJupiterQuote(
+      inputMint,
+      outputMint,
+      inputMintDecimals,
+      outputMintDecimals,
+      inputAmountLamports,
+      slippageBps,
+      swapMode,
+    );
+  }
+
+  /**
+   * Get Jupiter swap instructions
+   * @param payer Payer public key
+   * @param inputs Swap inputs
+   * @param quote Swap quote
+   * @returns Mocked swap instructions
+   */
+  async getJupiterSwap(
+    payer: web3.PublicKey,
+    inputs: SwapInputs,
+    quote: SwapQuote<QuoteResponse>,
+  ): Promise<SwapIxs> {
+    const key = this.getSwapKey(payer, inputs);
+    const swapIxs = this.mockSwaps.get(key);
+
+    if (!swapIxs) {
+      throw new Error(`Mock swap not set for ${key}`);
+    }
+
+    return swapIxs;
+  }
+
+  /**
+   * Generate a key for swap lookups
+   * @param payer Payer public key
+   * @param inputs Swap inputs
+   * @returns Key for the swap
+   */
+  private getSwapKey(payer: web3.PublicKey, inputs: SwapInputs): string {
+    return `${payer.toString()}-${inputs.inputMint.toString()}-${inputs.outputMint.toString()}`;
+  }
+}