mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-12-23 09:52:26 +01:00
4f3fc95d50
* Use @metamask/eslint-config@1.1.0 * Use eslint-plugin-mocha@6.2.2 * Mark root ESLint config as root * Update Mocha ESLint rules with shared ESLint config
138 lines
3.6 KiB
JavaScript
138 lines
3.6 KiB
JavaScript
import assert from 'assert'
|
|
import sinon from 'sinon'
|
|
import CachedBalancesController from '../../../../app/scripts/controllers/cached-balances'
|
|
|
|
describe('CachedBalancesController', function () {
|
|
describe('updateCachedBalances', function () {
|
|
it('should update the cached balances', async function () {
|
|
const controller = new CachedBalancesController({
|
|
getNetwork: () => Promise.resolve(17),
|
|
accountTracker: {
|
|
store: {
|
|
subscribe: () => {},
|
|
},
|
|
},
|
|
initState: {
|
|
cachedBalances: 'mockCachedBalances',
|
|
},
|
|
})
|
|
|
|
controller._generateBalancesToCache = sinon.stub().callsFake(() => Promise.resolve('mockNewCachedBalances'))
|
|
|
|
await controller.updateCachedBalances({ accounts: 'mockAccounts' })
|
|
|
|
assert.equal(controller._generateBalancesToCache.callCount, 1)
|
|
assert.deepEqual(controller._generateBalancesToCache.args[0], ['mockAccounts', 17])
|
|
assert.equal(controller.store.getState().cachedBalances, 'mockNewCachedBalances')
|
|
})
|
|
})
|
|
|
|
describe('_generateBalancesToCache', function () {
|
|
it('should generate updated account balances where the current network was updated', function () {
|
|
const controller = new CachedBalancesController({
|
|
accountTracker: {
|
|
store: {
|
|
subscribe: () => {},
|
|
},
|
|
},
|
|
initState: {
|
|
cachedBalances: {
|
|
17: {
|
|
a: '0x1',
|
|
b: '0x2',
|
|
c: '0x3',
|
|
},
|
|
16: {
|
|
a: '0xa',
|
|
b: '0xb',
|
|
c: '0xc',
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
const result = controller._generateBalancesToCache({
|
|
a: { balance: '0x4' },
|
|
b: { balance: null },
|
|
c: { balance: '0x5' },
|
|
}, 17)
|
|
|
|
assert.deepEqual(result, {
|
|
17: {
|
|
a: '0x4',
|
|
b: '0x2',
|
|
c: '0x5',
|
|
},
|
|
16: {
|
|
a: '0xa',
|
|
b: '0xb',
|
|
c: '0xc',
|
|
},
|
|
})
|
|
})
|
|
|
|
it('should generate updated account balances where the a new network was selected', function () {
|
|
const controller = new CachedBalancesController({
|
|
accountTracker: {
|
|
store: {
|
|
subscribe: () => {},
|
|
},
|
|
},
|
|
initState: {
|
|
cachedBalances: {
|
|
17: {
|
|
a: '0x1',
|
|
b: '0x2',
|
|
c: '0x3',
|
|
},
|
|
},
|
|
},
|
|
})
|
|
|
|
const result = controller._generateBalancesToCache({
|
|
a: { balance: '0x4' },
|
|
b: { balance: null },
|
|
c: { balance: '0x5' },
|
|
}, 16)
|
|
|
|
assert.deepEqual(result, {
|
|
17: {
|
|
a: '0x1',
|
|
b: '0x2',
|
|
c: '0x3',
|
|
},
|
|
16: {
|
|
a: '0x4',
|
|
c: '0x5',
|
|
},
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('_registerUpdates', function () {
|
|
it('should subscribe to the account tracker with the updateCachedBalances method', async function () {
|
|
const subscribeSpy = sinon.spy()
|
|
const controller = new CachedBalancesController({
|
|
getNetwork: () => Promise.resolve(17),
|
|
accountTracker: {
|
|
store: {
|
|
subscribe: subscribeSpy,
|
|
},
|
|
},
|
|
})
|
|
subscribeSpy.resetHistory()
|
|
|
|
const updateCachedBalancesSpy = sinon.spy()
|
|
controller.updateCachedBalances = updateCachedBalancesSpy
|
|
controller._registerUpdates({ accounts: 'mockAccounts' })
|
|
|
|
assert.equal(subscribeSpy.callCount, 1)
|
|
|
|
subscribeSpy.args[0][0]()
|
|
|
|
assert.equal(updateCachedBalancesSpy.callCount, 1)
|
|
})
|
|
})
|
|
|
|
})
|