vega-frontend-monorepo/apps/trading/components/vega-wallet-connect-button/vega-wallet-connect-button.spec.tsx

67 lines
2.3 KiB
TypeScript

import { fireEvent, render, screen } from '@testing-library/react';
import { VegaWalletContext } from '@vegaprotocol/wallet';
import type { VegaWalletContextShape } from '@vegaprotocol/wallet';
import { VegaWalletConnectButton } from './vega-wallet-connect-button';
import { truncateByChars } from '@vegaprotocol/utils';
import userEvent from '@testing-library/user-event';
const mockUpdateDialogOpen = jest.fn();
jest.mock('@vegaprotocol/wallet', () => ({
...jest.requireActual('@vegaprotocol/wallet'),
useVegaWalletDialogStore: () => mockUpdateDialogOpen,
}));
beforeEach(() => {
jest.clearAllMocks();
});
const generateJsx = (context: VegaWalletContextShape) => {
return (
<VegaWalletContext.Provider value={context}>
<VegaWalletConnectButton />
</VegaWalletContext.Provider>
);
};
describe('VegaWalletConnectButton', () => {
it('should fire dialog when not connected', () => {
render(generateJsx({ pubKey: null } as VegaWalletContextShape));
const button = screen.getByTestId('connect-vega-wallet');
expect(button).toHaveTextContent('Connect Vega wallet');
fireEvent.click(button);
expect(mockUpdateDialogOpen).toHaveBeenCalled();
});
it('should retrieve keys when connected', async () => {
const pubKey = { publicKey: '123456__123456', name: 'test' };
const pubKey2 = { publicKey: '123456__123457', name: 'test2' };
render(
generateJsx({
pubKey: pubKey.publicKey,
pubKeys: [pubKey],
fetchPubKeys: () => Promise.resolve([pubKey, pubKey2]),
} as VegaWalletContextShape)
);
const button = screen.getByTestId('manage-vega-wallet');
expect(button).toHaveTextContent(truncateByChars(pubKey.publicKey));
userEvent.click(button);
expect(mockUpdateDialogOpen).not.toHaveBeenCalled();
});
it('should fetch keys when connected', async () => {
const pubKey = { publicKey: '123456__123456', name: 'test' };
const context = {
pubKey: pubKey.publicKey,
pubKeys: [pubKey],
} as VegaWalletContextShape;
render(generateJsx(context));
const button = screen.getByTestId('manage-vega-wallet');
expect(button).toHaveTextContent(truncateByChars(pubKey.publicKey));
userEvent.click(button);
expect(mockUpdateDialogOpen).not.toHaveBeenCalled();
});
});