import sinon from 'sinon'; import configureStore from 'redux-mock-store'; import thunk from 'redux-thunk'; import enLocale from '../../app/_locales/en/messages.json'; import MetaMaskController from '../../app/scripts/metamask-controller'; import { TransactionStatus } from '../../shared/constants/transaction'; import { HardwareDeviceNames } from '../../shared/constants/hardware-wallets'; import { GAS_LIMITS } from '../../shared/constants/gas'; import { ORIGIN_METAMASK } from '../../shared/constants/app'; import { MetaMetricsNetworkEventSource } from '../../shared/constants/metametrics'; import * as actions from './actions'; import { _setBackgroundConnection } from './action-queue'; const middleware = [thunk]; const defaultState = { metamask: { currentLocale: 'test', selectedAddress: '0xFirstAddress', providerConfig: { chainId: '0x1' }, accounts: { '0xFirstAddress': { balance: '0x0', }, }, identities: { '0xFirstAddress': {}, }, cachedBalances: { '0x1': { '0xFirstAddress': '0x0', }, }, }, }; const mockStore = (state = defaultState) => configureStore(middleware)(state); const baseMockState = defaultState.metamask; describe('Actions', () => { let background; const currentNetworkId = '5'; beforeEach(async () => { background = sinon.createStubInstance(MetaMaskController, { getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); background.signMessage = sinon.stub(); background.signPersonalMessage = sinon.stub(); background.signTypedMessage = sinon.stub(); }); describe('#tryUnlockMetamask', () => { afterEach(() => { sinon.restore(); }); it('calls submitPassword', async () => { const store = mockStore(); const submitPassword = background.submitPassword.callsFake((_, cb) => cb(), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'UNLOCK_IN_PROGRESS' }, { type: 'UNLOCK_SUCCEEDED', value: undefined }, { type: 'UPDATE_METAMASK_STATE', value: baseMockState, }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.tryUnlockMetamask()); expect(submitPassword.callCount).toStrictEqual(1); expect(store.getActions()).toStrictEqual(expectedActions); }); it('errors on submitPassword will fail', async () => { const store = mockStore(); background.submitPassword.callsFake((_, cb) => cb(new Error('error'))); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'UNLOCK_IN_PROGRESS' }, { type: 'UNLOCK_FAILED', value: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.tryUnlockMetamask('test')), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#createNewVaultAndRestore', () => { afterEach(() => { sinon.restore(); }); it('calls createNewVaultAndRestore', async () => { const store = mockStore(); const createNewVaultAndRestore = background.createNewVaultAndRestore.callsFake((_, __, cb) => cb()); background.unMarkPasswordForgotten.callsFake((cb) => cb()); _setBackgroundConnection(background); await store.dispatch( actions.createNewVaultAndRestore('password', 'test'), ); expect(createNewVaultAndRestore.callCount).toStrictEqual(1); }); it('calls the expected actions', async () => { const store = mockStore(); background.createNewVaultAndRestore.callsFake((_, __, cb) => cb()); background.unMarkPasswordForgotten.callsFake((cb) => cb()); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'UPDATE_METAMASK_STATE', value: baseMockState, }, { type: 'SHOW_ACCOUNTS_PAGE' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch( actions.createNewVaultAndRestore('password', 'test'), ); expect(store.getActions()).toStrictEqual(expectedActions); }); it('errors when callback in createNewVaultAndRestore throws', async () => { const store = mockStore(); background.createNewVaultAndRestore.callsFake((_, __, cb) => cb(new Error('error')), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.createNewVaultAndRestore('password', 'test')), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#requestRevealSeedWords', () => { afterEach(() => { sinon.restore(); }); it('calls verifyPassword in background', async () => { const store = mockStore(); const verifyPassword = background.verifyPassword.callsFake((_, cb) => cb(), ); const verifySeedPhrase = background.verifySeedPhrase.callsFake((cb) => cb(null, Array.from(Buffer.from('test').values())), ); _setBackgroundConnection(background); await store.dispatch(actions.requestRevealSeedWords()); expect(verifyPassword.callCount).toStrictEqual(1); expect(verifySeedPhrase.callCount).toStrictEqual(1); }); it('displays warning error message then callback in background errors', async () => { const store = mockStore(); background.verifyPassword.callsFake((_, cb) => cb()); background.verifySeedPhrase.callsFake((cb) => { cb(new Error('error')); }); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.requestRevealSeedWords()), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#removeAccount', () => { afterEach(() => { sinon.restore(); }); it('calls removeAccount in background and expect actions to show account', async () => { const store = mockStore(); background.getState.callsFake((cb) => cb(null, { currentLocale: 'test', selectedAddress: '0xAnotherAddress', providerConfig: { chainId: '0x1', }, accounts: { '0xAnotherAddress': { balance: '0x0', }, }, cachedBalances: { '0x1': { '0xAnotherAddress': '0x0', }, }, }), ); const removeAccount = background.removeAccount.callsFake((_, cb) => cb()); _setBackgroundConnection(background); const expectedActions = [ 'SHOW_LOADING_INDICATION', 'SELECTED_ADDRESS_CHANGED', 'ACCOUNT_CHANGED', 'SELECTED_ACCOUNT_CHANGED', 'UPDATE_METAMASK_STATE', 'HIDE_LOADING_INDICATION', 'SHOW_ACCOUNTS_PAGE', ]; await store.dispatch( actions.removeAccount('0xe18035bf8712672935fdb4e5e431b1a0183d2dfc'), ); expect(removeAccount.callCount).toStrictEqual(1); const actionTypes = store.getActions().map((action) => action.type); expect(actionTypes).toStrictEqual(expectedActions); }); it('displays warning error message when removeAccount callback errors', async () => { const store = mockStore(); background.removeAccount.callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch( actions.removeAccount('0xe18035bf8712672935fdb4e5e431b1a0183d2dfc'), ), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#resetAccount', () => { afterEach(() => { sinon.restore(); }); it('resets account', async () => { const store = mockStore(); const resetAccount = background.resetAccount.callsFake((cb) => cb()); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'SHOW_ACCOUNTS_PAGE' }, ]; await store.dispatch(actions.resetAccount()); expect(resetAccount.callCount).toStrictEqual(1); expect(store.getActions()).toStrictEqual(expectedActions); }); it('throws if resetAccount throws', async () => { const store = mockStore(); background.resetAccount.callsFake((cb) => { cb(new Error('error')); }); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; await expect(store.dispatch(actions.resetAccount())).rejects.toThrow( 'error', ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#importNewAccount', () => { afterEach(() => { sinon.restore(); }); it('calls importAccountWithStrategies in background', async () => { const store = mockStore(); const importAccountWithStrategy = background.importAccountWithStrategy.callsFake((_, __, cb) => { cb(); }); _setBackgroundConnection(background); await store.dispatch( actions.importNewAccount( 'Private Key', ['c87509a1c067bbde78beb793e6fa76530b6382a4c0241e5e4a9ec0a0f44dc0d3'], '', ), ); expect(importAccountWithStrategy.callCount).toStrictEqual(1); }); it('displays warning error message when importAccount in background callback errors', async () => { const store = mockStore(); background.importAccountWithStrategy.callsFake((_, __, cb) => cb(new Error('error')), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined, }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect(store.dispatch(actions.importNewAccount())).rejects.toThrow( 'error', ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#addNewAccount', () => { it('adds a new account', async () => { const store = mockStore({ metamask: { identities: {}, ...defaultState.metamask }, }); const addNewAccount = background.addNewAccount.callsFake((_, cb) => cb(null, { identities: {}, }), ); _setBackgroundConnection(background); await store.dispatch(actions.addNewAccount(1)); expect(addNewAccount.callCount).toStrictEqual(1); }); it('displays warning error message when addNewAccount in background callback errors', async () => { const store = mockStore(); background.addNewAccount.callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect(store.dispatch(actions.addNewAccount(1))).rejects.toThrow( 'error', ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#checkHardwareStatus', () => { afterEach(() => { sinon.restore(); }); it('calls checkHardwareStatus in background', async () => { const store = mockStore(); const checkHardwareStatus = background.checkHardwareStatus.callsFake( (_, __, cb) => { cb(); }, ); _setBackgroundConnection(background); await store.dispatch( actions.checkHardwareStatus( HardwareDeviceNames.ledger, `m/44'/60'/0'/0`, ), ); expect(checkHardwareStatus.callCount).toStrictEqual(1); }); it('shows loading indicator and displays error', async () => { const store = mockStore(); background.checkHardwareStatus.callsFake((_, __, cb) => cb(new Error('error')), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.checkHardwareStatus()), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#forgetDevice', () => { afterEach(() => { sinon.restore(); }); it('calls forgetDevice in background', async () => { const store = mockStore(); const forgetDevice = background.forgetDevice.callsFake((_, cb) => cb()); _setBackgroundConnection(background); await store.dispatch(actions.forgetDevice(HardwareDeviceNames.ledger)); expect(forgetDevice.callCount).toStrictEqual(1); }); it('shows loading indicator and displays error', async () => { const store = mockStore(); background.forgetDevice.callsFake((_, cb) => cb(new Error('error'))); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect(store.dispatch(actions.forgetDevice())).rejects.toThrow( 'error', ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#connectHardware', () => { afterEach(() => { sinon.restore(); }); it('calls connectHardware in background', async () => { const store = mockStore(); const connectHardware = background.connectHardware.callsFake( (_, __, ___, cb) => cb(), ); background.establishLedgerTransportPreference.callsFake((cb) => cb()); _setBackgroundConnection(background); await store.dispatch( actions.connectHardware( HardwareDeviceNames.ledger, 0, `m/44'/60'/0'/0`, ), ); expect(connectHardware.callCount).toStrictEqual(1); }); it('shows loading indicator and displays error', async () => { const store = mockStore(); background.connectHardware.callsFake((_, __, ___, cb) => cb(new Error('error')), ); background.establishLedgerTransportPreference.callsFake((cb) => cb()); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: 'Looking for your Ledger...', }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.connectHardware(HardwareDeviceNames.ledger)), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#unlockHardwareWalletAccount', () => { afterEach(() => { sinon.restore(); }); it('calls unlockHardwareWalletAccount in background', async () => { const store = mockStore(); const unlockHardwareWalletAccount = background.unlockHardwareWalletAccount.callsFake( (_, __, ___, ____, cb) => cb(), ); _setBackgroundConnection(background); await store.dispatch( actions.unlockHardwareWalletAccounts( [0], HardwareDeviceNames.ledger, `m/44'/60'/0'/0`, '', ), ); expect(unlockHardwareWalletAccount.callCount).toStrictEqual(1); }); it('shows loading indicator and displays error', async () => { const store = mockStore(); background.unlockHardwareWalletAccount.callsFake((_, __, ___, ____, cb) => cb(new Error('error')), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.unlockHardwareWalletAccounts([null])), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setCurrentCurrency', () => { afterEach(() => { sinon.restore(); }); it('calls setCurrentCurrency', async () => { const store = mockStore(); background.setCurrentCurrency = sinon.stub().callsFake((_, cb) => cb()); _setBackgroundConnection(background); await store.dispatch(actions.setCurrentCurrency('jpy')); expect(background.setCurrentCurrency.callCount).toStrictEqual(1); }); it('throws if setCurrentCurrency throws', async () => { const store = mockStore(); background.setCurrentCurrency = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.setCurrentCurrency()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#updateTransaction', () => { const txParams = { from: '0x1', gas: GAS_LIMITS.SIMPLE, gasPrice: '0x3b9aca00', to: '0x2', value: '0x0', }; const txData = { id: '1', status: TransactionStatus.unapproved, metamaskNetworkId: currentNetworkId, txParams, }; afterEach(() => { sinon.restore(); }); it('updates transaction', async () => { const store = mockStore(); const updateTransactionStub = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ updateTransaction: updateTransactionStub, getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.updateTransaction(txData)); const resultantActions = store.getActions(); expect(updateTransactionStub.callCount).toStrictEqual(1); expect(resultantActions[1]).toStrictEqual({ type: 'UPDATE_TRANSACTION_PARAMS', id: txData.id, value: txParams, }); }); it('rejects with error message', async () => { const store = mockStore(); background.getApi.returns({ updateTransaction: (_, callback) => { callback(new Error('error')); }, getState: sinon.stub().callsFake((cb) => cb(null, { currentLocale: 'test', selectedAddress: '0xFirstAddress', }), ), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'UPDATE_TRANSACTION_PARAMS', id: '1', value: { from: '0x1', gas: GAS_LIMITS.SIMPLE, gasPrice: '0x3b9aca00', to: '0x2', value: '0x0', }, }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'GO_HOME' }, ]; await expect( store.dispatch(actions.updateTransaction(txData)), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#lockMetamask', () => { afterEach(() => { sinon.restore(); }); it('calls setLocked', async () => { const store = mockStore(); const backgroundSetLocked = background.setLocked.callsFake((cb) => cb()); _setBackgroundConnection(background); await store.dispatch(actions.lockMetamask()); expect(backgroundSetLocked.callCount).toStrictEqual(1); }); it('returns display warning error with value when setLocked in background callback errors', async () => { const store = mockStore(); background.setLocked.callsFake((cb) => { cb(new Error('error')); }); _setBackgroundConnection(background); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'LOCK_METAMASK' }, ]; await store.dispatch(actions.lockMetamask()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setSelectedAddress', () => { afterEach(() => { sinon.restore(); }); it('calls setSelectedAddress in background', async () => { const store = mockStore(); const setSelectedAddressSpy = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ setSelectedAddress: setSelectedAddressSpy, }); _setBackgroundConnection(background.getApi()); await store.dispatch( actions.setSelectedAddress( '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc', ), ); expect(setSelectedAddressSpy.callCount).toStrictEqual(1); }); it('errors when setSelectedAddress throws', async () => { const store = mockStore(); const setSelectedAddressSpy = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); background.getApi.returns({ setSelectedAddress: setSelectedAddressSpy, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.setSelectedAddress()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setSelectedAccount', () => { afterEach(() => { sinon.restore(); }); it('#setSelectedAccount', async () => { const store = mockStore({ activeTab: {}, metamask: { alertEnabledness: {}, selectedAddress: '0x123' }, }); const setSelectedAddressSpy = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ setSelectedAddress: setSelectedAddressSpy, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setSelectedAccount()); expect(setSelectedAddressSpy.callCount).toStrictEqual(1); }); it('displays warning if setSelectedAccount throws', async () => { const store = mockStore({ activeTab: {}, metamask: { alertEnabledness: {}, selectedAddress: '0x123' }, }); const setSelectedAddressSpy = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); background.getApi.returns({ setSelectedAddress: setSelectedAddressSpy, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.setSelectedAccount()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#addToken', () => { afterEach(() => { sinon.restore(); }); it('calls addToken in background', async () => { const store = mockStore(); const addTokenStub = sinon .stub() .callsFake((_, __, ___, ____, cb) => cb()); background.getApi.returns({ addToken: addTokenStub, getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); _setBackgroundConnection(background.getApi()); await store.dispatch( actions.addToken({ address: '0x514910771af9ca656af840dff83e8264ecf986ca', symbol: 'LINK', decimals: 18, }), ); expect(addTokenStub.callCount).toStrictEqual(1); }); it('expected actions', async () => { const store = mockStore(); const tokenDetails = { address: 'tokenAddress', symbol: 'token', decimal: 18, }; const addTokenStub = sinon .stub() .callsFake((_, __, ___, ____, cb) => cb(null, tokenDetails)); background.getApi.returns({ addToken: addTokenStub, getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'UPDATE_METAMASK_STATE', value: baseMockState, }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch( actions.addToken({ address: '0x514910771af9ca656af840dff83e8264ecf986ca', symbol: 'LINK', decimals: 18, }), ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#ignoreTokens', () => { afterEach(() => { sinon.restore(); }); it('calls ignoreTokens in background', async () => { const store = mockStore(); const ignoreTokensStub = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ ignoreTokens: ignoreTokensStub, getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); _setBackgroundConnection(background.getApi()); await store.dispatch( actions.ignoreTokens({ tokensToIgnore: '0x0000001' }), ); expect(ignoreTokensStub.callCount).toStrictEqual(1); }); it('should display warning when ignoreTokens in background fails', async () => { const store = mockStore(); background.getApi.returns({ ignoreTokens: sinon.stub().callsFake((_, cb) => cb(new Error('error'))), getState: sinon.stub().callsFake((cb) => cb(null, baseMockState)), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'UPDATE_METAMASK_STATE', value: baseMockState, }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch( actions.ignoreTokens({ tokensToIgnore: '0x0000001' }), ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setProviderType', () => { afterEach(() => { sinon.restore(); }); it('calls setProviderType', async () => { const store = mockStore(); const setProviderTypeStub = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ setProviderType: setProviderTypeStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setProviderType()); expect(setProviderTypeStub.callCount).toStrictEqual(1); }); it('displays warning when setProviderType throws', async () => { const store = mockStore(); background.getApi.returns({ setProviderType: sinon .stub() .callsFake((_, cb) => cb(new Error('error'))), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'DISPLAY_WARNING', payload: 'Had a problem changing networks!', }, ]; await store.dispatch(actions.setProviderType()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setActiveNetwork', () => { afterEach(() => { sinon.restore(); }); it('calls setActiveNetwork in the background with the correct arguments', async () => { const store = mockStore(); const setCurrentNetworkStub = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ setActiveNetwork: setCurrentNetworkStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setActiveNetwork('networkConfigurationId')); expect( setCurrentNetworkStub.calledOnceWith('networkConfigurationId'), ).toBe(true); }); it('displays warning when setActiveNetwork throws', async () => { const store = mockStore(); const setCurrentNetworkStub = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); background.getApi.returns({ setActiveNetwork: setCurrentNetworkStub, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'DISPLAY_WARNING', payload: 'Had a problem changing networks!', }, ]; await store.dispatch(actions.setActiveNetwork()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#editAndSetNetworkConfiguration', () => { afterEach(() => { sinon.restore(); }); it('removes then re-adds the given network configuration', async () => { const store = mockStore(); const removeNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb()); const upsertNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb()); background.getApi.returns({ removeNetworkConfiguration: removeNetworkConfigurationStub, upsertNetworkConfiguration: upsertNetworkConfigurationStub, }); _setBackgroundConnection(background.getApi()); const networkConfiguration = { rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }; await store.dispatch( actions.editAndSetNetworkConfiguration( { ...networkConfiguration, networkConfigurationId: 'networkConfigurationId', }, { source: 'https://test-dapp.com' }, ), ); expect( removeNetworkConfigurationStub.calledOnceWith('networkConfigurationId'), ).toBe(true); expect( upsertNetworkConfigurationStub.calledOnceWith(networkConfiguration, { setActive: true, referrer: ORIGIN_METAMASK, source: 'https://test-dapp.com', }), ).toBe(true); }); it('displays warning when removeNetworkConfiguration throws', async () => { const store = mockStore(); const upsertNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb()); const removeNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); background.getApi.returns({ removeNetworkConfiguration: removeNetworkConfigurationStub, upsertNetworkConfiguration: upsertNetworkConfigurationStub, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'DISPLAY_WARNING', payload: 'Had a problem removing network!' }, ]; await store.dispatch( actions.editAndSetNetworkConfiguration( { networkConfigurationId: 'networkConfigurationId', rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }, { source: 'https://test-dapp.com' }, ), ); expect(store.getActions()).toStrictEqual(expectedActions); }); it('throws when no options object is passed as a second argument', async () => { const store = mockStore(); await expect(() => store.dispatch( actions.editAndSetNetworkConfiguration({ networkConfigurationId: 'networkConfigurationId', rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }), ), ).toThrow( "Cannot destructure property 'source' of 'undefined' as it is undefined.", ); }); }); describe('#upsertNetworkConfiguration', () => { afterEach(() => { sinon.restore(); }); it('calls upsertNetworkConfiguration in the background with the correct arguments', async () => { const store = mockStore(); const upsertNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb()); background.getApi.returns({ upsertNetworkConfiguration: upsertNetworkConfigurationStub, }); _setBackgroundConnection(background.getApi()); const networkConfiguration = { rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }; await store.dispatch( actions.upsertNetworkConfiguration(networkConfiguration, { source: MetaMetricsNetworkEventSource.CustomNetworkForm, }), ); expect( upsertNetworkConfigurationStub.calledOnceWith(networkConfiguration, { referrer: ORIGIN_METAMASK, source: MetaMetricsNetworkEventSource.CustomNetworkForm, setActive: undefined, }), ).toBe(true); }); it('throws when no options object is passed as a second argument', async () => { const store = mockStore(); await expect(() => store.dispatch( actions.upsertNetworkConfiguration({ networkConfigurationId: 'networkConfigurationId', rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }), ), ).toThrow( "Cannot destructure property 'setActive' of 'undefined' as it is undefined.", ); }); }); describe('#requestUserApproval', () => { afterEach(() => { sinon.restore(); }); it('calls requestUserApproval in the background with the correct arguments', async () => { const store = mockStore(); const requestUserApprovalStub = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ requestUserApproval: requestUserApprovalStub, }); _setBackgroundConnection(background.getApi()); const networkConfiguration = { rpcUrl: 'newRpc', chainId: '0x', ticker: 'ETH', nickname: 'nickname', rpcPrefs: { blockExplorerUrl: 'etherscan.io' }, }; await store.dispatch( actions.requestUserApproval({ origin: ORIGIN_METAMASK, type: 'test', requestData: networkConfiguration, }), ); expect( requestUserApprovalStub.calledOnceWith({ origin: ORIGIN_METAMASK, type: 'test', requestData: networkConfiguration, }), ).toBe(true); }); }); describe('#removeNetworkConfiguration', () => { afterEach(() => { sinon.restore(); }); it('calls removeNetworkConfiguration in the background with the correct arguments', async () => { const store = mockStore(); const removeNetworkConfigurationStub = sinon .stub() .callsFake((_, cb) => cb()); background.getApi.returns({ removeNetworkConfiguration: removeNetworkConfigurationStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch( actions.removeNetworkConfiguration('testNetworkConfigurationId'), ); expect( removeNetworkConfigurationStub.calledOnceWith( 'testNetworkConfigurationId', ), ).toBe(true); }); }); describe('#setSelectedNetworkConfigurationId', () => { it('sets appState.networkConfigurationId to provided value', async () => { const store = mockStore(); const networkConfigurationId = 'testNetworkConfigurationId'; store.dispatch( actions.setSelectedNetworkConfigurationId(networkConfigurationId), ); const resultantActions = store.getActions(); expect(resultantActions[0]).toStrictEqual({ type: 'SET_SELECTED_NETWORK_CONFIGURATION_ID', payload: networkConfigurationId, }); }); }); describe('#setNewNetworkAdded', () => { it('sets appState.setNewNetworkAdded to provided value', async () => { const store = mockStore(); const newNetworkAddedDetails = { networkConfigurationId: 'testNetworkConfigurationId', nickname: 'test-chain', }; store.dispatch(actions.setNewNetworkAdded(newNetworkAddedDetails)); const resultantActions = store.getActions(); expect(resultantActions[0]).toStrictEqual({ type: 'SET_NEW_NETWORK_ADDED', payload: newNetworkAddedDetails, }); }); }); describe('#addToAddressBook', () => { it('calls setAddressBook', async () => { const store = mockStore(); const setAddressBookStub = sinon .stub() .callsFake((_, __, ___, ____, cb) => cb()); background.getApi.returns({ setAddressBook: setAddressBookStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.addToAddressBook('0x0000')); expect(setAddressBookStub.callCount).toStrictEqual(1); sinon.restore(); }); }); describe('#exportAccount', () => { afterEach(() => { sinon.restore(); }); it('returns expected actions for successful action', async () => { const store = mockStore(); const testPrivKey = 'a-test-priv-key'; const verifyPasswordStub = sinon.stub().callsFake((_, cb) => cb()); const exportAccountStub = sinon .stub() .callsFake((_, _2, cb) => cb(null, testPrivKey)); background.getApi.returns({ verifyPassword: verifyPasswordStub, exportAccount: exportAccountStub, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'SHOW_PRIVATE_KEY', payload: testPrivKey, }, ]; await store.dispatch( actions.exportAccount('a-test-password', '0xAddress'), ); expect(verifyPasswordStub.callCount).toStrictEqual(1); expect(exportAccountStub.callCount).toStrictEqual(1); expect(store.getActions()).toStrictEqual(expectedActions); }); it('returns action errors when first func callback errors', async () => { const store = mockStore(); const verifyPasswordStub = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); background.getApi.returns({ verifyPassword: verifyPasswordStub, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'Incorrect Password.' }, ]; await expect( store.dispatch(actions.exportAccount('a-test-password', '0xAddress')), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); it('returns action errors when second func callback errors', async () => { const store = mockStore(); const verifyPasswordStub = sinon.stub().callsFake((_, cb) => cb()); const exportAccountStub = sinon .stub() .callsFake((_, _2, cb) => cb(new Error('error'))); background.getApi.returns({ verifyPassword: verifyPasswordStub, exportAccount: exportAccountStub, }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'Had a problem exporting the account.', }, ]; await expect( store.dispatch(actions.exportAccount('a-test-password', '0xAddress')), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setAccountLabel', () => { afterEach(() => { sinon.restore(); }); it('calls setAccountLabel', async () => { const store = mockStore(); const setAccountLabelStub = sinon.stub().callsFake((_, __, cb) => cb()); background.getApi.returns({ setAccountLabel: setAccountLabelStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch( actions.setAccountLabel( '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc', 'test', ), ); expect(setAccountLabelStub.callCount).toStrictEqual(1); }); it('returns action errors when func callback errors', async () => { const store = mockStore(); background.getApi.returns({ setAccountLabel: sinon .stub() .callsFake((_, __, cb) => cb(new Error('error'))), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; await expect( store.dispatch( actions.setAccountLabel( '0x0dcd5d886577d5081b0c52e242ef29e70be3e7bc', 'test', ), ), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setFeatureFlag', () => { afterEach(() => { sinon.restore(); }); it('calls setFeatureFlag in the background', async () => { const store = mockStore(); const setFeatureFlagStub = sinon.stub().callsFake((_, __, cb) => cb()); background.getApi.returns({ setFeatureFlag: setFeatureFlagStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setFeatureFlag()); expect(setFeatureFlagStub.callCount).toStrictEqual(1); }); it('errors when setFeatureFlag in background throws', async () => { const store = mockStore(); background.getApi.returns({ setFeatureFlag: sinon .stub() .callsFake((_, __, cb) => cb(new Error('error'))), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; await expect(store.dispatch(actions.setFeatureFlag())).rejects.toThrow( 'error', ); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setCompletedOnboarding', () => { afterEach(() => { sinon.restore(); }); it('completes onboarding', async () => { const store = mockStore(); const completeOnboardingStub = sinon.stub().callsFake((cb) => cb()); background.getApi.returns({ completeOnboarding: completeOnboardingStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setCompletedOnboarding()); expect(completeOnboardingStub.callCount).toStrictEqual(1); }); it('errors when setCompletedOnboarding in background throws', async () => { const store = mockStore(); background.getApi.returns({ completeOnboarding: sinon .stub() .callsFake((cb) => cb(new Error('error'))), }); _setBackgroundConnection(background.getApi()); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await expect( store.dispatch(actions.setCompletedOnboarding()), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setParticipateInMetaMetrics', () => { beforeAll(() => { window.sentry = { toggleSession: jest.fn(), endSession: jest.fn(), }; }); it('sets participateInMetaMetrics to true', async () => { const store = mockStore(); const setParticipateInMetaMetricsStub = jest.fn((_, cb) => cb()); background.getApi.returns({ setParticipateInMetaMetrics: setParticipateInMetaMetricsStub, }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setParticipateInMetaMetrics(true)); expect(setParticipateInMetaMetricsStub).toHaveBeenCalledWith( true, expect.anything(), ); expect(window.sentry.toggleSession).toHaveBeenCalled(); }); }); describe('#setUseBlockie', () => { afterEach(() => { sinon.restore(); }); it('calls setUseBlockie in background', async () => { const store = mockStore(); const setUseBlockieStub = sinon.stub().callsFake((_, cb) => cb()); _setBackgroundConnection({ setUseBlockie: setUseBlockieStub }); await store.dispatch(actions.setUseBlockie()); expect(setUseBlockieStub.callCount).toStrictEqual(1); }); it('errors when setUseBlockie in background throws', async () => { const store = mockStore(); const setUseBlockieStub = sinon.stub().callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection({ setUseBlockie: setUseBlockieStub }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; await store.dispatch(actions.setUseBlockie()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setUsePhishDetect', () => { afterEach(() => { sinon.restore(); }); it('calls setUsePhishDetect in background', () => { const store = mockStore(); const setUsePhishDetectStub = sinon.stub().callsFake((_, cb) => cb()); _setBackgroundConnection({ setUsePhishDetect: setUsePhishDetectStub, }); store.dispatch(actions.setUsePhishDetect()); expect(setUsePhishDetectStub.callCount).toStrictEqual(1); }); it('errors when setUsePhishDetect in background throws', () => { const store = mockStore(); const setUsePhishDetectStub = sinon.stub().callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection({ setUsePhishDetect: setUsePhishDetectStub, }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; store.dispatch(actions.setUsePhishDetect()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setUseMultiAccountBalanceChecker', () => { afterEach(() => { sinon.restore(); }); it('calls setUseMultiAccountBalanceChecker in background', () => { const store = mockStore(); const setUseMultiAccountBalanceCheckerStub = sinon .stub() .callsFake((_, cb) => cb()); _setBackgroundConnection({ setUseMultiAccountBalanceChecker: setUseMultiAccountBalanceCheckerStub, }); store.dispatch(actions.setUseMultiAccountBalanceChecker()); expect(setUseMultiAccountBalanceCheckerStub.callCount).toStrictEqual(1); }); it('errors when setUseMultiAccountBalanceChecker in background throws', () => { const store = mockStore(); const setUseMultiAccountBalanceCheckerStub = sinon .stub() .callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection({ setUseMultiAccountBalanceChecker: setUseMultiAccountBalanceCheckerStub, }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'HIDE_LOADING_INDICATION' }, { type: 'DISPLAY_WARNING', payload: 'error' }, ]; store.dispatch(actions.setUseMultiAccountBalanceChecker()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#setUse4ByteResolution', () => { afterEach(() => { sinon.restore(); }); it('calls setUse4ByteResolution in background', async () => { const store = mockStore(); const setUse4ByteResolutionStub = sinon.stub().callsFake((_, cb) => cb()); _setBackgroundConnection({ setUse4ByteResolution: setUse4ByteResolutionStub, }); await store.dispatch(actions.setUse4ByteResolution()); expect(setUse4ByteResolutionStub.callCount).toStrictEqual(1); }); it('errors when setUse4ByteResolution in background throws', async () => { const store = mockStore(); const setUse4ByteResolutionStub = sinon.stub().callsFake((_, cb) => { cb(new Error('error')); }); _setBackgroundConnection({ setUse4ByteResolution: setUse4ByteResolutionStub, }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.setUse4ByteResolution()); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#updateCurrentLocale', () => { beforeEach(() => { sinon.stub(window, 'fetch').resolves({ json: async () => enLocale, }); }); afterEach(() => { sinon.restore(); }); it('calls expected actions', async () => { const store = mockStore(); const setCurrentLocaleStub = sinon.stub().callsFake((_, cb) => cb()); _setBackgroundConnection({ setCurrentLocale: setCurrentLocaleStub, }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'SET_CURRENT_LOCALE', payload: { locale: 'test', messages: enLocale }, }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.updateCurrentLocale('test')); expect(setCurrentLocaleStub.callCount).toStrictEqual(1); expect(store.getActions()).toStrictEqual(expectedActions); }); it('errors when setCurrentLocale throws', async () => { const store = mockStore(); const setCurrentLocaleStub = sinon .stub() .callsFake((_, cb) => cb(new Error('error'))); _setBackgroundConnection({ setCurrentLocale: setCurrentLocaleStub, }); const expectedActions = [ { type: 'SHOW_LOADING_INDICATION', payload: undefined }, { type: 'DISPLAY_WARNING', payload: 'error' }, { type: 'HIDE_LOADING_INDICATION' }, ]; await store.dispatch(actions.updateCurrentLocale('test')); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#markPasswordForgotten', () => { afterEach(() => { sinon.restore(); }); it('calls markPasswordForgotten', async () => { const store = mockStore(); background.markPasswordForgotten.callsFake((cb) => cb()); _setBackgroundConnection(background); await store.dispatch(actions.markPasswordForgotten()); expect(background.markPasswordForgotten.callCount).toStrictEqual(1); }); it('errors when markPasswordForgotten throws', async () => { const store = mockStore(); background.markPasswordForgotten.callsFake((cb) => cb(new Error('error')), ); _setBackgroundConnection(background); const expectedActions = [ { type: 'HIDE_LOADING_INDICATION' }, { type: 'UPDATE_METAMASK_STATE', value: baseMockState, }, ]; await expect( store.dispatch(actions.markPasswordForgotten('test')), ).rejects.toThrow('error'); expect(store.getActions()).toStrictEqual(expectedActions); }); }); describe('#unMarkPasswordForgotten', () => { it('calls unMarkPasswordForgotten', async () => { const store = mockStore(); background.unMarkPasswordForgotten.callsFake((cb) => cb()); _setBackgroundConnection(background); store.dispatch(actions.unMarkPasswordForgotten()); expect(background.unMarkPasswordForgotten.callCount).toStrictEqual(1); }); }); describe('#displayWarning', () => { it('sets appState.warning to provided value', async () => { const store = mockStore(); const warningText = 'This is a sample warning message'; store.dispatch(actions.displayWarning(warningText)); const resultantActions = store.getActions(); expect(resultantActions[0]).toStrictEqual({ type: 'DISPLAY_WARNING', payload: warningText, }); }); }); describe('#cancelTx', () => { it('creates COMPLETED_TX with the cancelled transaction ID', async () => { const store = mockStore(); background.getApi.returns({ rejectPendingApproval: sinon.stub().callsFake((_1, _2, cb) => { cb(); }), getState: sinon.stub().callsFake((cb) => cb(null, { currentLocale: 'test', selectedAddress: '0xFirstAddress', providerConfig: { chainId: '0x1', }, accounts: { '0xFirstAddress': { balance: '0x0', }, }, cachedBalances: { '0x1': { '0xFirstAddress': '0x0', }, }, }), ), }); _setBackgroundConnection(background.getApi()); const txId = 1457634084250832; await store.dispatch(actions.cancelTx({ id: txId })); const resultantActions = store.getActions(); const expectedAction = resultantActions.find( (action) => action.type === 'COMPLETED_TX', ); expect(expectedAction.value.id).toStrictEqual(txId); }); }); describe('Desktop', () => { describe('#setDesktopEnabled', () => { it('calls background setDesktopEnabled method', async () => { const store = mockStore(); const setDesktopEnabled = sinon.stub().callsFake((_, cb) => cb()); background.getApi.returns({ setDesktopEnabled, getState: sinon.stub().callsFake((cb) => cb(null, { desktopEnabled: true, }), ), }); _setBackgroundConnection(background.getApi()); await store.dispatch(actions.setDesktopEnabled(true)); expect(setDesktopEnabled.calledOnceWith(true)).toBeTruthy(); }); }); }); });