metamask-extension/ui/app/pages/first-time-flow/end-of-flow/tests/end-of-flow.test.js

import assert from 'assert'
import React from 'react'
import sinon from 'sinon'
import { mountWithRouter } from '../../../../../../test/lib/render-helpers'
import { DEFAULT_ROUTE } from '../../../../helpers/constants/routes'
import EndOfFlowScreen from '..'

describe('End of Flow Screen', function () {
  let wrapper

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

  beforeEach(function () {
    wrapper = mountWithRouter(
      <EndOfFlowScreen.WrappedComponent {...props} />,
    )
  })

  it('renders', function () {
    assert.equal(wrapper.length, 1)
  })

  it('should navigate to the default route on click', function (done) {
    const endOfFlowButton = wrapper.find('.btn-primary.first-time-flow__button')
    endOfFlowButton.simulate('click')

    setImmediate(() => {
      assert(props.history.push.calledOnceWithExactly(DEFAULT_ROUTE))
      done()
    })
  })
})