test(preact-query/suspense): add test cases for 'static' staleTime with number and function (#10134)

Author: sukvvon

packages/preact-query/src/__tests__/suspense.test.tsx

@@ -5,6 +5,7 @@ import { Suspense } from 'preact/compat'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
 import { QueryClient, QueryClientProvider, useSuspenseQuery } from '..'
+import type { StaleTime } from '@tanstack/query-core'
 import type { QueryKey } from '..'
 
 function renderWithSuspense(client: QueryClient, ui: ComponentChildren) {
@@ -18,7 +19,7 @@ function renderWithSuspense(client: QueryClient, ui: ComponentChildren) {
 function createTestQuery(options: {
   fetchCount: { count: number }
   queryKey: QueryKey
-  staleTime?: number | (() => number)
+  staleTime?: StaleTime | (() => StaleTime)
 }) {
   return function TestComponent() {
     const { data } = useSuspenseQuery({
@@ -158,6 +159,58 @@ describe('Suspense Timer Tests', () => {
     expect(fetchCount.count).toBe(1)
   })
 
+  it('should preserve staleTime when value is static', async () => {
+    const TestComponent = createTestQuery({
+      fetchCount,
+      queryKey: queryKey(),
+      staleTime: 'static',
+    })
+
+    const rendered = renderWithSuspense(queryClient, <TestComponent />)
+
+    expect(rendered.getByText('loading')).toBeInTheDocument()
+    await vi.advanceTimersByTimeAsync(10)
+    expect(rendered.getByText('data: data')).toBeInTheDocument()
+
+    rendered.rerender(
+      <QueryClientProvider client={queryClient}>
+        <Suspense fallback="loading">
+          <TestComponent />
+        </Suspense>
+      </QueryClientProvider>,
+    )
+
+    await vi.advanceTimersByTimeAsync(2000)
+
+    expect(fetchCount.count).toBe(1)
+  })
+
+  it('should preserve staleTime when function returns static', async () => {
+    const TestComponent = createTestQuery({
+      fetchCount,
+      queryKey: queryKey(),
+      staleTime: () => 'static',
+    })
+
+    const rendered = renderWithSuspense(queryClient, <TestComponent />)
+
+    expect(rendered.getByText('loading')).toBeInTheDocument()
+    await vi.advanceTimersByTimeAsync(10)
+    expect(rendered.getByText('data: data')).toBeInTheDocument()
+
+    rendered.rerender(
+      <QueryClientProvider client={queryClient}>
+        <Suspense fallback="loading">
+          <TestComponent />
+        </Suspense>
+      </QueryClientProvider>,
+    )
+
+    await vi.advanceTimersByTimeAsync(2000)
+
+    expect(fetchCount.count).toBe(1)
+  })
+
   it('should respect staleTime when function returns value greater than 1000ms', async () => {
     const TestComponent = createTestQuery({
       fetchCount,