In tests, only create token account if needed (#99)

orca-so · Apr 24, 2023 · bf60726 · bf60726
1 parent b21c82e
commit bf60726
Showing 1 changed file with 19 additions and 6 deletions.
diff --git a/sdk/tests/utils/token.ts b/sdk/tests/utils/token.ts
@@ -176,12 +176,25 @@ export async function createAndMintToAssociatedTokenAccount(
     return tokenAccount;
   }
 
-  const tokenAccount = await createAssociatedTokenAccount(
-    provider,
-    mint,
-    destinationWalletKey,
-    payerKey
-  );
+  const tokenAccounts = await provider.connection.getParsedTokenAccountsByOwner(destinationWalletKey, {
+    programId: TOKEN_PROGRAM_ID,
+  });
+
+  let tokenAccount = tokenAccounts.value.map((account) => {
+    if(account.account.data.parsed.info.mint === mint.toString()) {
+      return account.pubkey
+    }
+  }).filter(Boolean)[0];
+
+  if(!tokenAccount) {
+    tokenAccount = await createAssociatedTokenAccount(
+      provider,
+      mint,
+      destinationWalletKey,
+      payerKey
+    );
+  }
+
   await mintToByAuthority(provider, mint, tokenAccount, new u64(amount.toString()));
   return tokenAccount;
 }