test(extension): mock initializeTrezorTransport method

input-output-hk · Feb 27, 2024 · bd11e2e · bd11e2e
1 parent e4ed066
commit bd11e2e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/apps/browser-extension-wallet/src/hooks/__tests__/useWalletManager.test.tsx b/apps/browser-extension-wallet/src/hooks/__tests__/useWalletManager.test.tsx
@@ -17,6 +17,7 @@ const mockRestoreWalletFromKeyAgent = jest.fn();
 const mockSwitchKeyAgents = jest.fn();
 const mockLedgerGetXpub = jest.fn();
 const mockTrezorGetXpub = jest.fn();
+const mockInitializeTrezorTransport = jest.fn();
 const mockLedgerCreateWithDevice = jest.fn();
 const mockUseAppSettingsContext = jest.fn().mockReturnValue([{}, jest.fn()]);
 import React from 'react';
@@ -70,7 +71,8 @@ jest.mock('@lace/cardano', () => {
       },
       Trezor: {
         TrezorKeyAgent: {
-          getXpub: mockTrezorGetXpub
+          getXpub: mockTrezorGetXpub,
+          initializeTrezorTransport: mockInitializeTrezorTransport
         }
       },
       restoreWalletFromKeyAgent: mockRestoreWalletFromKeyAgent,