test: check invalidateQueries

open-craft · Dec 8, 2023 · 8d2c437 · 8d2c437
1 parent 94f1bf7
commit 8d2c437
Showing 1 changed file with 11 additions and 16 deletions.
diff --git a/src/taxonomy/import-tags/data/api.test.jsx b/src/taxonomy/import-tags/data/api.test.jsx
@@ -1,5 +1,6 @@
 import { initializeMockApp } from '@edx/frontend-platform';
 import { getAuthenticatedHttpClient } from '@edx/frontend-platform/auth';
+import { waitFor } from '@testing-library/react';
 import { renderHook } from '@testing-library/react-hooks';
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
 
@@ -26,16 +27,6 @@ const queryClient = new QueryClient({
   },
 });
 
-const mockInvalidateQueries = jest.fn();
-
-jest.mock('@tanstack/react-query', () => ({
-  ...jest.requireActual('@tanstack/react-query'),
-  useQueryClient: () => ({
-    ...jest.requireActual('@tanstack/react-query').useQueryClient(),
-    invalidateQueries: mockInvalidateQueries,
-  }),
-}));
-
 const wrapper = ({ children }) => (
   <QueryClientProvider client={queryClient}>
     {children}
@@ -69,16 +60,12 @@ describe('import taxonomy api calls', () => {
 
   it('should call import tags', async () => {
     axiosMock.onPut(getTagsImportApiUrl(1)).reply(200);
+    const mockInvalidateQueries = jest.spyOn(queryClient, 'invalidateQueries');
+
     const { result } = renderHook(() => useImportTags(), { wrapper });
 
     await result.current.mutateAsync({ taxonomyId: 1 });
     expect(axiosMock.history.put[0].url).toEqual(getTagsImportApiUrl(1));
-  });
-
-  it('should call plan import tags', async () => {
-    axiosMock.onPut(getTagsPlanImportApiUrl(1)).reply(200, { plan: 'plan' });
-    await planImportTags(1);
-    expect(axiosMock.history.put[0].url).toEqual(getTagsPlanImportApiUrl(1));
     expect(mockInvalidateQueries).toHaveBeenCalledWith({
       queryKey: ['tagList', 1],
     });
@@ -87,8 +74,16 @@ describe('import taxonomy api calls', () => {
     });
   });
 
+  it('should call plan import tags', async () => {
+    axiosMock.onPut(getTagsPlanImportApiUrl(1)).reply(200, { plan: 'plan' });
+    await planImportTags(1);
+    expect(axiosMock.history.put[0].url).toEqual(getTagsPlanImportApiUrl(1));
+  });
+
   it('should handle errors in plan import tags', async () => {
     axiosMock.onPut(getTagsPlanImportApiUrl(1)).reply(400, { error: 'test error' });
+    const mockInvalidateQueries = jest.spyOn(queryClient, 'invalidateQueries');
+
     expect(planImportTags(1)).rejects.toEqual(Error('test error'));
     expect(axiosMock.history.put[0].url).toEqual(getTagsPlanImportApiUrl(1));
     expect(mockInvalidateQueries).not.toHaveBeenCalled();