fix test to include transfer_fee calculation

jup-ag · Oct 23, 2024 · c412431 · c412431
1 parent 7ad2f3e
commit c412431
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/locker_utils/index.ts b/tests/locker_utils/index.ts
@@ -13,6 +13,7 @@ import {
   createAssociatedTokenAccountInstruction,
   getAssociatedTokenAddressSync,
   getEpochFee,
+  getMint,
   getTransferFeeConfig,
   TOKEN_2022_PROGRAM_ID,
   TOKEN_PROGRAM_ID,
@@ -586,7 +587,9 @@ export async function cancelVestingPlan(
     .signers([signer])
     .rpc();
 
-  const feeConfig = getTransferFeeConfig(escrowState.tokenMint);
+  const feeConfig = getTransferFeeConfig(
+    await getMint(program.provider.connection, escrowState.tokenMint)
+  );
   const epoch = BigInt(await getCurrentEpoch(program.provider.connection));
   const creator_fee = feeConfig
     ? Number(

diff --git a/tests/v2/test_cancel_with_harvest_fees.ts b/tests/v2/test_cancel_with_harvest_fees.ts
@@ -23,8 +23,6 @@ import { ADMIN, createMintTransaction } from "../locker_utils/token_2022/mint";
 
 const provider = anchor.AnchorProvider.env();
 provider.opts.commitment = "confirmed";
-const FEE = 100; // 100 BPS
-const MAX_FEE = 10000; 
 
 describe("[V2] Cancel, with transfer fees and harvesting", () => {
   let TOKEN: web3.PublicKey;