add test

Author: crispheaney

programs/drift/src/instructions/if_staker.rs

@@ -821,7 +821,6 @@ pub fn handle_transfer_protocol_if_shares_to_revenue_pool<'c: 'info, 'info>(
     Ok(())
 }
 
-
 pub fn handle_deposit_into_insurance_fund_stake<'c: 'info, 'info>(
     ctx: Context<'_, '_, 'c, 'info, DepositIntoInsuranceFundStake<'info>>,
     market_index: u16,
@@ -1221,6 +1220,7 @@ pub struct DepositIntoInsuranceFundStake<'info> {
     )]
     pub spot_market_vault: Box<InterfaceAccount<'info, TokenAccount>>,
     #[account(
+        mut,
         seeds = [b"insurance_fund_vault".as_ref(), market_index.to_le_bytes().as_ref()],
         bump,
     )]
@@ -1232,6 +1232,6 @@ pub struct DepositIntoInsuranceFundStake<'info> {
     )]
     pub user_token_account: Box<InterfaceAccount<'info, TokenAccount>>,
     pub token_program: Interface<'info, TokenInterface>,
-        /// CHECK: forced drift_signer
-        pub drift_signer: AccountInfo<'info>,
+    /// CHECK: forced drift_signer
+    pub drift_signer: AccountInfo<'info>,
 }

sdk/src/driftClient.ts

@@ -9277,6 +9277,56 @@ export class DriftClient {
 		return ix;
 	}
 
+	public async depositIntoInsuranceFundStake(
+		marketIndex: number,
+		amount: BN,
+		userStatsPublicKey: PublicKey,
+		insuranceFundStakePublicKey: PublicKey,
+		userTokenAccountPublicKey: PublicKey,
+		txParams?: TxParams
+	): Promise<TransactionSignature> {
+		const tx = await this.buildTransaction(
+			await this.getDepositIntoInsuranceFundStakeIx(
+				marketIndex,
+				amount,
+				userStatsPublicKey,
+				insuranceFundStakePublicKey,
+				userTokenAccountPublicKey
+			),
+			txParams
+		);
+		const { txSig } = await this.sendTransaction(tx, [], this.opts);
+		return txSig;
+	}
+
+	public async getDepositIntoInsuranceFundStakeIx(
+		marketIndex: number,
+		amount: BN,
+		userStatsPublicKey: PublicKey,
+		insuranceFundStakePublicKey: PublicKey,
+		userTokenAccountPublicKey: PublicKey
+	): Promise<TransactionInstruction> {
+		const spotMarket = this.getSpotMarketAccount(marketIndex);
+		return await this.program.instruction.depositIntoInsuranceFundStake(
+			marketIndex,
+			amount,
+			{
+				accounts: {
+					signer: this.wallet.publicKey,
+					state: await this.getStatePublicKey(),
+					spotMarket: spotMarket.pubkey,
+					insuranceFundStake: insuranceFundStakePublicKey,
+					userStats: userStatsPublicKey,
+					spotMarketVault: spotMarket.vault,
+					insuranceFundVault: spotMarket.insuranceFund.vault,
+					userTokenAccount: userTokenAccountPublicKey,
+					tokenProgram: this.getTokenProgramForSpotMarket(spotMarket),
+					driftSigner: this.getSignerPublicKey(),
+				},
+			}
+		);
+	}
+
 	public async settleRevenueToInsuranceFund(
 		spotMarketIndex: number,
 		txParams?: TxParams

sdk/src/idl/drift.json

@@ -3141,6 +3141,42 @@
       ],
       "args": []
     },
+    {
+      "name": "updateDelegateUserGovTokenInsuranceStake",
+      "accounts": [
+        {
+          "name": "spotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "userStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "admin",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "state",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": []
+    },
     {
       "name": "initializeInsuranceFundStake",
       "accounts": [
@@ -3634,6 +3670,71 @@
         }
       ]
     },
+    {
+      "name": "depositIntoInsuranceFundStake",
+      "accounts": [
+        {
+          "name": "signer",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "state",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "spotMarket",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundStake",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "userStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "spotMarketVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "insuranceFundVault",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "userTokenAccount",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "tokenProgram",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "driftSigner",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "marketIndex",
+          "type": "u16"
+        },
+        {
+          "name": "amount",
+          "type": "u64"
+        }
+      ]
+    },
     {
       "name": "updatePythPullOracle",
       "accounts": [
@@ -16218,5 +16319,8 @@
       "name": "InvalidIfRebalanceSwap",
       "msg": "Invalid If Rebalance Swap"
     }
-  ]
+  ],
+  "metadata": {
+    "address": "dRiftyHA39MWEi3m9aunc5MzRF1JYuBsbn6VPcn33UH"
+  }
 }

tests/insuranceFundStake.ts

@@ -29,6 +29,7 @@ import {
 	unstakeSharesToAmount,
 	MarketStatus,
 	LIQUIDATION_PCT_PRECISION,
+	getUserStatsAccountPublicKey,
 } from '../sdk/src';
 
 import {
@@ -40,6 +41,7 @@ import {
 	sleep,
 	mockOracleNoProgram,
 	setFeedPriceNoProgram,
+	mintUSDCToUser,
 } from './testHelpers';
 import { ContractTier, PERCENTAGE_PRECISION, UserStatus } from '../sdk';
 import { startAnchor } from 'solana-bankrun';
@@ -1163,6 +1165,33 @@ describe('insurance fund stake', () => {
 		// assert(usdcBefore.eq(usdcAfter));
 	});
 
+	it('admin deposit into insurance fund stake', async () => {
+		await mintUSDCToUser(
+			usdcMint,
+			userUSDCAccount.publicKey,
+			usdcAmount,
+			bankrunContextWrapper
+		);
+		const marketIndex = 0;
+		const insuranceFundStakePublicKey = getInsuranceFundStakeAccountPublicKey(
+			driftClient.program.programId,
+			driftClient.wallet.publicKey,
+			marketIndex
+		);
+		const userStatsPublicKey = getUserStatsAccountPublicKey(
+			driftClient.program.programId,
+			driftClient.wallet.publicKey
+		);
+		const txSig = await driftClient.depositIntoInsuranceFundStake(
+			marketIndex,
+			usdcAmount,
+			userStatsPublicKey,
+			insuranceFundStakePublicKey,
+			userUSDCAccount.publicKey
+		);
+		bankrunContextWrapper.printTxLogs(txSig);
+	});
+
 	// it('settle spotMarket to insurance vault', async () => {
 	// 	const marketIndex = new BN(0);