metamask-extension/ui/app/pages/add-token/tests/add-token.test.js

import assert from 'assert'
import React from 'react'
import { Provider } from 'react-redux'
import sinon from 'sinon'
import configureMockStore from 'redux-mock-store'
import { mountWithRouter } from '../../../../../test/lib/render-helpers'
import AddToken from '..'

describe('Add Token', function () {
  let wrapper

  const state = {
    metamask: {
      tokens: [],
    },
  }

  const store = configureMockStore()(state)

  const props = {
    history: {
      push: sinon.stub().callsFake(() => undefined),
    },
    setPendingTokens: sinon.spy(),
    clearPendingTokens: sinon.spy(),
    tokens: [],
    identities: {},
    mostRecentOverviewPage: '/',
  }

  describe('Add Token', function () {
    before(function () {
      wrapper = mountWithRouter(
        <Provider store={store}>
          <AddToken.WrappedComponent {...props} />
        </Provider>,
        store,
      )

      wrapper.find({ name: 'customToken' }).simulate('click')
    })

    afterEach(function () {
      props.history.push.reset()
    })

    it('next button is disabled when no fields are populated', function () {
      const nextButton = wrapper.find(
        '.button.btn-secondary.page-container__footer-button',
      )

      assert.equal(nextButton.props().disabled, true)
    })

    it('edits token address', function () {
      const tokenAddress = '0x617b3f8050a0BD94b6b1da02B4384eE5B4DF13F4'
      const event = { target: { value: tokenAddress } }
      const customAddress = wrapper.find('input#custom-address')

      customAddress.simulate('change', event)
      assert.equal(
        wrapper.find('AddToken').instance().state.customAddress,
        tokenAddress,
      )
    })

    it('edits token symbol', function () {
      const tokenSymbol = 'META'
      const event = { target: { value: tokenSymbol } }
      const customAddress = wrapper.find('#custom-symbol')
      customAddress.last().simulate('change', event)

      assert.equal(
        wrapper.find('AddToken').instance().state.customSymbol,
        tokenSymbol,
      )
    })

    it('edits token decimal precision', function () {
      const tokenPrecision = '2'
      const event = { target: { value: tokenPrecision } }
      const customAddress = wrapper.find('#custom-decimals')
      customAddress.last().simulate('change', event)

      assert.equal(
        wrapper.find('AddToken').instance().state.customDecimals,
        tokenPrecision,
      )
    })

    it('next', function () {
      const nextButton = wrapper.find(
        '.button.btn-secondary.page-container__footer-button',
      )
      nextButton.simulate('click')

      assert(props.setPendingTokens.calledOnce)
      assert(props.history.push.calledOnce)
      assert.equal(props.history.push.getCall(0).args[0], '/confirm-add-token')
    })

    it('cancels', function () {
      const cancelButton = wrapper.find(
        'button.btn-default.page-container__footer-button',
      )
      cancelButton.simulate('click')

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