secret-polar-reworks/src/lib/chai/matchers/changeScrtBalance.ts

Polar is a development environment to compile, deploy, test, run scrt contracts on different networks.

import chalk from "chalk";
import { Account as WasmAccount } from "secretjs";

import { PolarContext } from "../../../internal/context";
import { PolarError } from "../../../internal/core/errors";
import { ERRORS } from "../../../internal/core/errors-list";
import type {
  Account, Coin, UserAccount
} from "../../../types";
import { getClient } from "../../client";
import { defaultFees } from "../../contants";

export function supportChangeScrtBalance (Assertion: Chai.AssertionStatic): void {
  Assertion.addMethod('changeScrtBalance', function (
    this: any, // eslint-disable-line  @typescript-eslint/no-explicit-any
    account: UserAccount | Account | string,
    balanceChange: number,
    includeFee?: boolean,
    logResponse?: boolean
  ) {
    const subject = this._obj;

    if ((account as UserAccount).account !== undefined) {
      account = (account as UserAccount).account;
    }

    const accountAddr: string = (account as Account).address !== undefined
      ? (account as Account).address : (account as string);
    const derivedPromise = Promise.all([
      getBalanceChange(subject, accountAddr, includeFee, logResponse)
    ]).then(([actualChange]) => {
      this.assert(
        actualChange === balanceChange,
        `Expected "${accountAddr}" to change balance by ${balanceChange} uscrt, ` +
          `but it has changed by ${actualChange} uscrt`,
        `Expected "${accountAddr}" to not change balance by ${balanceChange} uscrt,`,
        balanceChange,
        actualChange
      );
    });

    this.then = derivedPromise.then.bind(derivedPromise);
    this.catch = derivedPromise.catch.bind(derivedPromise);
    this.promise = derivedPromise;
    return this;
  });
}

function extractScrtBalance (
  balances: readonly Coin[]
): number {
  console.log(balances);
  for (const coin of balances) {
    if (coin.denom === 'uscrt') {
      return Number(coin.amount);
    }
  }
  return 0;
}

export async function getBalanceChange ( // eslint-disable-line sonarjs/cognitive-complexity
  transaction: (() => Promise<any>), // eslint-disable-line  @typescript-eslint/no-explicit-any
  accountAddr: string,
  includeFee?: boolean,
  logResponse?: boolean
): Promise<number> {
  if (typeof transaction !== 'function') {
    throw new PolarError(ERRORS.GENERAL.NOT_A_FUNCTION, {
      param: transaction
    });
  }

  const client = getClient(PolarContext.getPolarContext().getRuntimeEnv().network);

  const balanceBefore = extractScrtBalance(
    (await client.getAccount(accountAddr) as WasmAccount).balance
  );

  const txResponse = await transaction();
  if (logResponse === true) {
    console.log(`${chalk.green("Transaction response:")} ${txResponse as string}`);
  }
  const txnEvents = txResponse.logs[0].events;
  let msgEvent;
  for (const event of txnEvents) {
    if (event.type === 'message') {
      msgEvent = event;
      break;
    }
  }
  const msgEventKeys: { [key: string]: string } = {};
  for (const attr of msgEvent.attributes) {
    msgEventKeys[attr.key] = attr.value;
  }

  const balanceAfter = extractScrtBalance(
    (await client.getAccount(accountAddr) as WasmAccount).balance
  );

  const fees = Object.assign(
    Object.assign({}, defaultFees),
    (PolarContext.getPolarContext().getRuntimeEnv().network.config.fees ?? {})
  );

  if (
    includeFee !== true &&
    (await client.getAccount(accountAddr) as WasmAccount).address === msgEventKeys.signer
  ) {
    if ((await client.getAccount(accountAddr) as WasmAccount).address === msgEventKeys.signer) {
      return balanceAfter - balanceBefore;
    } else {
      let txnFees = 0;
      for (const [key, value] of Object.entries(fees)) {
        if (key === msgEventKeys.action) {
          txnFees = Number(value);
          break;
        }
      }
      return balanceAfter + txnFees - balanceBefore;
    }
  } else {
    return balanceBefore - balanceAfter;
  }
}