fix learningpath invalidation (#635)

* fix learningpath invalidation * invalidate on removal, too
mitodl · Mar 26, 2024 · fb5de02 · fb5de02
1 parent eef4c7f
commit fb5de02
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 17 deletions.
diff --git a/frontends/api/src/hooks/learningResources/index.test.ts b/frontends/api/src/hooks/learningResources/index.test.ts
@@ -273,21 +273,22 @@ describe("LearningPath CRUD", () => {
     setMockResponse.post(url, relationship)
 
     const { wrapper, queryClient } = setupReactQueryTest()
-    jest.spyOn(queryClient, "invalidateQueries")
     const { result } = renderHook(useLearningpathRelationshipCreate, {
       wrapper,
     })
     result.current.mutate(requestData)
 
     await waitFor(() => expect(result.current.isSuccess).toBe(true))
     expect(makeRequest).toHaveBeenCalledWith("post", url, requestData)
-    expect(queryClient.invalidateQueries).toHaveBeenCalledWith(
-      keys.relationshipListing,
-    )
+
     expect(invalidateResourceQueries).toHaveBeenCalledWith(
       queryClient,
       relationship.child,
     )
+    expect(invalidateResourceQueries).toHaveBeenCalledWith(
+      queryClient,
+      relationship.parent,
+    )
 
     // Check patches cached result
     expect(queryClient.getQueryData(keys.childResource)).toEqual(
@@ -302,7 +303,6 @@ describe("LearningPath CRUD", () => {
     setMockResponse.delete(url, null)
     const { wrapper, queryClient } = setupReactQueryTest()
     queryClient.setQueryData(keys.childResource, relationship.resource)
-    jest.spyOn(queryClient, "invalidateQueries")
     const { result } = renderHook(useLearningpathRelationshipDestroy, {
       wrapper,
     })
@@ -311,13 +311,14 @@ describe("LearningPath CRUD", () => {
     await waitFor(() => expect(result.current.isSuccess).toBe(true))
 
     expect(makeRequest).toHaveBeenCalledWith("delete", url, undefined)
-    expect(queryClient.invalidateQueries).toHaveBeenCalledWith(
-      keys.relationshipListing,
-    )
     expect(invalidateResourceQueries).toHaveBeenCalledWith(
       queryClient,
       relationship.child,
     )
+    expect(invalidateResourceQueries).toHaveBeenCalledWith(
+      queryClient,
+      relationship.parent,
+    )
 
     // Patched existing resource
     expect(queryClient.getQueryData(keys.childResource)).toEqual({

diff --git a/frontends/api/src/hooks/learningResources/index.ts b/frontends/api/src/hooks/learningResources/index.ts
@@ -158,10 +158,7 @@ const useLearningpathRelationshipCreate = () => {
     },
     onSettled: (response, _err, vars) => {
       invalidateResourceQueries(queryClient, vars.child)
-      queryClient.invalidateQueries(
-        learningResources.learningpaths._ctx.detail(vars.parent)._ctx
-          .infiniteItems._def,
-      )
+      invalidateResourceQueries(queryClient, vars.parent)
     },
   })
 }
@@ -188,12 +185,9 @@ const useLearningpathRelationshipDestroy = () => {
         },
       )
     },
-    onSettled: (response, _err, vars) => {
+    onSettled: (_response, _err, vars) => {
       invalidateResourceQueries(queryClient, vars.child)
-      queryClient.invalidateQueries(
-        learningResources.learningpaths._ctx.detail(vars.parent)._ctx
-          .infiniteItems._def,
-      )
+      invalidateResourceQueries(queryClient, vars.parent)
     },
   })
 }