import * as anchor from '@coral-xyz/anchor';
import { Program } from '@coral-xyz/anchor';
import { Keypair, LAMPORTS_PER_SOL, PublicKey } from '@solana/web3.js';
import { TransferSol } from '../target/types/transfer_sol';
describe('transfer-sol', () => {
  // Configure the client to use the local cluster.
  const provider = anchor.AnchorProvider.env();
  anchor.setProvider(provider);
  const program = anchor.workspace.TransferSol as Program<TransferSol>;
  // Generate new user keypairs for testing
  const user = Keypair.generate();
  const receiver = Keypair.generate();
  // Set the transfer amount to 1 SOL
  const transferAmount = 1 * LAMPORTS_PER_SOL;
  before(async () => {
    const latestBlockHash = await provider.connection.getLatestBlockhash();
    // Airdrop 5 SOL to the user that will send SOL to the other user
    const airdropUser = await provider.connection.requestAirdrop(user.publicKey, 5 * LAMPORTS_PER_SOL);
    await provider.connection.confirmTransaction({
      blockhash: latestBlockHash.blockhash,
      lastValidBlockHeight: latestBlockHash.lastValidBlockHeight,
      signature: airdropUser,
    });
  });
  it('Transfer SOL with CPI', async () => {
    await getBalances(user.publicKey, receiver.publicKey, '\n Beginning');
    // Transfer SOL instruction invoked from the program
    await program.methods
      .transferSolWithCpi(new anchor.BN(transferAmount))
      .accountsPartial({
        payer: user.publicKey,
        recipient: receiver.publicKey,
      })
      .signers([user])
      .rpc();
    await getBalances(user.publicKey, receiver.publicKey, '\n Resulting');
  });
  // Helper function to display balance of the accounts
  async function getBalances(payerPubkey: PublicKey, recipientPubkey: PublicKey, timeframe: string) {
    const payerBalance = await provider.connection.getBalance(payerPubkey);
    const recipientBalance = await provider.connection.getBalance(recipientPubkey);
    console.log(`${timeframe} balances:`);
    console.log(`   Payer: ${payerBalance / LAMPORTS_PER_SOL}`);
    console.log(`   Recipient: ${recipientBalance / LAMPORTS_PER_SOL}`);
  }
});