wallet observer tests

wallet-observer-test.ts

import { WalletCore } from "@aptos-labs/wallet-adapter-core";
import { MartianWallet } from "@martianwallet/aptos-wallet-adapter";
import { PetraWallet } from "petra-plugin-wallet-adapter";
import { connectWalletAction, disconnectWalletAction } from "@store/wallet/wallet.reducer";
import { getAptosWalletState } from "@store/wallet/wallet.utils";
import { aptosWalletCore, connectAptosWallet, injectStoreToAptosWalletConfig } from './wallet.config.aptos';

jest.mock("@aptos-labs/wallet-adapter-core");
jest.mock("@martianwallet/aptos-wallet-adapter");
jest.mock("petra-plugin-wallet-adapter");
jest.mock("@store/wallet/wallet.reducer");
jest.mock("@store/wallet/wallet.utils");

describe("Aptos Wallet Config", () => {
  let storeMock;

  beforeEach(() => {
    storeMock = {
      dispatch: jest.fn(),
    };
    injectStoreToAptosWalletConfig(storeMock);
  });

  it("should connect to wallet", async () => {
    const walletName = "MartianWallet";
    await connectAptosWallet(walletName);
    expect(WalletCore.prototype.connect).toHaveBeenCalledWith(walletName);
    expect(WalletCore.prototype.onAccountChange).toHaveBeenCalled();
    expect(WalletCore.prototype.onNetworkChange).toHaveBeenCalled();
  });

  it("should update wallet state when connected", () => {
    WalletCore.prototype.isConnected.mockReturnValue(true);
    aptosWalletCore.emit("connect");
    expect(getAptosWalletState).toHaveBeenCalled();
    expect(storeMock.dispatch).toHaveBeenCalledWith(connectWalletAction(expect.any(Object)));
  });

  it("should dispatch disconnect action when disconnected", () => {
    aptosWalletCore.emit("disconnect");
    expect(storeMock.dispatch).toHaveBeenCalledWith(disconnectWalletAction("aptos"));
  });

  // Add more tests for other events like "readyStateChange", "networkChange", "accountChange"
});