test(query-core/queryObserver): add tests for 'refetchOnWindowFocus' as a function in 'shouldFetchOnWindowFocus'

Author: sukvvon

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

@@ -1337,6 +1337,53 @@ describe('queryObserver', () => {
     expect(observer.shouldFetchOnWindowFocus()).toBe(true)
   })
 
+  it('should return true from shouldFetchOnWindowFocus when refetchOnWindowFocus is a function returning true', () => {
+    const key = queryKey()
+    const refetchOnWindowFocus = vi.fn(() => true)
+
+    const observer = new QueryObserver(queryClient, {
+      queryKey: key,
+      queryFn: () => 'data',
+      refetchOnWindowFocus,
+    })
+
+    expect(observer.shouldFetchOnWindowFocus()).toBe(true)
+    expect(refetchOnWindowFocus).toHaveBeenCalledWith(
+      queryClient.getQueryCache().find({ queryKey: key }),
+    )
+  })
+
+  it('should return false from shouldFetchOnWindowFocus when refetchOnWindowFocus is a function returning false', () => {
+    const key = queryKey()
+
+    const observer = new QueryObserver(queryClient, {
+      queryKey: key,
+      queryFn: () => 'data',
+      refetchOnWindowFocus: () => false,
+    })
+
+    expect(observer.shouldFetchOnWindowFocus()).toBe(false)
+  })
+
+  it('should return true from shouldFetchOnWindowFocus when refetchOnWindowFocus is a function returning "always" even if the query is fresh', async () => {
+    const key = queryKey()
+
+    queryClient.prefetchQuery({
+      queryKey: key,
+      queryFn: () => sleep(10).then(() => 'data'),
+    })
+    await vi.advanceTimersByTimeAsync(10)
+
+    const observer = new QueryObserver(queryClient, {
+      queryKey: key,
+      queryFn: () => sleep(10).then(() => 'data'),
+      staleTime: Infinity,
+      refetchOnWindowFocus: () => 'always',
+    })
+
+    expect(observer.shouldFetchOnWindowFocus()).toBe(true)
+  })
+
   it('should fetch and return optimistic result via fetchOptimistic', async () => {
     const key = queryKey()
     const observer = new QueryObserver(queryClient, {