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

const assert = require('assert')
const sinon = require('sinon')
const DetectTokensController = require('../../../../app/scripts/controllers/detect-tokens')
const PreferencesController = require('../../../../app/scripts/controllers/preferences')
const ObservableStore = require('obs-store')

describe('DetectTokensController', () => {
  it('should poll on correct interval', async () => {
    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 not check tokens while in test network', async () => {
    var network = new ObservableStore({provider: {type: 'rinkeby'}})
    const preferences = new PreferencesController()
    const controller = new DetectTokensController({preferences: preferences, network: network})
    controller.isActive = true
    controller.contracts = {
        '0x0D262e5dC4A06a0F1c90cE79C7a60C09DfC884E4': {
            'name': 'JET8 Token',
            'logo': 'J8T.svg',
            'erc20': true,
            'symbol': 'J8T',
            'decimals': 8,
        },
        '0xBC86727E770de68B1060C91f6BB6945c73e10388': {
            'name': 'Ink Protocol',
            'logo': 'ink_protocol.svg',
            'erc20': true,
            'symbol': 'XNK',
            'decimals': 18,
        },
      }
    controller.fetchContractAccountBalance = address => address
    await controller.exploreNewTokens()
    assert.deepEqual(preferences.store.getState().tokens, [])
})

  it('should only check and add tokens while in main network', async () => {
    const network = new ObservableStore({provider: {type: 'mainnet'}})
    const preferences = new PreferencesController()
    const controller = new DetectTokensController({preferences: preferences, network: network})
    controller.isActive = true
    controller.contracts = {
        '0x0D262e5dC4A06a0F1c90cE79C7a60C09DfC884E4': {
            'name': 'JET8 Token',
            'logo': 'J8T.svg',
            'erc20': true,
            'symbol': 'J8T',
            'decimals': 8,
        },
        '0xBC86727E770de68B1060C91f6BB6945c73e10388': {
            'name': 'Ink Protocol',
            'logo': 'ink_protocol.svg',
            'erc20': true,
            'symbol': 'XNK',
            'decimals': 18,
        },
    }
    controller.fetchContractAccountBalance = address => address
    await controller.exploreNewTokens()
    assert.deepEqual(preferences.store.getState().tokens, [{address: '0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', decimals: 8, symbol: 'J8T'}, {address: '0xbc86727e770de68b1060c91f6bb6945c73e10388', decimals: 18, symbol: 'XNK'}])
  })

  it('should not detect same token while in main network', async () => {
    const network = new ObservableStore({provider: {type: 'mainnet'}})
    const preferences = new PreferencesController()
    preferences.addToken('0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', 8, 'J8T')
    const controller = new DetectTokensController({preferences: preferences, network: network})
    controller.isActive = true
    controller.contracts = {
        '0x0D262e5dC4A06a0F1c90cE79C7a60C09DfC884E4': {
            'name': 'JET8 Token',
            'logo': 'J8T.svg',
            'erc20': true,
            'symbol': 'J8T',
            'decimals': 8,
        },
        '0xBC86727E770de68B1060C91f6BB6945c73e10388': {
            'name': 'Ink Protocol',
            'logo': 'ink_protocol.svg',
            'erc20': true,
            'symbol': 'XNK',
            'decimals': 18,
        },
    }
    controller.fetchContractAccountBalance = address => address
    await controller.exploreNewTokens()
    assert.deepEqual(preferences.store.getState().tokens, [{address: '0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', decimals: 8, symbol: 'J8T'}, {address: '0xbc86727e770de68b1060c91f6bb6945c73e10388', decimals: 18, symbol: 'XNK'}])
  })
})