test: Fix tests

Author: rekmarks

packages/selected-network-controller/tests/SelectedNetworkController.test.ts

@@ -13,7 +13,10 @@ import {
   getDefaultNetworkControllerState,
   RpcEndpointType,
 } from '@metamask/network-controller';
-import { createSwappableProxy } from '@metamask/swappable-obj-proxy';
+import {
+  createEventEmitterProxy,
+  createSwappableProxy,
+} from '@metamask/swappable-obj-proxy';
 import type { Hex } from '@metamask/utils';
 
 import type {
@@ -169,17 +172,23 @@ const setup = ({
     once: jest.fn(),
   };
 
-  const createSwappableProxyMock = jest.mocked(createSwappableProxy);
+  // Non-polluting use of any for test mock.
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  createSwappableProxyMock.mockImplementation((initialTarget: any) => {
+  const swappableProxyImplementation = (initialTarget: any) => {
     if (initialTarget?.request !== undefined) {
       return mockProviderProxy;
     }
     if (initialTarget?.getLatestBlock !== undefined) {
       return mockBlockTrackerProxy;
     }
     return mockProviderProxy;
-  });
+  };
+  const createSwappableProxyMock = jest
+    .mocked(createSwappableProxy)
+    .mockImplementation(swappableProxyImplementation);
+  const createEventEmitterProxyMock = jest
+    .mocked(createEventEmitterProxy)
+    .mockImplementation(swappableProxyImplementation);
   const rootMessenger = getRootMessenger();
   const { messenger, ...mockMessengerActions } =
     buildSelectedNetworkControllerMessenger({
@@ -201,6 +210,7 @@ const setup = ({
     mockBlockTrackerProxy,
     domainProxyMap,
     createSwappableProxyMock,
+    createEventEmitterProxyMock,
     ...mockMessengerActions,
   };
 };