fix(query-core): propagate AbortSignal reason in infiniteQueryBehavior fetchPage cancellation (TanStack#10476)

* fix: propagate AbortSignal reason when cancelling infinite query fetchPage (Closes TanStack#10475)

Signed-off-by: Cocoon-Break <54054995+kuishou68@users.noreply.github.com>

* add a test

---------

Signed-off-by: Cocoon-Break <54054995+kuishou68@users.noreply.github.com>
Co-authored-by: Dominik Dorfmeister 🔮 <office@dorfmeister.cc>

Author: kuishou68

packages/query-core/src/__tests__/infiniteQueryBehavior.test.tsx

@@ -1,6 +1,7 @@
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 import { queryKey, sleep } from '@tanstack/query-test-utils'
 import { CancelledError, InfiniteQueryObserver, QueryClient } from '..'
+import { infiniteQueryBehavior } from '../infiniteQueryBehavior'
 import type { InfiniteData, InfiniteQueryObserverResult, QueryCache } from '..'
 
 describe('InfiniteQueryBehavior', () => {
@@ -332,6 +333,46 @@ describe('InfiniteQueryBehavior', () => {
     unsubscribe()
   })
 
+  it('should surface the abort reason when cancellation happens between refetched pages', async () => {
+    const key = queryKey()
+    const abortController = new AbortController()
+    const queryFn = vi.fn().mockImplementation(({ pageParam, signal }) => {
+      void signal.aborted
+
+      if (pageParam === 1) {
+        abortController.abort()
+      }
+
+      return pageParam
+    })
+
+    const behavior = infiniteQueryBehavior<number, Error, number, number>()
+    const context = {
+      client: queryClient,
+      queryKey: key,
+      fetchOptions: undefined,
+      options: {
+        queryKey: key,
+        queryFn,
+        initialPageParam: 1,
+        getNextPageParam: (lastPage: number) => lastPage + 1,
+      },
+      state: {
+        data: {
+          pages: [1, 2],
+          pageParams: [1, 2],
+        },
+      },
+      fetchFn: () => Promise.resolve(),
+      signal: abortController.signal,
+    }
+
+    behavior.onFetch(context as any, {} as any)
+
+    await expect(context.fetchFn()).rejects.toBe(abortController.signal.reason)
+    expect(queryFn).toHaveBeenCalledTimes(1)
+  })
+
   it('should not enter an infinite loop when a page errors while retry is on #8046', async () => {
     let errorCount = 0
     const key = queryKey()

packages/query-core/src/infiniteQueryBehavior.ts

@@ -44,7 +44,7 @@ export function infiniteQueryBehavior<TQueryFnData, TError, TData, TPageParam>(
           previous?: boolean,
         ): Promise<InfiniteData<unknown>> => {
           if (cancelled) {
-            return Promise.reject()
+            return Promise.reject(context.signal.reason)
           }
 
           if (param == null && data.pages.length) {