52 lines
1.5 KiB
TypeScript
52 lines
1.5 KiB
TypeScript
|
import { fireEvent, render, screen } from '@testing-library/react';
|
||
|
import { VegaWalletContext } from '@vegaprotocol/wallet';
|
||
|
import type { VegaWalletContextShape } from '@vegaprotocol/wallet';
|
||
|
import type { VegaWalletConnectButtonProps } from './vega-wallet-connect-button';
|
||
|
import { VegaWalletConnectButton } from './vega-wallet-connect-button';
|
||
|
|
||
|
let props: VegaWalletConnectButtonProps;
|
||
|
|
||
|
beforeEach(() => {
|
||
|
props = {
|
||
|
setConnectDialog: jest.fn(),
|
||
|
setManageDialog: jest.fn(),
|
||
|
};
|
||
|
});
|
||
|
|
||
|
const generateJsx = (
|
||
|
context: VegaWalletContextShape,
|
||
|
props: VegaWalletConnectButtonProps
|
||
|
) => {
|
||
|
return (
|
||
|
<VegaWalletContext.Provider value={context}>
|
||
|
<VegaWalletConnectButton {...props} />
|
||
|
</VegaWalletContext.Provider>
|
||
|
);
|
||
|
};
|
||
|
|
||
|
test('Not connected', () => {
|
||
|
render(generateJsx({ keypair: null } as VegaWalletContextShape, props));
|
||
|
|
||
|
const button = screen.getByRole('button');
|
||
|
expect(button).toHaveTextContent('Connect Vega wallet');
|
||
|
fireEvent.click(button);
|
||
|
expect(props.setConnectDialog).toHaveBeenCalledWith(true);
|
||
|
expect(props.setManageDialog).not.toHaveBeenCalled();
|
||
|
});
|
||
|
|
||
|
test('Connected', () => {
|
||
|
render(
|
||
|
generateJsx(
|
||
|
{ keypair: { pub: '123456__123456' } } as VegaWalletContextShape,
|
||
|
props
|
||
|
)
|
||
|
);
|
||
|
|
||
|
expect(screen.getByText('Vega key:')).toBeInTheDocument();
|
||
|
const button = screen.getByRole('button');
|
||
|
expect(button).toHaveTextContent('123456\u2026123456');
|
||
|
fireEvent.click(button);
|
||
|
expect(props.setManageDialog).toHaveBeenCalledWith(true);
|
||
|
expect(props.setConnectDialog).not.toHaveBeenCalled();
|
||
|
});
|