deposit through example program (#105)

Author: skosito

Cargo.lock

@@ -20,3 +20,4 @@ idl-build = ["anchor-lang/idl-build", "anchor-spl/idl-build"]
 anchor-lang = { version = "=0.31.1" }
 anchor-spl = { version = "=0.31.1" }
 spl-associated-token-account = { version = "6.0.0", features = ["no-entrypoint"] }
+gateway = { path = "../../gateway", features = ["no-entrypoint", "cpi"] }

programs/examples/connected/Cargo.toml

@@ -84,6 +84,28 @@ pub mod connected {
 
         Ok(())
     }
+
+    pub fn trigger_deposit(
+        ctx: Context<TriggerDeposit>,
+        amount: u64,
+        receiver: [u8; 20],
+        revert_options: Option<gateway::RevertOptions>,
+    ) -> Result<()> {
+        let gateway_program = ctx.accounts.gateway_program.to_account_info();
+
+        let cpi_accounts = gateway::cpi::accounts::Deposit {
+            signer: ctx.accounts.signer.to_account_info(),
+            pda: ctx.accounts.gateway_pda.to_account_info(),
+            system_program: ctx.accounts.system_program.to_account_info(),
+        };
+
+        let cpi_program = gateway_program;
+        let cpi_ctx = CpiContext::new(cpi_program, cpi_accounts);
+
+        gateway::cpi::deposit(cpi_ctx, amount, receiver, revert_options)?;
+
+        Ok(())
+    }
 }
 
 #[derive(Accounts)]
@@ -122,6 +144,21 @@ pub struct OnRevert<'info> {
     pub system_program: Program<'info, System>,
 }
 
+#[derive(Accounts)]
+pub struct TriggerDeposit<'info> {
+    #[account(mut)]
+    pub signer: Signer<'info>,
+
+    #[account(mut)]
+    /// CHECK: Validated by the gateway program via seeds
+    pub gateway_pda: UncheckedAccount<'info>,
+
+    /// CHECK: Only used for CPI
+    pub gateway_program: UncheckedAccount<'info>,
+
+    pub system_program: Program<'info, System>,
+}
+
 #[account]
 pub struct Pda {
     pub last_sender: [u8; 20],

programs/examples/connected/src/lib.rs

@@ -535,13 +535,33 @@ describe("Gateway", () => {
     }
   });
 
+  it("Deposit through connected program", async () => {
+    const balanceBefore = await conn.getBalance(pdaAccount);
+    await connectedProgram.methods
+      .triggerDeposit(
+        new anchor.BN(1_000_000_000),
+        Array.from(address),
+        revertOptions
+      )
+      .accounts({
+        gatewayPda: pdaAccount,
+        gatewayProgram: gatewayProgram.programId,
+      })
+      .rpc();
+
+    const balanceAfter = await conn.getBalance(pdaAccount);
+    // amount + deposit fee
+    expect(balanceAfter - balanceBefore).to.eq(1_000_000_000 + 2_000_000);
+  });
+
   it("Deposit and withdraw 0.5 SOL from Gateway with ECDSA signature", async () => {
+    const balanceBefore = await conn.getBalance(pdaAccount);
     await gatewayProgram.methods
       .deposit(new anchor.BN(1_000_000_000), Array.from(address), revertOptions)
       .rpc();
-    let bal1 = await conn.getBalance(pdaAccount);
+    let balanceAfter = await conn.getBalance(pdaAccount);
     // amount + deposit fee
-    expect(bal1).to.be.gte(1_000_000_000 + 2_000_000);
+    expect(balanceAfter - balanceBefore).to.eq(1_000_000_000 + 2_000_000);
     const pdaAccountData = await gatewayProgram.account.pda.fetch(pdaAccount);
     const nonce = pdaAccountData.nonce;
     const amount = new anchor.BN(500000000);
@@ -578,7 +598,7 @@ describe("Gateway", () => {
       })
       .rpc();
     let bal2 = await conn.getBalance(pdaAccount);
-    expect(bal2).to.be.eq(bal1 - 500_000_000);
+    expect(bal2).to.be.eq(balanceAfter - 500_000_000);
     let bal3 = await conn.getBalance(to);
     expect(bal3).to.be.gte(500_000_000);
   });