fix tests

drift-labs · Dec 24, 2024 · 2d27e8d · 2d27e8d
1 parent b3891df
commit 2d27e8d
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 10 deletions.
diff --git a/sdk/src/idl/drift.json b/sdk/src/idl/drift.json
@@ -1508,7 +1508,7 @@
         },
         {
           "name": "authority",
-          "isMut": false,
+          "isMut": true,
           "isSigner": true
         }
       ],

diff --git a/tests/admin.ts b/tests/admin.ts
@@ -19,9 +19,11 @@ import {
 	mockUSDCMint,
 } from './testHelpers';
 import { PublicKey } from '@solana/web3.js';
-import { BankrunContextWrapper } from '../sdk/src/bankrun/bankrunConnection';
+import {
+	BankrunContextWrapper,
+	Connection,
+} from '../sdk/src/bankrun/bankrunConnection';
 import { TestBulkAccountLoader } from '../sdk/src/accounts/testBulkAccountLoader';
-import { checkIfAccountExists } from './placeAndMakeSwiftPerpBankrun';
 
 describe('admin', () => {
 	const chProgram = anchor.workspace.Drift as Program;
@@ -417,3 +419,16 @@ describe('admin', () => {
 		await driftClient.unsubscribe();
 	});
 });
+
+async function checkIfAccountExists(
+	connection: Connection,
+	account: PublicKey
+): Promise<boolean> {
+	try {
+		const accountInfo = await connection.getAccountInfo(account);
+		return accountInfo != null;
+	} catch (e) {
+		// Doesn't already exist
+		return false;
+	}
+}
diff --git a/tests/placeAndMakeSwiftPerpBankrun.ts b/tests/placeAndMakeSwiftPerpBankrun.ts
@@ -414,12 +414,13 @@ describe('place and make swift order', () => {
 		);
 
 		// Get pyth lazer instruction
-		const pythLazerCrankIxs = makerDriftClient.getPostPythLazerOracleUpdateIxs(
-			[6],
-			PYTH_LAZER_HEX_STRING_SOL,
-			undefined,
-			1
-		);
+		const pythLazerCrankIxs =
+			await makerDriftClient.getPostPythLazerOracleUpdateIxs(
+				[6],
+				PYTH_LAZER_HEX_STRING_SOL,
+				undefined,
+				1
+			);
 
 		const placeSwiftTakerOrderIxs =
 			await makerDriftClient.getPlaceSwiftTakerPerpOrderIxs(
@@ -1083,7 +1084,7 @@ function getSizeOfCompressedU16(n: number) {
 	return 1 + Number(n >= 128) + Number(n >= 16384);
 }
 
-export async function checkIfAccountExists(
+async function checkIfAccountExists(
 	connection: Connection,
 	account: PublicKey
 ): Promise<boolean> {