test: Fix unit test to use messenger pattern for `getNetworkClientByI…

…d` action
MetaMask · Mar 9, 2024 · c6a3a05 · c6a3a05
1 parent b1d99d8
commit c6a3a05
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/app/scripts/metamask-controller.actions.test.js b/app/scripts/metamask-controller.actions.test.js
@@ -253,7 +253,7 @@ describe('MetaMaskController', function () {
           Promise.resolve({ supportsInterface: supportsInterfaceStub }),
         );
       sinon
-        .stub(metamaskController.tokensController, 'getNetworkClientById')
+        .stub(metamaskController.controllerMessenger, 'call')
         .callsFake(() => ({
           configuration: {
             chainId: '0xa',
@@ -267,8 +267,11 @@ describe('MetaMaskController', function () {
         networkClientId: 'networkClientId1',
       });
       assert.strictEqual(
-        metamaskController.tokensController.getNetworkClientById.getCall(0)
-          .args[0],
+        metamaskController.controllerMessenger.call.getCall(0).args[0],
+        'NetworkController:getNetworkClientById',
+      );
+      assert.strictEqual(
+        metamaskController.controllerMessenger.call.getCall(0).args[1],
         'networkClientId1',
       );
     });