vega-frontend-monorepo/apps/trading/components/welcome-dialog/risk-notice-dialog.spec.tsx

import { render, screen, fireEvent } from '@testing-library/react';
import { Networks, useEnvironment } from '@vegaprotocol/environment';
import { RiskNoticeDialog } from './risk-notice-dialog';

jest.mock('@vegaprotocol/environment');

const mockEnvDefinitions = {
  VEGA_CONFIG_URL: 'https://config.url',
  VEGA_URL: 'https://test.url',
  VEGA_NETWORKS: JSON.stringify({}),
};

describe('Risk notice dialog', () => {
  const mockOnClose = jest.fn();
  afterEach(() => {
    jest.clearAllMocks();
  });

  it.each`
    assertion     | network
    ${'displays'} | ${Networks.CUSTOM}
    ${'displays'} | ${Networks.DEVNET}
    ${'displays'} | ${Networks.TESTNET}
  `(
    '$assertion a generic message on $network',
    async ({ assertion, network }) => {
      // @ts-ignore ignore mock implementation
      useEnvironment.mockImplementation(() => ({
        ...mockEnvDefinitions,
        VEGA_ENV: network,
      }));

      render(<RiskNoticeDialog onClose={mockOnClose} network={network} />);

      expect(
        screen.getByText(
          new RegExp(
            `This application for trading on Vega is connected to ${network}`
          )
        )
      ).toBeInTheDocument();

      const button = screen.getByRole('button', {
        name: 'Continue',
      });
      fireEvent.click(button);
      expect(mockOnClose).toHaveBeenCalled();
    }
  );
});