add ts tests

Author: Perelyn-sama

basics/transfer-sol/asm/tests/instruction.ts

@@ -0,0 +1,22 @@
+import { Buffer } from "buffer";
+import { type PublicKey, SystemProgram, TransactionInstruction } from "@solana/web3.js";
+
+export function createTransferInstruction(
+  senderPubkey: PublicKey,
+  recipientPubkey: PublicKey,
+  programId: PublicKey,
+  lamports: number,
+): TransactionInstruction {
+  const data = Buffer.alloc(8);
+  data.writeBigUInt64LE(BigInt(lamports));
+
+  return new TransactionInstruction({
+    keys: [
+      { pubkey: senderPubkey, isSigner: true, isWritable: true },
+      { pubkey: recipientPubkey, isSigner: false, isWritable: true },
+      { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
+    ],
+    programId,
+    data,
+  });
+}

basics/transfer-sol/asm/tests/test.ts

@@ -1,13 +1,43 @@
 import { describe, test } from "node:test";
-import {
-  Keypair,
-  LAMPORTS_PER_SOL,
-  PublicKey,
-  SystemProgram,
-  Transaction,
-} from "@solana/web3.js";
+import { Keypair, LAMPORTS_PER_SOL, PublicKey, Transaction } from "@solana/web3.js";
 import { start } from "solana-bankrun";
+import { createTransferInstruction } from "./instruction";
 
-describe("transfer-sol", async () => {
-  console.log("transfer-sol");
+describe("transfer-sol (asm)", async () => {
+  const PROGRAM_ID = PublicKey.unique();
+  const context = await start([{ name: "transfer-sol-cpi", programId: PROGRAM_ID }], []);
+  const client = context.banksClient;
+  const payer = context.payer;
+
+  const transferAmount = 1 * LAMPORTS_PER_SOL;
+  const recipient = Keypair.generate();
+
+  test("Transfer SOL via CPI to the system program", async () => {
+    await getBalances(payer.publicKey, recipient.publicKey, "Beginning");
+
+    const ix = createTransferInstruction(
+      payer.publicKey,
+      recipient.publicKey,
+      PROGRAM_ID,
+      transferAmount,
+    );
+
+    const tx = new Transaction();
+    const [blockhash, _] = await client.getLatestBlockhash();
+    tx.recentBlockhash = blockhash;
+    tx.add(ix).sign(payer);
+
+    await client.processTransaction(tx);
+
+    await getBalances(payer.publicKey, recipient.publicKey, "Resulting");
+  });
+
+  async function getBalances(payerPubkey: PublicKey, recipientPubkey: PublicKey, timeframe: string) {
+    const payerBalance = await client.getBalance(payerPubkey);
+    const recipientBalance = await client.getBalance(recipientPubkey);
+
+    console.log(`${timeframe} balances:`);
+    console.log(`   Payer: ${payerBalance}`);
+    console.log(`   Recipient: ${recipientBalance}`);
+  }
 });