feat: fix unit tests

stakwork · Oct 16, 2024 · 7fa0b15 · 7fa0b15
1 parent 5463402
commit 7fa0b15
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/components/App/SideBar/FilterSearch/__tests__/index.tsx b/src/components/App/SideBar/FilterSearch/__tests__/index.tsx
@@ -32,9 +32,15 @@ const mockSetAnchorEl = jest.fn()
 const mockFetchData = jest.fn()
 const mockSetAbortRequests = jest.fn()
 const mockOnClose = jest.fn()
-const mockSchemaAll = [{ type: 'Type1' }, { type: 'Type2' }, { type: 'Type3' }]
+
+jest.mock('~/stores/useSchemaStore', () => ({
+  useSchemaStore: jest.fn(),
+}))
 
 describe('FilterSearch Component', () => {
+  const mockSetSchemas = jest.fn()
+  const mockSchemaAll = [{ type: 'Type1' }, { type: 'Type2' }, { type: 'Type3' }]
+
   beforeEach(() => {
     jest.clearAllMocks()
 
@@ -46,9 +52,10 @@ describe('FilterSearch Component', () => {
     })
 
     //
-    ;(useSchemaStore as jest.Mock)
-      .mockReturnValue([mockSchemaAll, jest.fn()])(useFeatureFlagStore as jest.Mock)
-      .mockReturnValue({ fastFiltersFeatureFlag: true })
+    ;(useSchemaStore as jest.Mock).mockReturnValue([mockSchemaAll, mockSetSchemas]) // Return an array
+
+    //
+    ;(useFeatureFlagStore as jest.Mock).mockReturnValue({ fastFiltersFeatureFlag: true })
 
     //
     ;(getSchemaAll as jest.Mock).mockResolvedValue({ schemas: mockSchemaAll })