@drift-labs/sdk 2.85.0-beta.1 → 2.85.0-beta.10

package/src/testClient.ts

@@ -10,8 +10,6 @@ export class TestClient extends AdminClient {
 			throw new Error('Test client must be polling');
 		}
 		super(config);
-		// @ts-ignore
-		this.txHandler.blockhashCommitment = 'recent';
 	}
 
 	async sendTransaction(
@@ -30,6 +28,7 @@ export class TestClient extends AdminClient {
 		let lastFetchedSlot = (
 			this.accountSubscriber as PollingDriftClientAccountSubscriber
 		).accountLoader.mostRecentSlot;
+		await this.fetchAccounts();
 		while (lastFetchedSlot < slot) {
 			await this.fetchAccounts();
 			lastFetchedSlot = (

package/src/tokenFaucet.ts

@@ -2,10 +2,10 @@ import * as anchor from '@coral-xyz/anchor';
 import { AnchorProvider, Idl, Program } from '@coral-xyz/anchor';
 import {
 	TOKEN_PROGRAM_ID,
-	getAccount,
 	Account,
 	createAssociatedTokenAccountInstruction,
 	getAssociatedTokenAddress,
+	getAccount,
 } from '@solana/spl-token';
 import {
 	ConfirmOptions,
@@ -19,8 +19,10 @@ import {
 import { BN } from '.';
 import tokenFaucet from './idl/token_faucet.json';
 import { IWallet } from './types';
+import { BankrunContextWrapper } from './bankrun/bankrunConnection';
 
 export class TokenFaucet {
+	context?: BankrunContextWrapper;
 	connection: Connection;
 	wallet: IWallet;
 	public program: Program;
@@ -33,13 +35,20 @@ export class TokenFaucet {
 		wallet: IWallet,
 		programId: PublicKey,
 		mint: PublicKey,
-		opts?: ConfirmOptions
+		opts?: ConfirmOptions,
+		context?: BankrunContextWrapper
 	) {
 		this.connection = connection;
+		this.context = context;
 		this.wallet = wallet;
 		this.opts = opts || AnchorProvider.defaultOptions();
 		// @ts-ignore
-		const provider = new AnchorProvider(connection, wallet, this.opts);
+		const provider = new AnchorProvider(
+			context ? context.connection.toConnection() : this.connection,
+			// @ts-ignore
+			wallet,
+			this.opts
+		);
 		this.provider = provider;
 		this.program = new Program(tokenFaucet as Idl, programId, provider);
 		this.mint = mint;
@@ -76,7 +85,7 @@ export class TokenFaucet {
 	public async initialize(): Promise<TransactionSignature> {
 		const [faucetConfigPublicKey] =
 			await this.getFaucetConfigPublicKeyAndNonce();
-		return await this.program.rpc.initialize({
+		const ix = this.program.instruction.initialize({
 			accounts: {
 				faucetConfig: faucetConfigPublicKey,
 				admin: this.wallet.publicKey,
@@ -86,6 +95,9 @@ export class TokenFaucet {
 				tokenProgram: TOKEN_PROGRAM_ID,
 			},
 		});
+		const tx = new Transaction().add(ix);
+		const txSig = await this.context.sendTransaction(tx);
+		return txSig;
 	}
 
 	public async fetchState(): Promise<any> {
@@ -114,12 +126,29 @@ export class TokenFaucet {
 
 		const tx = new Transaction().add(mintIx);
 
-		const txSig = await this.program.provider.sendAndConfirm(tx, [], this.opts);
-
-		return txSig;
+		if (this.context) {
+			return await this.context.sendTransaction(tx);
+		} else {
+			return await this.program.provider.sendAndConfirm(tx, [], this.opts);
+		}
 	}
 
 	public async transferMintAuthority(): Promise<TransactionSignature> {
+		if (this.context) {
+			const ix = this.program.instruction.transferMintAuthority({
+				accounts: {
+					faucetConfig: await this.getFaucetConfigPublicKey(),
+					mintAccount: this.mint,
+					mintAuthority: await this.getMintAuthority(),
+					tokenProgram: TOKEN_PROGRAM_ID,
+					admin: this.wallet.publicKey,
+				},
+			});
+			const tx = new Transaction().add(ix);
+			const txSig = await this.context.sendTransaction(tx);
+			return txSig;
+		}
+
 		return await this.program.rpc.transferMintAuthority({
 			accounts: {
 				faucetConfig: await this.getFaucetConfigPublicKey(),
@@ -146,9 +175,8 @@ export class TokenFaucet {
 		let associatedTokenAccountExists = false;
 
 		try {
-			const assosciatedTokenAccount = await this.connection.getAccountInfo(
-				associatedTokenPublicKey
-			);
+			const assosciatedTokenAccount =
+				await this.context.connection.getAccountInfo(associatedTokenPublicKey);
 
 			associatedTokenAccountExists = !!assosciatedTokenAccount;
 		} catch (e) {
@@ -162,7 +190,13 @@ export class TokenFaucet {
 
 		tx.add(mintToTx);
 
-		const txSig = await this.program.provider.sendAndConfirm(tx, [], this.opts);
+		let txSig;
+		if (this.context) {
+			txSig = await this.context.sendTransaction(tx);
+		} else {
+			txSig = await this.program.provider.sendAndConfirm(tx, [], this.opts);
+		}
+
 		return [associatedTokenPublicKey, txSig];
 	}
 
@@ -200,6 +234,9 @@ export class TokenFaucet {
 		userPubKey: PublicKey;
 	}): Promise<Account> {
 		const associatedKey = await this.getAssosciatedMockUSDMintAddress(props);
+		if (this.context) {
+			return await this.context.connection.getTokenAccount(associatedKey);
+		}
 		return await getAccount(this.connection, associatedKey);
 	}
 
@@ -215,7 +252,7 @@ export class TokenFaucet {
 			props.callback(await this.getTokenAccountInfo(props));
 
 			// Couldn't find a way to do it using anchor framework subscription, someone on serum discord recommended this way
-			this.connection.onAccountChange(
+			this.context.connection.onAccountChange(
 				tokenAccountKey,
 				async (
 					_accountInfo /* accountInfo is a buffer which we don't know how to deserialize */

package/src/tx/txParamProcessor.ts

@@ -32,7 +32,8 @@ export class TransactionParamProcessor {
 	public static async getTxSimComputeUnits(
 		tx: VersionedTransaction,
 		connection: Connection,
-		bufferMultiplier: number // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+		bufferMultiplier: number, // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+		lowerBoundCu?: number
 	): Promise<{ success: boolean; computeUnits: number }> {
 		try {
 			if (TEST_SIMS_ALWAYS_FAIL)
@@ -49,10 +50,18 @@ export class TransactionParamProcessor {
 			const computeUnits = await this.getComputeUnitsFromSim(simTxResult);
 
 			// Apply the buffer, but round down to the MAX_COMPUTE_UNITS, and round up to the nearest whole number
-			const bufferedComputeUnits = Math.ceil(
+			let bufferedComputeUnits = Math.ceil(
 				Math.min(computeUnits * bufferMultiplier, MAX_COMPUTE_UNITS)
 			);
 
+			// If a lower bound CU is passed then enforce it
+			if (lowerBoundCu) {
+				bufferedComputeUnits = Math.max(
+					bufferedComputeUnits,
+					Math.min(lowerBoundCu, MAX_COMPUTE_UNITS)
+				);
+			}
+
 			return {
 				success: true,
 				computeUnits: bufferedComputeUnits,

package/src/types.ts

@@ -1042,6 +1042,7 @@ export type ProcessingTxParams = {
 	computeUnitsBufferMultiplier?: number;
 	useSimulatedComputeUnitsForCUPriceCalculation?: boolean;
 	getCUPriceFromComputeUnits?: (computeUnits: number) => number;
+	lowerBoundCu?: number;
 };
 
 export type TxParams = BaseTxParams & ProcessingTxParams;

package/src/user.ts

@@ -108,9 +108,12 @@ export class User {
 		this.userAccountPublicKey = config.userAccountPublicKey;
 		if (config.accountSubscription?.type === 'polling') {
 			this.accountSubscriber = new PollingUserAccountSubscriber(
-				config.driftClient.program,
+				config.driftClient.connection,
 				config.userAccountPublicKey,
-				config.accountSubscription.accountLoader
+				config.accountSubscription.accountLoader,
+				this.driftClient.program.account.user.coder.accounts.decodeUnchecked.bind(
+					this.driftClient.program.account.user.coder.accounts
+				)
 			);
 		} else if (config.accountSubscription?.type === 'custom') {
 			this.accountSubscriber = config.accountSubscription.userAccountSubscriber;