1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 01:39:44 +01:00
metamask-extension/app/scripts/controllers/cached-balances.test.js
ryanml 0bc1eeaf37
Deprecating the Rinkeby, Ropsten, and Kovan test networks (#15989)
* Deprecating Rinkeby, setting default debug network to Goerli

* Deprecating Ropsten and Kovan

* Conflict fix

* Remove unused localization, test fixes

* Add migration for moving used deprecated testnets to custom networks

* Fix migrator test

* Add more unit tests

* Migration updates provider type to rpc if deprecated network is selected

* Migration fully and correctly updates the provider if selected network is a deprecated testnet

* Continue to show deprecation warning on each of rinkeby, ropsten and kovan

* Add rpcUrl deprecation message to loading screen

* Removing mayBeFauceting prop

Co-authored-by: Dan Miller <danjm.com@gmail.com>
2022-09-28 20:26:01 -07:00

152 lines
3.9 KiB
JavaScript

import { strict as assert } from 'assert';
import sinon from 'sinon';
import { CHAIN_IDS } from '../../../shared/constants/network';
import CachedBalancesController from './cached-balances';
describe('CachedBalancesController', function () {
describe('updateCachedBalances', function () {
it('should update the cached balances', async function () {
const controller = new CachedBalancesController({
getCurrentChainId: () => CHAIN_IDS.GOERLI,
accountTracker: {
store: {
subscribe: () => undefined,
},
},
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',
CHAIN_IDS.GOERLI,
]);
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: () => undefined,
},
},
initState: {
cachedBalances: {
[CHAIN_IDS.GOERLI]: {
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' },
},
CHAIN_IDS.GOERLI,
);
assert.deepEqual(result, {
[CHAIN_IDS.GOERLI]: {
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: () => undefined,
},
},
initState: {
cachedBalances: {
[CHAIN_IDS.GOERLI]: {
a: '0x1',
b: '0x2',
c: '0x3',
},
},
},
});
const result = controller._generateBalancesToCache(
{
a: { balance: '0x4' },
b: { balance: null },
c: { balance: '0x5' },
},
16,
);
assert.deepEqual(result, {
[CHAIN_IDS.GOERLI]: {
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({
getCurrentChainId: () => CHAIN_IDS.GOERLI,
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);
});
});
});