metamask-extension/ui/pages/first-time-flow/welcome/welcome.test.js

import React from 'react';
import sinon from 'sinon';
import configureMockStore from 'redux-mock-store';
import { mountWithRouter } from '../../../../test/lib/render-helpers';
import Welcome from './welcome.container';

describe('Welcome', () => {
  const mockStore = {
    metamask: {},
  };

  const store = configureMockStore()(mockStore);

  afterAll(() => {
    sinon.restore();
  });

  it('routes to select action when participateInMetaMetrics is not initialized', () => {
    const props = {
      history: {
        push: sinon.spy(),
      },
    };

    const wrapper = mountWithRouter(
      <Welcome.WrappedComponent {...props} />,
      store,
    );

    const getStartedButton = wrapper.find(
      '.btn-primary.first-time-flow__button',
    );
    getStartedButton.simulate('click');
    expect(props.history.push.getCall(0).args[0]).toStrictEqual(
      '/initialize/select-action',
    );
  });

  it('routes to correct password when participateInMetaMetrics is initialized', () => {
    const props = {
      welcomeScreenSeen: true,
      participateInMetaMetrics: false,
      history: {
        push: sinon.spy(),
      },
    };

    const wrapper = mountWithRouter(
      <Welcome.WrappedComponent {...props} />,
      store,
    );

    const getStartedButton = wrapper.find(
      '.btn-primary.first-time-flow__button',
    );
    getStartedButton.simulate('click');
    expect(props.history.push.getCall(0).args[0]).toStrictEqual(
      '/initialize/create-password',
    );
  });
});