@ledgerhq/live-common/src/exchange/swap/hooks/useToState.test.ts

Common ground for the Ledger Live apps

/**
 * @jest-environment jsdom
 */
import "../../../__tests__/test-helpers/dom-polyfill";
import { renderHook, act } from "@testing-library/react";
import { getCryptoCurrencyById, findTokenById } from "@ledgerhq/cryptoassets";
import { selectorStateDefaultValues, useToState } from ".";
import { genTokenAccount } from "@ledgerhq/coin-framework/mocks/account";
import { genAccount } from "../../../mock/account";
import BigNumber from "bignumber.js";
import type { Account } from "@ledgerhq/types-live";

const BTC = getCryptoCurrencyById("bitcoin");
const ETH = getCryptoCurrencyById("ethereum");
const usdtToken = findTokenById("ethereum/erc20/usd_tether__erc20_");
if (!usdtToken) throw new Error("USDT token not found");
const USDT = usdtToken;

const selectedAccount = genAccount("mocked-account-selected", { currency: ETH });

const accounts: Account[] = [
  genAccount("mocked-account-1", { currency: BTC }),
  genAccount("mocked-account-2", { currency: ETH }),
];
const subAccount = genTokenAccount(1, accounts[1], USDT);

describe("useToState", () => {
  const defaultProps = { accounts, fromCurrencyAccount: selectedAccount };

  it("should initialize a blank state", () => {
    const { result } = renderHook(useToState, { initialProps: defaultProps });
    expect(result.current.toState).toMatchObject(selectorStateDefaultValues);
  });

  it("should set the account", async () => {
    const [currency, account, parentAccount] = [USDT, subAccount, accounts[1]];
    const { result } = renderHook(useToState, { initialProps: defaultProps });
    act(() => result.current.setToAccount(currency, account, parentAccount));
    expect(result.current.toState).toMatchObject({
      account,
      parentAccount,
      currency,
    });
  });

  it("should set the amount", () => {
    const amount = new BigNumber(10);
    const { result } = renderHook(useToState, { initialProps: defaultProps });
    act(() => result.current.setToAmount(amount));
    expect(result.current.toState.amount).toBe(amount);
  });

  it("should set the currency", () => {
    const { result } = renderHook(useToState, { initialProps: defaultProps });
    act(() => result.current.setToCurrency(BTC));
    expect(result.current.toState).toMatchObject({
      account: accounts[0],
      parentAccount: null,
      currency: BTC,
    });
  });
});