From d2215f8d99e45c8426569b21e1c941da878a0223 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=EA=B9=80=EC=84=9D=EC=A7=84?= <sj01048260279@gmail.com>
Date: Wed, 7 May 2025 22:34:02 +0900
Subject: [PATCH] test(react-query): improve branch coverage for
 useSuspenseInfiniteQuery skipToken error handling

---
 .../useSuspenseInfiniteQuery.test.tsx         | 68 +++++++++++++++++++
 1 file changed, 68 insertions(+)

diff --git a/packages/react-query/src/__tests__/useSuspenseInfiniteQuery.test.tsx b/packages/react-query/src/__tests__/useSuspenseInfiniteQuery.test.tsx
index b5d7b4a49df..ac33557836c 100644
--- a/packages/react-query/src/__tests__/useSuspenseInfiniteQuery.test.tsx
+++ b/packages/react-query/src/__tests__/useSuspenseInfiniteQuery.test.tsx
@@ -40,4 +40,72 @@ describe('useSuspenseInfiniteQuery', () => {
     )
     consoleErrorSpy.mockRestore()
   })
+
+  it('should log an error when skipToken is used in development environment', () => {
+    const envCopy = process.env.NODE_ENV
+    process.env.NODE_ENV = 'development'
+
+    const consoleErrorSpy = vi
+      .spyOn(console, 'error')
+      .mockImplementation(() => undefined)
+    const key = queryKey()
+
+    function Page() {
+      useSuspenseInfiniteQuery({
+        queryKey: key,
+        queryFn: skipToken as any,
+        initialPageParam: 1,
+        getNextPageParam: () => 1,
+      })
+
+      return null
+    }
+
+    renderWithClient(
+      queryClient,
+      <React.Suspense fallback="Loading...">
+        <Page />
+      </React.Suspense>,
+    )
+
+    expect(consoleErrorSpy).toHaveBeenCalledWith(
+      'skipToken is not allowed for useSuspenseInfiniteQuery',
+    )
+
+    consoleErrorSpy.mockRestore()
+    process.env.NODE_ENV = envCopy
+  })
+
+  it('should not log an error when skipToken is used in production environment', () => {
+    const envCopy = process.env.NODE_ENV
+    process.env.NODE_ENV = 'production'
+
+    const consoleErrorSpy = vi
+      .spyOn(console, 'error')
+      .mockImplementation(() => undefined)
+    const key = queryKey()
+
+    function Page() {
+      useSuspenseInfiniteQuery({
+        queryKey: key,
+        queryFn: skipToken as any,
+        initialPageParam: 1,
+        getNextPageParam: () => 1,
+      })
+
+      return null
+    }
+
+    renderWithClient(
+      queryClient,
+      <React.Suspense fallback="Loading...">
+        <Page />
+      </React.Suspense>,
+    )
+
+    expect(consoleErrorSpy).not.toHaveBeenCalled()
+
+    consoleErrorSpy.mockRestore()
+    process.env.NODE_ENV = envCopy
+  })
 })