import { describe, it } from "node:test";
import * as anchor from "@coral-xyz/anchor";
import { Keypair, PublicKey } from "@solana/web3.js";
import { BankrunProvider } from "anchor-bankrun";
import { startAnchor } from "solana-bankrun";
import type { CounterProgram } from "../target/types/counter_program";
import { expect } from "chai";

const IDL = require("../target/idl/counter_program.json");
const PROGRAM_ID = new PublicKey(IDL.address);

describe("counter_program", async () => {
  const context = await startAnchor(
    "",
    [{ name: "counter_program", programId: PROGRAM_ID }],
    []
  );
  const provider = new BankrunProvider(context);
  const payer = provider.wallet as anchor.Wallet;
  const program = new anchor.Program<CounterProgram>(IDL, provider);

  const [stateAccount, _] = anchor.web3.PublicKey.findProgramAddressSync(
    [anchor.utils.bytes.utf8.encode("count")],
    program.programId
  );

  it("Initialize the counter", async () => {
    await program.methods
      .initialize()
      .accounts({
        state: stateAccount as PublicKey,
        user: payer.publicKey,
      })
      .signers([
        { publicKey: payer.publicKey, secretKey: payer.payer.secretKey },
      ])
      .rpc();

    const account = await program.account.counterState.fetch(stateAccount);
    console.log("Counter after initialization:", account.count.toString());
    // Expecting the count to be 0
    expect(account.count.toString()).to.equal("0");
  });

  it("Increment the counter", async () => {
    await program.methods
      .increment()
      .accounts({
        state: stateAccount,
      })
      .rpc();

    const account = await program.account.counterState.fetch(stateAccount);
    console.log("Counter after increment:", account.count.toString());
    expect(account.count.toString()).to.equal("1");
  });
});