unit tests: add parameter image to addToken calls

This commit is contained in:
dennistikhomirov 2018-10-25 18:07:05 -07:00
parent d8ba17dbc1
commit bfd7247d9d
2 changed files with 16 additions and 16 deletions

View File

@ -82,13 +82,12 @@ describe('DetectTokensController', () => {
const controller = new DetectTokensController({ preferences: preferences, network: network, keyringMemStore: keyringMemStore })
controller.isOpen = true
controller.isUnlocked = true
const image = 'image'
sandbox.stub(controller, 'detectTokenBalance')
.withArgs('0x0D262e5dC4A06a0F1c90cE79C7a60C09DfC884E4')
.returns(preferences.addToken('0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', 'J8T', 8, 1))
.returns(preferences.addToken('0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', 'J8T', 8, image, 1))
.withArgs('0xBC86727E770de68B1060C91f6BB6945c73e10388')
.returns(preferences.addToken('0xbc86727e770de68b1060c91f6bb6945c73e10388', 'XNK', 18, 1))
.returns(preferences.addToken('0xbc86727e770de68b1060c91f6bb6945c73e10388', 'XNK', 18, image, 1))
await controller.detectNewTokens()
assert.deepEqual(preferences.store.getState().tokens, [{address: '0x0d262e5dc4a06a0f1c90ce79c7a60c09dfc884e4', decimals: 8, symbol: 'J8T', network: 1},
{address: '0xbc86727e770de68b1060c91f6bb6945c73e10388', decimals: 18, symbol: 'XNK', network: 1}])

View File

@ -6,6 +6,7 @@ const sinon = require('sinon')
describe('preferences controller', function () {
let preferencesController
let network
const image = 'image'
beforeEach(() => {
network = {providerStore: new ObservableStore({ type: 'mainnet' })}
@ -161,10 +162,10 @@ describe('preferences controller', function () {
const network = 1
await preferencesController.setSelectedAddress('0x7e57e2')
await preferencesController.addToken(address, symbol, decimals, network)
await preferencesController.addToken(address, symbol, decimals, image, network)
const newDecimals = 6
await preferencesController.addToken(address, symbol, newDecimals, network)
await preferencesController.addToken(address, symbol, newDecimals, image, network)
const tokens = preferencesController.getTokens()
assert.equal(tokens.length, 1, 'one token added')
@ -240,8 +241,8 @@ describe('preferences controller', function () {
it('should remove a token from its state', async function () {
await preferencesController.setSelectedAddress('0x7e57e2')
await preferencesController.addToken('0xa', 'A', 4, 1)
await preferencesController.addToken('0xb', 'B', 5, 1)
await preferencesController.addToken('0xa', 'A', 4, image, 1)
await preferencesController.addToken('0xb', 'B', 5, image, 1)
await preferencesController.removeToken('0xa')
const tokens = preferencesController.getTokens()
@ -279,11 +280,11 @@ describe('preferences controller', function () {
it('should remove a token from its state on corresponding address', async function () {
await preferencesController.setSelectedAddress('0x7e57e2')
await preferencesController.addToken('0xa', 'A', 4, 1)
await preferencesController.addToken('0xb', 'B', 5, 1)
await preferencesController.addToken('0xa', 'A', 4, image, 1)
await preferencesController.addToken('0xb', 'B', 5, image, 1)
await preferencesController.setSelectedAddress('0x7e57e3')
await preferencesController.addToken('0xa', 'A', 4, 1)
await preferencesController.addToken('0xb', 'B', 5, 1)
await preferencesController.addToken('0xa', 'A', 4, image, 1)
await preferencesController.addToken('0xb', 'B', 5, image, 1)
const initialTokensSecond = preferencesController.getTokens()
await preferencesController.setSelectedAddress('0x7e57e2')
await preferencesController.removeToken('0xa')
@ -301,11 +302,11 @@ describe('preferences controller', function () {
it('should remove a token from its state on corresponding network', async function () {
network.providerStore.updateState({ type: 'mainnet' })
await preferencesController.addToken('0xa', 'A', 4, 1)
await preferencesController.addToken('0xb', 'B', 5, 1)
await preferencesController.addToken('0xa', 'A', 4, image, 1)
await preferencesController.addToken('0xb', 'B', 5, image, 1)
network.providerStore.updateState({ type: 'rinkeby' })
await preferencesController.addToken('0xa', 'A', 4, 1)
await preferencesController.addToken('0xb', 'B', 5, 1)
await preferencesController.addToken('0xa', 'A', 4, image, 1)
await preferencesController.addToken('0xb', 'B', 5, image, 1)
const initialTokensSecond = preferencesController.getTokens()
network.providerStore.updateState({ type: 'mainnet' })
await preferencesController.removeToken('0xa')