test(react-query): add test case for useSuspenseQuery accept skipToken as queryFn (#8269)

manudeli · web-flow · commit e802711e7bc2 · 2024-11-13T08:06:15.000+01:00
diff --git a/packages/react-query/src/__tests__/useSuspenseQuery.test.tsx b/packages/react-query/src/__tests__/useSuspenseQuery.test.tsx
@@ -5,6 +5,7 @@ import { ErrorBoundary } from 'react-error-boundary'
 import {
   QueryCache,
   QueryErrorResetBoundary,
+  skipToken,
   useQueryErrorResetBoundary,
   useSuspenseInfiniteQuery,
   useSuspenseQuery,
@@ -870,4 +871,36 @@ describe('useSuspenseQuery', () => {
     await waitFor(() => rendered.getByText('loading'))
     await waitFor(() => rendered.getByText('data: 2'))
   })
+
+  it('should log an error when skipToken is passed as queryFn', () => {
+    const consoleErrorSpy = vi
+      .spyOn(console, 'error')
+      .mockImplementation(() => {})
+    const key = queryKey()
+
+    function Page() {
+      useSuspenseQuery({
+        queryKey: key,
+        // @ts-expect-error
+        queryFn: Math.random() >= 0 ? skipToken : () => Promise.resolve(5),
+      })
+
+      return null
+    }
+
+    function App() {
+      return (
+        <React.Suspense fallback="Loading...">
+          <Page />
+        </React.Suspense>
+      )
+    }
+
+    renderWithClient(queryClient, <App />)
+
+    expect(consoleErrorSpy).toHaveBeenCalledWith(
+      'skipToken is not allowed for useSuspenseQuery',
+    )
+    consoleErrorSpy.mockRestore()
+  })
 })
