From c2bbbb1dbc6f5d5f3c811476d6db1e144f96daf3 Mon Sep 17 00:00:00 2001 From: Elliot Winkler Date: Fri, 26 May 2023 10:35:52 -0600 Subject: [PATCH] NetworkController: Simplify event types (#19252) In the `core` version of this controller, event types are not held in an enum but are specified directly as strings instead. Simplifying the event types will make it easier to compare the NetworkController unit tests between extension and core. --- app/scripts/controllers/detect-tokens.test.js | 4 +- .../network/network-controller.test.ts | 103 +++++++++--------- .../controllers/network/network-controller.ts | 56 +++------- app/scripts/metamask-controller.js | 32 +++--- 4 files changed, 82 insertions(+), 113 deletions(-) diff --git a/app/scripts/controllers/detect-tokens.test.js b/app/scripts/controllers/detect-tokens.test.js index 8e7bddd28..a85230668 100644 --- a/app/scripts/controllers/detect-tokens.test.js +++ b/app/scripts/controllers/detect-tokens.test.js @@ -13,7 +13,7 @@ import { convertHexToDecimal } from '@metamask/controller-utils'; import { NETWORK_TYPES } from '../../../shared/constants/network'; import { toChecksumHexAddress } from '../../../shared/modules/hexstring-utils'; import DetectTokensController from './detect-tokens'; -import { NetworkController, NetworkControllerEventType } from './network'; +import { NetworkController } from './network'; import PreferencesController from './preferences'; describe('DetectTokensController', function () { @@ -248,7 +248,7 @@ describe('DetectTokensController', function () { ), onNetworkStateChange: (cb) => networkControllerMessenger.subscribe( - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', () => { const networkState = network.store.getState(); const modifiedNetworkState = { diff --git a/app/scripts/controllers/network/network-controller.test.ts b/app/scripts/controllers/network/network-controller.test.ts index b1362a92e..abb1a3bfa 100644 --- a/app/scripts/controllers/network/network-controller.test.ts +++ b/app/scripts/controllers/network/network-controller.test.ts @@ -19,7 +19,6 @@ import { MetaMetricsNetworkEventSource } from '../../../../shared/constants/meta import { NetworkController, NetworkControllerEvent, - NetworkControllerEventType, NetworkControllerOptions, NetworkControllerState, ProviderConfiguration, @@ -871,7 +870,7 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await controller.lookupNetwork(); @@ -890,7 +889,7 @@ describe('NetworkController', () => { const promiseForNoInfuraIsBlockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await controller.lookupNetwork(); @@ -954,7 +953,7 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await controller.lookupNetwork(); @@ -985,7 +984,7 @@ describe('NetworkController', () => { const promiseForNoInfuraIsBlockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await controller.lookupNetwork(); @@ -1368,12 +1367,12 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, }); const promiseForInfuraIsBlocked = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', }); await controller.lookupNetwork(); @@ -1788,12 +1787,12 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, }); const promiseForInfuraIsBlocked = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', }); await controller.lookupNetwork(); @@ -2198,12 +2197,12 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, }); const promiseForInfuraIsBlocked = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', }); await controller.lookupNetwork(); @@ -2582,7 +2581,7 @@ describe('NetworkController', () => { operation: async () => { const networkWillChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkWillChange, + eventType: 'NetworkController:networkWillChange', operation: async () => { await controller.rollbackToPreviousProvider(); }, @@ -2957,7 +2956,7 @@ describe('NetworkController', () => { operation: async () => { const networkDidChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkDidChange, + eventType: 'NetworkController:networkDidChange', operation: async () => { await controller.rollbackToPreviousProvider(); }, @@ -3024,12 +3023,12 @@ describe('NetworkController', () => { const promiseForNoInfuraIsUnblockedEvents = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, }); const promiseForInfuraIsBlocked = waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', }); await waitForLookupNetworkToComplete({ @@ -3321,7 +3320,7 @@ describe('NetworkController', () => { operation: async () => { const networkWillChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkWillChange, + eventType: 'NetworkController:networkWillChange', operation: async () => { await controller.rollbackToPreviousProvider(); }, @@ -3644,7 +3643,7 @@ describe('NetworkController', () => { operation: async () => { const networkDidChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkDidChange, + eventType: 'NetworkController:networkDidChange', operation: async () => { await controller.rollbackToPreviousProvider(); }, @@ -3695,7 +3694,7 @@ describe('NetworkController', () => { operation: async () => { const infuraIsUnblocked = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await controller.rollbackToPreviousProvider(); }, @@ -4537,7 +4536,7 @@ function refreshNetworkTests({ const networkWillChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkWillChange, + eventType: 'NetworkController:networkWillChange', operation: () => { // Intentionally not awaited because we're capturing an event // emitted partway through the operation @@ -4562,7 +4561,7 @@ function refreshNetworkTests({ const networkDidChange = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.NetworkDidChange, + eventType: 'NetworkController:networkDidChange', operation: () => { // Intentionally not awaited because we're capturing an event // emitted partway through the operation @@ -5083,7 +5082,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5106,7 +5105,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5162,7 +5161,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5191,7 +5190,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5222,7 +5221,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5278,7 +5277,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5307,7 +5306,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5360,7 +5359,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5390,7 +5389,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', operation: async () => { await operation(controller); }, @@ -5446,7 +5445,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5475,7 +5474,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5506,7 +5505,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5562,7 +5561,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5591,7 +5590,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5622,7 +5621,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5684,7 +5683,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5716,7 +5715,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5750,7 +5749,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5812,7 +5811,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -5844,7 +5843,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -5903,7 +5902,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -5936,7 +5935,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', operation: async () => { await operation(controller); }, @@ -5997,7 +5996,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', operation: async () => { await operation(controller); }, @@ -6029,7 +6028,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsUnblocked, + eventType: 'NetworkController:infuraIsUnblocked', count: 0, operation: async () => { await operation(controller); @@ -6063,7 +6062,7 @@ function lookupNetworkTests({ const payloads = await waitForPublishedEvents({ messenger, - eventType: NetworkControllerEventType.InfuraIsBlocked, + eventType: 'NetworkController:infuraIsBlocked', count: 0, operation: async () => { await operation(controller); @@ -6093,17 +6092,13 @@ function buildMessenger() { * @returns The network controller restricted messenger. */ function buildNetworkControllerMessenger(messenger = buildMessenger()) { - return messenger.getRestricted< - 'NetworkController', - never, - NetworkControllerEventType - >({ + return messenger.getRestricted({ name: 'NetworkController', allowedEvents: [ - NetworkControllerEventType.NetworkDidChange, - NetworkControllerEventType.NetworkWillChange, - NetworkControllerEventType.InfuraIsBlocked, - NetworkControllerEventType.InfuraIsUnblocked, + 'NetworkController:networkDidChange', + 'NetworkController:networkWillChange', + 'NetworkController:infuraIsBlocked', + 'NetworkController:infuraIsUnblocked', ], }); } diff --git a/app/scripts/controllers/network/network-controller.ts b/app/scripts/controllers/network/network-controller.ts index fdb14b0f1..2f4fd9763 100644 --- a/app/scripts/controllers/network/network-controller.ts +++ b/app/scripts/controllers/network/network-controller.ts @@ -82,35 +82,13 @@ type NetworkConfigurationId = string; */ type ChainId = Hex; -/** - * The set of event types that NetworkController can publish via its messenger. - */ -export enum NetworkControllerEventType { - /** - * @see {@link NetworkControllerNetworkDidChangeEvent} - */ - NetworkDidChange = 'NetworkController:networkDidChange', - /** - * @see {@link NetworkControllerNetworkWillChangeEvent} - */ - NetworkWillChange = 'NetworkController:networkWillChange', - /** - * @see {@link NetworkControllerInfuraIsBlockedEvent} - */ - InfuraIsBlocked = 'NetworkController:infuraIsBlocked', - /** - * @see {@link NetworkControllerInfuraIsUnblockedEvent} - */ - InfuraIsUnblocked = 'NetworkController:infuraIsUnblocked', -} - /** * `networkWillChange` is published when the current network is about to be * switched, but the new provider has not been created and no state changes have * occurred yet. */ export type NetworkControllerNetworkWillChangeEvent = { - type: NetworkControllerEventType.NetworkWillChange; + type: 'NetworkController:networkWillChange'; payload: []; }; @@ -119,7 +97,7 @@ export type NetworkControllerNetworkWillChangeEvent = { * switched network (but before the network has been confirmed to be available). */ export type NetworkControllerNetworkDidChangeEvent = { - type: NetworkControllerEventType.NetworkDidChange; + type: 'NetworkController:networkDidChange'; payload: []; }; @@ -129,7 +107,7 @@ export type NetworkControllerNetworkDidChangeEvent = { * location. */ export type NetworkControllerInfuraIsBlockedEvent = { - type: NetworkControllerEventType.InfuraIsBlocked; + type: 'NetworkController:infuraIsBlocked'; payload: []; }; @@ -139,7 +117,7 @@ export type NetworkControllerInfuraIsBlockedEvent = { * their location, or the network is switched to a non-Infura network. */ export type NetworkControllerInfuraIsUnblockedEvent = { - type: NetworkControllerEventType.InfuraIsUnblocked; + type: 'NetworkController:infuraIsUnblocked'; payload: []; }; @@ -159,8 +137,8 @@ export type NetworkControllerMessenger = RestrictedControllerMessenger< typeof name, never, NetworkControllerEvent, - never, - NetworkControllerEventType + string, + string >; /** @@ -611,14 +589,11 @@ export class NetworkController extends EventEmitter { const listener = () => { networkChanged = true; this.#messenger.unsubscribe( - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', listener, ); }; - this.#messenger.subscribe( - NetworkControllerEventType.NetworkDidChange, - listener, - ); + this.#messenger.subscribe('NetworkController:networkDidChange', listener); try { const results = await Promise.all([ @@ -664,10 +639,7 @@ export class NetworkController extends EventEmitter { // in the process of being called, so we don't need to go further. return; } - this.#messenger.unsubscribe( - NetworkControllerEventType.NetworkDidChange, - listener, - ); + this.#messenger.unsubscribe('NetworkController:networkDidChange', listener); this.store.updateState({ networkStatus, @@ -692,15 +664,15 @@ export class NetworkController extends EventEmitter { if (isInfura) { if (networkStatus === NetworkStatus.Available) { - this.#messenger.publish(NetworkControllerEventType.InfuraIsUnblocked); + this.#messenger.publish('NetworkController:infuraIsUnblocked'); } else if (networkStatus === NetworkStatus.Blocked) { - this.#messenger.publish(NetworkControllerEventType.InfuraIsBlocked); + this.#messenger.publish('NetworkController:infuraIsBlocked'); } } else { // Always publish infuraIsUnblocked regardless of network status to // prevent consumers from being stuck in a blocked state if they were // previously connected to an Infura network that was blocked - this.#messenger.publish(NetworkControllerEventType.InfuraIsUnblocked); + this.#messenger.publish('NetworkController:infuraIsUnblocked'); } } @@ -892,12 +864,12 @@ export class NetworkController extends EventEmitter { * the new network. */ async #switchNetwork(providerConfig: ProviderConfiguration) { - this.#messenger.publish(NetworkControllerEventType.NetworkWillChange); + this.#messenger.publish('NetworkController:networkWillChange'); this.#resetNetworkId(); this.#resetNetworkStatus(); this.#resetNetworkDetails(); this.#configureProvider(providerConfig); - this.#messenger.publish(NetworkControllerEventType.NetworkDidChange); + this.#messenger.publish('NetworkController:networkDidChange'); await this.lookupNetwork(); } diff --git a/app/scripts/metamask-controller.js b/app/scripts/metamask-controller.js index 2ac98c2f9..892f0ec4d 100644 --- a/app/scripts/metamask-controller.js +++ b/app/scripts/metamask-controller.js @@ -140,10 +140,7 @@ import createTabIdMiddleware from './lib/createTabIdMiddleware'; import createOnboardingMiddleware from './lib/createOnboardingMiddleware'; import { setupMultiplex } from './lib/stream-utils'; import EnsController from './controllers/ens'; -import { - NetworkController, - NetworkControllerEventType, -} from './controllers/network'; +import { NetworkController } from './controllers/network'; import PreferencesController from './controllers/preferences'; import AppStateController from './controllers/app-state'; import CachedBalancesController from './controllers/cached-balances'; @@ -269,7 +266,12 @@ export default class MetamaskController extends EventEmitter { const networkControllerMessenger = this.controllerMessenger.getRestricted({ name: 'NetworkController', - allowedEvents: Object.values(NetworkControllerEventType), + allowedEvents: [ + 'NetworkController:networkWillChange', + 'NetworkController:networkDidChange', + 'NetworkController:infuraIsBlocked', + 'NetworkController:infuraIsUnblocked', + ], }); this.networkController = new NetworkController({ messenger: networkControllerMessenger, @@ -316,11 +318,11 @@ export default class MetamaskController extends EventEmitter { initLangCode: opts.initLangCode, onInfuraIsBlocked: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.InfuraIsBlocked, + 'NetworkController:infuraIsBlocked', ), onInfuraIsUnblocked: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.InfuraIsUnblocked, + 'NetworkController:infuraIsUnblocked', ), tokenListController: this.tokenListController, provider: this.provider, @@ -366,7 +368,7 @@ export default class MetamaskController extends EventEmitter { // network provider by one update. onNetworkStateChange: (cb) => networkControllerMessenger.subscribe( - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', () => { const networkState = this.networkController.store.getState(); const modifiedNetworkState = { @@ -477,7 +479,7 @@ export default class MetamaskController extends EventEmitter { preferencesStore: this.preferencesController.store, onNetworkDidChange: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', ), getNetworkIdentifier: () => { const { type, rpcUrl } = @@ -516,7 +518,7 @@ export default class MetamaskController extends EventEmitter { // onNetworkDidChange onNetworkStateChange: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', ), getCurrentNetworkEIP1559Compatibility: this.networkController.getEIP1559Compatibility.bind( @@ -642,7 +644,7 @@ export default class MetamaskController extends EventEmitter { this.networkController.store.getState().providerConfig.chainId, onNetworkDidChange: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', ), }); @@ -654,7 +656,7 @@ export default class MetamaskController extends EventEmitter { blockTracker: this.blockTracker, onNetworkDidChange: networkControllerMessenger.subscribe.bind( networkControllerMessenger, - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', ), getCurrentChainId: () => this.networkController.store.getState().providerConfig.chainId, @@ -1151,7 +1153,7 @@ export default class MetamaskController extends EventEmitter { }); networkControllerMessenger.subscribe( - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', async () => { const { ticker } = this.networkController.store.getState().providerConfig; @@ -1279,7 +1281,7 @@ export default class MetamaskController extends EventEmitter { // ensure accountTracker updates balances after network change networkControllerMessenger.subscribe( - NetworkControllerEventType.NetworkDidChange, + 'NetworkController:networkDidChange', () => { this.accountTracker._updateAccounts(); }, @@ -1287,7 +1289,7 @@ export default class MetamaskController extends EventEmitter { // clear unapproved transactions and messages when the network will change networkControllerMessenger.subscribe( - NetworkControllerEventType.NetworkWillChange, + 'NetworkController:networkWillChange', () => { this.txController.txStateManager.clearUnapprovedTxs(); this.encryptionPublicKeyController.clearUnapproved();