solana-program-examples/tokens/token-fundraiser/anchor/tests/fundraiser.ts

import * as anchor from '@coral-xyz/anchor';
import type { Program } from '@coral-xyz/anchor';
import type NodeWallet from '@coral-xyz/anchor/dist/cjs/nodewallet';
import {
  ASSOCIATED_TOKEN_PROGRAM_ID,
  TOKEN_PROGRAM_ID,
  createMint,
  getAssociatedTokenAddressSync,
  getOrCreateAssociatedTokenAccount,
  mintTo,
} from '@solana/spl-token';
import type { Fundraiser } from '../target/types/fundraiser';

describe('fundraiser', () => {
  // Configure the client to use the local cluster.
  const provider = anchor.AnchorProvider.env();
  anchor.setProvider(provider);

  const program = anchor.workspace.Fundraiser as Program<Fundraiser>;

  const maker = anchor.web3.Keypair.generate();

  let mint: anchor.web3.PublicKey;

  let contributorATA: anchor.web3.PublicKey;

  let makerATA: anchor.web3.PublicKey;

  const wallet = provider.wallet as NodeWallet;

  const fundraiser = anchor.web3.PublicKey.findProgramAddressSync([Buffer.from('fundraiser'), maker.publicKey.toBuffer()], program.programId)[0];

  const contributor = anchor.web3.PublicKey.findProgramAddressSync(
    [Buffer.from('contributor'), fundraiser.toBuffer(), provider.publicKey.toBuffer()],
    program.programId,
  )[0];

  const confirm = async (signature: string): Promise<string> => {
    const block = await provider.connection.getLatestBlockhash();
    await provider.connection.confirmTransaction({
      signature,
      ...block,
    });
    return signature;
  };

  it('Test Preparation', async () => {
    const airdrop = await provider.connection.requestAirdrop(maker.publicKey, 1 * anchor.web3.LAMPORTS_PER_SOL).then(confirm);
    console.log('\nAirdropped 1 SOL to maker', airdrop);

    mint = await createMint(provider.connection, wallet.payer, provider.publicKey, provider.publicKey, 6);
    console.log('Mint created', mint.toBase58());

    contributorATA = (await getOrCreateAssociatedTokenAccount(provider.connection, wallet.payer, mint, wallet.publicKey)).address;

    makerATA = (await getOrCreateAssociatedTokenAccount(provider.connection, wallet.payer, mint, maker.publicKey)).address;

    const mintTx = await mintTo(provider.connection, wallet.payer, mint, contributorATA, provider.publicKey, 1_000_000_0);
    console.log('Minted 10 tokens to contributor', mintTx);
  });

  it('Initialize Fundaraiser', async () => {
    const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

    const tx = await program.methods
      .initialize(new anchor.BN(30000000), 0)
      .accountsPartial({
        maker: maker.publicKey,
        fundraiser,
        mintToRaise: mint,
        vault,
        systemProgram: anchor.web3.SystemProgram.programId,
        tokenProgram: TOKEN_PROGRAM_ID,
        associatedTokenProgram: ASSOCIATED_TOKEN_PROGRAM_ID,
      })
      .signers([maker])
      .rpc()
      .then(confirm);

    console.log('\nInitialized fundraiser Account');
    console.log('Your transaction signature', tx);
  });

  it('Contribute to Fundraiser', async () => {
    const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

    const tx = await program.methods
      .contribute(new anchor.BN(1000000))
      .accountsPartial({
        contributor: provider.publicKey,
        fundraiser,
        contributorAccount: contributor,
        contributorAta: contributorATA,
        vault,
        tokenProgram: TOKEN_PROGRAM_ID,
      })
      .rpc()
      .then(confirm);

    console.log('\nContributed to fundraiser', tx);
    console.log('Your transaction signature', tx);
    console.log('Vault balance', (await provider.connection.getTokenAccountBalance(vault)).value.amount);

    const contributorAccount = await program.account.contributor.fetch(contributor);
    console.log('Contributor balance', contributorAccount.amount.toString());
  });
  it('Contribute to Fundraiser', async () => {
    const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

    const tx = await program.methods
      .contribute(new anchor.BN(1000000))
      .accountsPartial({
        contributor: provider.publicKey,
        fundraiser,
        contributorAccount: contributor,
        contributorAta: contributorATA,
        vault,
        tokenProgram: TOKEN_PROGRAM_ID,
      })
      .rpc()
      .then(confirm);

    console.log('\nContributed to fundraiser', tx);
    console.log('Your transaction signature', tx);
    console.log('Vault balance', (await provider.connection.getTokenAccountBalance(vault)).value.amount);

    const contributorAccount = await program.account.contributor.fetch(contributor);
    console.log('Contributor balance', contributorAccount.amount.toString());
  });

  it('Contribute to Fundraiser - Robustness Test', async () => {
    try {
      const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

      const tx = await program.methods
        .contribute(new anchor.BN(2000000))
        .accountsPartial({
          contributor: provider.publicKey,
          fundraiser,
          contributorAccount: contributor,
          contributorAta: contributorATA,
          vault,
          tokenProgram: TOKEN_PROGRAM_ID,
        })
        .rpc()
        .then(confirm);

      console.log('\nContributed to fundraiser', tx);
      console.log('Your transaction signature', tx);
      console.log('Vault balance', (await provider.connection.getTokenAccountBalance(vault)).value.amount);
    } catch (error) {
      console.log('\nError contributing to fundraiser');
      console.log(error.msg);
    }
  });

  it('Check contributions - Robustness Test', async () => {
    try {
      const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

      const tx = await program.methods
        .checkContributions()
        .accountsPartial({
          maker: maker.publicKey,
          mintToRaise: mint,
          fundraiser,
          makerAta: makerATA,
          vault,
          tokenProgram: TOKEN_PROGRAM_ID,
        })
        .signers([maker])
        .rpc()
        .then(confirm);

      console.log('\nChecked contributions');
      console.log('Your transaction signature', tx);
      console.log('Vault balance', (await provider.connection.getTokenAccountBalance(vault)).value.amount);
    } catch (error) {
      console.log('\nError checking contributions');
      console.log(error.msg);
    }
  });

  it('Refund Contributions', async () => {
    const vault = getAssociatedTokenAddressSync(mint, fundraiser, true);

    const contributorAccount = await program.account.contributor.fetch(contributor);
    console.log('\nContributor balance', contributorAccount.amount.toString());

    const tx = await program.methods
      .refund()
      .accountsPartial({
        contributor: provider.publicKey,
        maker: maker.publicKey,
        mintToRaise: mint,
        fundraiser,
        contributorAccount: contributor,
        contributorAta: contributorATA,
        vault,
        tokenProgram: TOKEN_PROGRAM_ID,
        systemProgram: anchor.web3.SystemProgram.programId,
      })
      .rpc()
      .then(confirm);

    console.log('\nRefunded contributions', tx);
    console.log('Your transaction signature', tx);
    console.log('Vault balance', (await provider.connection.getTokenAccountBalance(vault)).value.amount);
  });
});