How to test queryClient.removeQueries() in the OnSuccess call back using useMutation and jest?

[BoyanLiuu]

hey friends,
I am trying to use jest test the useMutation call back hook and having trouble to find a way to test the queryClient.removeQueries, My main goal is to see if we ever called queryClient.removeQueries() function or not.
I have been searching on internet, it seems like all the tutorial/blog did not test queryClient.removeQueries(), Thanks for any kind of ideas or help!.

Here is the code I want to test:

export const useUpdateResources = (): UseMutationResult<
  ResourceResponse,
  Error,
  ResourceInput
> => {
  const queryClient = useQueryClient();
  return useMutation((request) => updateResource(request), {
    onSuccess: (_, request) => {
      queryClient.removeQueries(["update-resource"]);
    }
  });
};

Here is test I got so far:

  describe('useUpdateResources', () => {
    const queryClient = renderHook(() => useQueryClient());

    const updateResourcesSpy = jest.spyOn(apiClient, 'updateResource');
   // this would throw error 
    const queryClientSpy = jest.spyOn(queryClient, '');
 

    it('should call apiClient.updateResource endpoint with expected params', async () => {
      const params = {
        id: 'abc123',
        resources: { key1: 'val1', key2: 'val2', key3: 'val3', key4: 'val4' }
      };

      expect(updateResourcesSpy ).not.toHaveBeenCalledWith(params);

      const { result } = renderHook(() => useUpdateResources(), { wrapper: QueryClientProvider });

      result.current.mutate(params);

      //  wait until the query has transitioned to success state
      await waitFor(() => result.current.isSuccess);

      expect(queryClientSpy).toHaveBeenCalled();
      expect( updateResourcesSpy).toHaveBeenCalledWith(params);
    });
  });

[TkDodo]

don't mock the queryClient. check if the query still exists in the client. if remove was called, the query should no longer be there.