metamask-extension/ui/app/pages/first-time-flow/select-action/tests/select-action.test.js

import React from 'react'
import assert from 'assert'
import sinon from 'sinon'
import { mountWithRouter } from '../../../../../../test/lib/render-helpers'
import SelectAction from '../index'

describe('Selection Action', () => {
  let wrapper

  const props = {
    isInitialized: false,
    setFirstTimeFlowType: sinon.spy(),
    history: {
      push: sinon.spy(),
    },
  }

  beforeEach(() => {
    wrapper = mountWithRouter(
      <SelectAction.WrappedComponent {...props} />
    )
  })

  afterEach(() => {
    props.setFirstTimeFlowType.resetHistory()
    props.history.push.resetHistory()
  })

  it('clicks import wallet to route to import FTF', () => {
    const importWalletButton = wrapper.find('.btn-primary.first-time-flow__button').at(0)
    importWalletButton.simulate('click')

    assert(props.setFirstTimeFlowType.calledOnce)
    assert.equal(props.setFirstTimeFlowType.getCall(0).args[0], 'import')
    assert(props.history.push.calledOnce)
  })

  it('clicks create wallet to route to create FTF ', () => {
    const createWalletButton = wrapper.find('.btn-primary.first-time-flow__button').at(1)
    createWalletButton.simulate('click')

    assert(props.setFirstTimeFlowType.calledOnce)
    assert.equal(props.setFirstTimeFlowType.getCall(0).args[0], 'create')
    assert(props.history.push.calledOnce)
  })
})