tokens done

solana-developers · Aug 21, 2024 · 4a3a92b · 4a3a92b
1 parent a69fcb1
commit 4a3a92b
Show file tree

Hide file tree

Showing 5 changed files with 275 additions and 33 deletions.
diff --git a/tokens/token-2022/basics/anchor/tests/anchor.ts b/tokens/token-2022/basics/anchor/tests/anchor.ts
@@ -1,5 +1,6 @@
 import * as anchor from '@coral-xyz/anchor';
 import type { Program } from '@coral-xyz/anchor';
+import { sendAndConfirmTransaction } from '@solana/web3.js';
 import type { Anchor } from '../target/types/anchor';
 
 describe('anchor', () => {
@@ -45,7 +46,7 @@ describe('anchor', () => {
 
     tx.add(ix);
 
-    const sig = await anchor.web3.sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
+    const sig = await sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
     console.log('Your transaction signature', sig);
   });
 
@@ -64,7 +65,7 @@ describe('anchor', () => {
 
     tx.add(ix);
 
-    const sig = await anchor.web3.sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
+    const sig = await sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
     console.log('Your transaction signature', sig);
   });
 
@@ -108,7 +109,7 @@ describe('anchor', () => {
 
     tx.add(ix);
 
-    const sig = await anchor.web3.sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
+    const sig = await sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
     console.log('Your transaction signature', sig);
   });
 
@@ -130,7 +131,7 @@ describe('anchor', () => {
 
     tx.add(ix);
 
-    const sig = await anchor.web3.sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
+    const sig = await sendAndConfirmTransaction(program.provider.connection, tx, [wallet.payer]);
     console.log('Your transaction signature', sig);
   });
 });
diff --git a/tokens/token-2022/basics/anchor/tests/bankrun.test.ts b/tokens/token-2022/basics/anchor/tests/bankrun.test.ts
@@ -0,0 +1,149 @@
+import { describe, it } from 'node:test';
+import * as anchor from '@coral-xyz/anchor';
+import { PublicKey, sendAndConfirmTransaction } from '@solana/web3.js';
+import { BankrunProvider } from 'anchor-bankrun';
+import { startAnchor } from 'solana-bankrun';
+import type { Anchor } from '../target/types/anchor';
+
+const IDL = require('../target/idl/anchor.json');
+const PROGRAM_ID = new PublicKey(IDL.address);
+
+describe('anchor', async () => {
+  const context = await startAnchor('', [{ name: 'anchor', programId: PROGRAM_ID }], []);
+  const provider = new BankrunProvider(context);
+  anchor.setProvider(provider);
+  const program = new anchor.Program<Anchor>(IDL, provider);
+  const client = context.banksClient;
+  const TOKEN_2022_PROGRAM_ID = new anchor.web3.PublicKey('TokenzQdBNbLqP5VEhdkAS6EPFLC1PHnBqCXEpPxuEb');
+  const wallet = provider.wallet as anchor.Wallet;
+  const ATA_PROGRAM_ID = new anchor.web3.PublicKey('ATokenGPvbdGVxr1b2hvZbsiqW5xWH25efTNsLJA8knL');
+
+  const tokenName = 'TestToken';
+  const [mint] = anchor.web3.PublicKey.findProgramAddressSync(
+    [Buffer.from('token-2022-token'), wallet.publicKey.toBytes(), Buffer.from(tokenName)],
+    program.programId,
+  );
+  const [payerATA] = anchor.web3.PublicKey.findProgramAddressSync(
+    [wallet.publicKey.toBytes(), TOKEN_2022_PROGRAM_ID.toBytes(), mint.toBytes()],
+    ATA_PROGRAM_ID,
+  );
+
+  const receiver = anchor.web3.Keypair.generate();
+
+  const [receiverATA] = anchor.web3.PublicKey.findProgramAddressSync(
+    [receiver.publicKey.toBytes(), TOKEN_2022_PROGRAM_ID.toBytes(), mint.toBytes()],
+    ATA_PROGRAM_ID,
+  );
+
+  it('Create Token-2022 Token', async () => {
+    // await connection.requestAirdrop(receiver.publicKey, 1000000000);
+    // await connection.requestAirdrop(wallet.publicKey, 1000000000);
+    const tx = new anchor.web3.Transaction();
+    const [blockhash, _height] = await client.getLatestBlockhash();
+
+    const ix = await program.methods
+      .createToken(tokenName)
+      .accounts({
+        signer: wallet.publicKey,
+        tokenProgram: TOKEN_2022_PROGRAM_ID,
+      })
+      .instruction();
+
+    tx.recentBlockhash = blockhash;
+    tx.add(ix);
+    tx.sign(wallet.payer);
+    const sig = await client.processTransaction(tx);
+    console.log('Your transaction signature', sig);
+  });
+
+  it('Initialize payer ATA', async () => {
+    const tx = new anchor.web3.Transaction();
+    const [blockhash, _height] = await client.getLatestBlockhash();
+
+    const ix = await program.methods
+      .createAssociatedTokenAccount()
+      .accounts({
+        tokenAccount: payerATA,
+        mint: mint,
+        signer: wallet.publicKey,
+        tokenProgram: TOKEN_2022_PROGRAM_ID,
+      })
+      .instruction();
+
+    tx.recentBlockhash = blockhash;
+    tx.add(ix);
+    tx.sign(wallet.payer);
+    const sig = await client.processTransaction(tx);
+    console.log('Your transaction signature', sig);
+  });
+
+  /*
+  // This instruction is included only as a reference, but is not required to run this test, because we are using "init" in the program's transfer instruction. The create_associated_token_account instruction on the program is provided as a reference as well.
+  it("Initialize receiver ATA", async () => {
+    const tx = new anchor.web3.Transaction();
+    const ix = await program.methods
+      .createAssociatedTokenAccount()
+      .accounts({
+        tokenAccount: receiverATA,
+        mint: mint,
+        signer: receiver.publicKey,
+        tokenProgram: TOKEN_2022_PROGRAM_ID,
+        associatedTokenProgram: ATA_PROGRAM_ID,
+      })
+      .signers([receiver])
+      .instruction();
+    tx.add(ix);
+    const sig = await anchor.web3.sendAndConfirmTransaction(
+      program.provider.connection,
+      tx,
+      [receiver]
+    );
+    console.log("Your transaction signature", sig);
+  });
+*/
+
+  it('Mint Token to payer', async () => {
+    const tx = new anchor.web3.Transaction();
+    const [blockhash, _height] = await client.getLatestBlockhash();
+
+    const ix = await program.methods
+      .mintToken(new anchor.BN(200000000))
+      .accounts({
+        mint: mint,
+        signer: wallet.publicKey,
+        receiver: payerATA,
+        tokenProgram: TOKEN_2022_PROGRAM_ID,
+      })
+      .instruction();
+
+    tx.recentBlockhash = blockhash;
+    tx.add(ix);
+    tx.sign(wallet.payer);
+    const sig = await client.processTransaction(tx);
+    console.log('Your transaction signature', sig);
+  });
+
+  // Using init in the transfer instruction, as init if needed is bot working with Token 2022 yet.
+  it('Transfer Token', async () => {
+    const tx = new anchor.web3.Transaction();
+    const [blockhash, _height] = await client.getLatestBlockhash();
+
+    const ix = await program.methods
+      .transferToken(new anchor.BN(100))
+      .accounts({
+        mint: mint,
+        signer: wallet.publicKey,
+        from: payerATA,
+        to: receiver.publicKey,
+        tokenProgram: TOKEN_2022_PROGRAM_ID,
+        toAta: receiverATA,
+      })
+      .instruction();
+
+    tx.recentBlockhash = blockhash;
+    tx.add(ix);
+    tx.sign(wallet.payer);
+    const sig = await client.processTransaction(tx);
+    console.log('Your transaction signature', sig);
+  });
+});
diff --git a/tokens/token-2022/cpi-guard/anchor/package.json b/tokens/token-2022/cpi-guard/anchor/package.json
@@ -11,9 +11,11 @@
     "@types/bn.js": "^5.1.0",
     "@types/chai": "^4.3.0",
     "@types/mocha": "^9.0.0",
+    "anchor-bankrun": "^0.4.0",
     "chai": "^4.3.4",
     "mocha": "^9.0.3",
     "prettier": "^2.6.2",
+    "solana-bankrun": "^0.3.0",
     "ts-mocha": "^10.0.0",
     "typescript": "^4.3.5"
   }