fix test errors

Opentrons · May 28, 2024 · dc7d652 · dc7d652
1 parent c6acd68
commit dc7d652
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/app/src/organisms/QuickTransferFlow/__tests__/VolumeEntry.test.tsx b/app/src/organisms/QuickTransferFlow/__tests__/VolumeEntry.test.tsx
@@ -9,10 +9,17 @@ import { NumericalKeyboard } from '../../../atoms/SoftwareKeyboard'
 import { getVolumeRange } from '../utils'
 import { VolumeEntry } from '../VolumeEntry'
 
-vi.mock('../../../atoms/InputField')
 vi.mock('../../../atoms/SoftwareKeyboard')
 vi.mock('../utils')
 
+vi.mock('../../../atoms/InputField', async importOriginal => {
+  const actualComponents = await importOriginal<typeof InputField>()
+  return {
+    ...actualComponents,
+    InputField: vi.fn(),
+  }
+})
+
 const render = (props: React.ComponentProps<typeof VolumeEntry>) => {
   return renderWithProviders(<VolumeEntry {...props} />, {
     i18nInstance: i18n,