Update ProtocolSetupLabware.test.tsx

Opentrons · Feb 29, 2024 · ed01ecf · ed01ecf
1 parent 2855a85
commit ed01ecf
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/app/src/organisms/ProtocolSetupLabware/__tests__/ProtocolSetupLabware.test.tsx b/app/src/organisms/ProtocolSetupLabware/__tests__/ProtocolSetupLabware.test.tsx
@@ -25,7 +25,17 @@ import {
   mockUseModulesQueryUnknown,
 } from '../__fixtures__'
 
-vi.mock('@opentrons/react-api-client')
+import type * as ReactApiClient from '@opentrons/react-api-client'
+
+vi.mock('@opentrons/react-api-client', async importOriginal => {
+  const actual = await importOriginal<typeof ReactApiClient>()
+  return {
+    ...actual,
+    useCreateLiveCommandMutation: vi.fn(),
+    useModulesQuery: vi.fn(),
+  }
+})
+
 vi.mock(
   '../../../organisms/LabwarePositionCheck/useMostRecentCompletedAnalysis'
 )