const assert = require('assert') const sinon = require('sinon') const TokenRatesController = require('../../../../app/scripts/controllers/token-rates') const ObservableStore = require('obs-store') describe('TokenRatesController', () => { it('should listen for preferences store updates', () => { const preferences = new ObservableStore({ tokens: [] }) const controller = new TokenRatesController({ preferences }) preferences.putState({ tokens: ['foo'] }) assert.deepEqual(controller._tokens, ['foo']) }) it('should poll on correct interval', async () => { const stub = sinon.stub(global, 'setInterval') new TokenRatesController({ interval: 1337 }) // eslint-disable-line no-new assert.strictEqual(stub.getCall(0).args[1], 1337) stub.restore() }) it('should fetch each token rate based on address', async () => { const controller = new TokenRatesController() controller.isActive = true controller.fetchExchangeRate = address => address controller.tokens = [{ address: 'foo' }, { address: 'bar' }] await controller.updateExchangeRates() assert.deepEqual(controller.store.getState().contractExchangeRates, { foo: 'foo', bar: 'bar' }) }) })