metamask-extension/test/unit/app/controllers/detect-tokens-test.js

import assert from 'assert'
import sinon from 'sinon'
import ObservableStore from 'obs-store'
import contracts from 'eth-contract-metadata'
import BigNumber from 'bignumber.js'

import DetectTokensController from '../../../../app/scripts/controllers/detect-tokens'
import NetworkController from '../../../../app/scripts/controllers/network/network'
import PreferencesController from '../../../../app/scripts/controllers/preferences'
import { MAINNET, ROPSTEN } from '../../../../app/scripts/controllers/network/enums'

describe('DetectTokensController', function () {
  const sandbox = sinon.createSandbox()
  let keyringMemStore, network, preferences

  const noop = () => undefined

  const networkControllerProviderConfig = {
    getAccounts: noop,
  }

  beforeEach(async function () {
    keyringMemStore = new ObservableStore({ isUnlocked: false })
    network = new NetworkController()
    network.setInfuraProjectId('foo')
    preferences = new PreferencesController({ network })
    preferences.setAddresses([
      '0x7e57e2',
      '0xbc86727e770de68b1060c91f6bb6945c73e10388',
    ])
    network.initializeProvider(networkControllerProviderConfig)

  })

  after(function () {
    sandbox.restore()
  })

  it('should poll on correct interval', async function () {
    const stub = sinon.stub(global, 'setInterval')
    new DetectTokensController({ interval: 1337 }) // eslint-disable-line no-new
    assert.strictEqual(stub.getCall(0).args[1], 1337)
    stub.restore()
  })

  it('should be called on every polling period', async function () {
    const clock = sandbox.useFakeTimers()
    network.setProviderType(MAINNET)
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.isUnlocked = true

    const stub = sandbox.stub(controller, 'detectNewTokens')

    clock.tick(1)
    sandbox.assert.notCalled(stub)
    clock.tick(180000)
    sandbox.assert.called(stub)
    clock.tick(180000)
    sandbox.assert.calledTwice(stub)
    clock.tick(180000)
    sandbox.assert.calledThrice(stub)
  })

  it('should not check tokens while on test network', async function () {
    sandbox.useFakeTimers()
    network.setProviderType(ROPSTEN)
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.isUnlocked = true

    const stub = sandbox.stub(controller, '_getTokenBalances')

    await controller.detectNewTokens()
    sandbox.assert.notCalled(stub)
  })

  it('should check and add tokens while on main network', async function () {
    sandbox.useFakeTimers()
    network.setProviderType(MAINNET)
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.isUnlocked = true

    const contractAddresses = Object.keys(contracts)
    const erc20ContractAddresses = contractAddresses
      .filter((contractAddress) => contracts[contractAddress].erc20 === true)

    const existingTokenAddress = erc20ContractAddresses[0]
    const existingToken = contracts[existingTokenAddress]
    await preferences.addToken(existingTokenAddress, existingToken.symbol, existingToken.decimals)

    const tokenAddressToAdd = erc20ContractAddresses[1]
    const tokenToAdd = contracts[tokenAddressToAdd]

    const contractAddresssesToDetect = contractAddresses
      .filter((address) => address !== existingTokenAddress)
    const indexOfTokenToAdd = contractAddresssesToDetect.indexOf(tokenAddressToAdd)

    const balances = new Array(contractAddresssesToDetect.length)
    balances[indexOfTokenToAdd] = new BigNumber(10)

    sandbox.stub(controller, '_getTokenBalances')
      .returns(Promise.resolve(balances))

    await controller.detectNewTokens()

    assert.deepEqual(
      preferences.store.getState().tokens,
      [
        { address: existingTokenAddress.toLowerCase(), decimals: existingToken.decimals, symbol: existingToken.symbol },
        { address: tokenAddressToAdd.toLowerCase(), decimals: tokenToAdd.decimals, symbol: tokenToAdd.symbol },
      ],
    )
  })

  it('should trigger detect new tokens when change address', async function () {
    sandbox.useFakeTimers()
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.isUnlocked = true
    const stub = sandbox.stub(controller, 'detectNewTokens')
    await preferences.setSelectedAddress('0xbc86727e770de68b1060c91f6bb6945c73e10388')
    sandbox.assert.called(stub)
  })

  it('should trigger detect new tokens when submit password', async function () {
    sandbox.useFakeTimers()
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.selectedAddress = '0x0'
    const stub = sandbox.stub(controller, 'detectNewTokens')
    await controller._keyringMemStore.updateState({ isUnlocked: true })
    sandbox.assert.called(stub)
  })

  it('should not trigger detect new tokens when not unlocked', async function () {
    const clock = sandbox.useFakeTimers()
    network.setProviderType(MAINNET)
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    controller.isOpen = true
    controller.isUnlocked = false
    const stub = sandbox.stub(controller, '_getTokenBalances')
    clock.tick(180000)
    sandbox.assert.notCalled(stub)
  })

  it('should not trigger detect new tokens when not open', async function () {
    const clock = sandbox.useFakeTimers()
    network.setProviderType(MAINNET)
    const controller = new DetectTokensController({ preferences, network, keyringMemStore })
    // trigger state update from preferences controller
    await preferences.setSelectedAddress('0xbc86727e770de68b1060c91f6bb6945c73e10388')
    controller.isOpen = false
    controller.isUnlocked = true
    const stub = sandbox.stub(controller, '_getTokenBalances')
    clock.tick(180000)
    sandbox.assert.notCalled(stub)
  })
})