fix: refresh user query on continue from org selection (#2975)

codecov · Jul 2, 2024 · cf4e398 · cf4e398
1 parent bb4833d
commit cf4e398
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/services/defaultOrganization/useUpdateDefaultOrganization.spec.tsx b/src/services/defaultOrganization/useUpdateDefaultOrganization.spec.tsx
@@ -60,12 +60,17 @@ describe('useUpdateDefaultOrganization', () => {
           wrapper,
         })
         result.current.mutate({ username: 'codecov' })
+        const invalidateQueries = jest.spyOn(queryClient, 'invalidateQueries')
 
         await waitFor(() => expect(result.current.isSuccess).toBeTruthy())
 
         const username =
           result.current.data.data.updateDefaultOrganization.username
         await waitFor(() => expect(username).toBe('Gilmore'))
+
+        expect(invalidateQueries).toHaveBeenCalledTimes(2)
+        expect(invalidateQueries).toHaveBeenNthCalledWith(1, ['DetailOwner'])
+        expect(invalidateQueries).toHaveBeenNthCalledWith(2, ['currentUser'])
       })
     })
   })

diff --git a/src/services/defaultOrganization/useUpdateDefaultOrganization.ts b/src/services/defaultOrganization/useUpdateDefaultOrganization.ts
@@ -41,6 +41,7 @@ export function useUpdateDefaultOrganization() {
         )
       } else {
         queryClient.invalidateQueries(['DetailOwner'])
+        queryClient.invalidateQueries(['currentUser'])
       }
     },
     onError: (e: any) => {