[token-detection-controller] test: Restore publishes for `networkDidC…

…hange` event, mock polygon network client
MetaMask · Jan 8, 2024 · a49fbb1 · a49fbb1
1 parent 4e65ccc
commit a49fbb1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/packages/assets-controllers/jest.config.js b/packages/assets-controllers/jest.config.js
@@ -17,7 +17,7 @@ module.exports = merge(baseConfig, {
   // An object that configures minimum threshold enforcement for coverage results
   coverageThreshold: {
     global: {
-      branches: 88.36,
+      branches: 89.09,
       functions: 97.08,
       lines: 97.23,
       statements: 97.28,

diff --git a/packages/assets-controllers/src/TokenDetectionController.test.ts b/packages/assets-controllers/src/TokenDetectionController.test.ts
@@ -10,11 +10,8 @@ import {
 } from '@metamask/controller-utils';
 import { defaultState as defaultNetworkState } from '@metamask/network-controller';
 import type {
-  BlockTrackerProxy,
   NetworkController,
-  NetworkState,
   ProviderConfig,
-  ProviderProxy,
 } from '@metamask/network-controller';
 import type { AutoManagedNetworkClient } from '@metamask/network-controller/src/create-auto-managed-network-client';
 import type {
@@ -177,7 +174,6 @@ describe('TokenDetectionController', () => {
     ReturnType<AssetsContractController['getBalancesInSingleCall']>
   >;
 
-  const networkDidChangeEvent = jest.fn<NetworkState, []>();
   let getNetworkClientByIdHandler = jest.fn();
   const getNetworkClientByIdInfuraHandler = jest.fn<
     AutoManagedNetworkClient<InfuraNetworkClientConfiguration>,
@@ -188,7 +184,7 @@ describe('TokenDetectionController', () => {
     Parameters<NetworkController['getNetworkClientById']>
   >();
   const changeNetwork = (providerConfig: ProviderConfig) => {
-    networkDidChangeEvent.mockReturnValue({
+    controllerMessenger.publish('NetworkController:networkDidChange', {
       ...defaultNetworkState,
       providerConfig,
       selectedNetworkClientId: providerConfig.id ?? providerConfig.type,
@@ -242,9 +238,6 @@ describe('TokenDetectionController', () => {
       ...mainnet,
       type: NetworkClientType.Infura,
     },
-    provider: {} as ProviderProxy,
-    blockTracker: {} as BlockTrackerProxy,
-    destroy: jest.fn(),
   };
 
   beforeEach(async () => {