diff --git a/src/assets/TokenBalancesController.test.ts b/src/assets/TokenBalancesController.test.ts index c0fb7fe17b..c919379e8c 100644 --- a/src/assets/TokenBalancesController.test.ts +++ b/src/assets/TokenBalancesController.test.ts @@ -7,7 +7,7 @@ import { PreferencesController } from '../user/PreferencesController'; import { AssetsController } from './AssetsController'; import { Token } from './TokenRatesController'; import { AssetsContractController } from './AssetsContractController'; -import TokenBalancesController from './TokenBalancesController'; +import { BN as exportedBn, TokenBalancesController } from './TokenBalancesController'; const MAINNET_PROVIDER = new HttpProvider('https://mainnet.infura.io'); @@ -28,6 +28,10 @@ describe('TokenBalancesController', () => { sandbox.restore(); }); + it('should re-export BN', () => { + expect(exportedBn).toEqual(BN); + }); + it('should set default state', () => { expect(tokenBalances.state).toEqual({ contractBalances: {} }); }); diff --git a/src/assets/TokenBalancesController.ts b/src/assets/TokenBalancesController.ts index 47ed6b2fed..6f53751e17 100644 --- a/src/assets/TokenBalancesController.ts +++ b/src/assets/TokenBalancesController.ts @@ -5,6 +5,9 @@ import AssetsController from './AssetsController'; import { Token } from './TokenRatesController'; import { AssetsContractController } from './AssetsContractController'; +// TODO: Remove this export in the next major release +export { BN }; + /** * @type TokenBalancesConfig *