All files / tests/utils token-balances.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
1x
 
 
1x
 
1x
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { expect } from "@jest/globals";
import { Connection, PublicKey } from "@solana/web3.js";
import BN from "bn.js";
import { AccountLayout } from "@solana/spl-token";
 
export const checkTokenBalances = async (
  connection: Connection,
  tokenAccount: PublicKey,
  expected: BN
) => {
  const raw = await connection.getAccountInfo(tokenAccount);
  Iif (!raw) {
    throw new Error(
      `Account ${tokenAccount.toBase58()} could not be retrieved`
    );
  }
  const des = AccountLayout.decode(raw.data);
  expect(des.amount.toString()).toBe(expected.toString());
};