From dc7d652b63fe529167c2d7c11bc2226db3b4f926 Mon Sep 17 00:00:00 2001
From: koji <kk2796@nyu.edu>
Date: Tue, 28 May 2024 15:21:15 -0400
Subject: [PATCH] fix test errors

---
 .../QuickTransferFlow/__tests__/VolumeEntry.test.tsx     | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/app/src/organisms/QuickTransferFlow/__tests__/VolumeEntry.test.tsx b/app/src/organisms/QuickTransferFlow/__tests__/VolumeEntry.test.tsx
index a130ef6075f..9b686349445 100644
--- 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,