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

…d` action
MetaMask · Mar 9, 2024 · 5d43932 · 5d43932
1 parent b1d99d8
commit 5d43932
Showing 1 changed file with 4 additions and 5 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',
@@ -266,10 +266,9 @@ describe('MetaMaskController', function () {
         decimals,
         networkClientId: 'networkClientId1',
       });
-      assert.strictEqual(
-        metamaskController.tokensController.getNetworkClientById.getCall(0)
-          .args[0],
-        'networkClientId1',
+      assert.deepStrictEqual(
+        metamaskController.controllerMessenger.call.getCall(0).args,
+        ['NetworkController:getNetworkClientById', 'networkClientId1'],
       );
     });
   });