test(react-query/fine-grained-persister): remove 'waitFor' and add 'advanceTimersByTime'

Author: sukvvon

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

@@ -1,4 +1,4 @@
-import { act, render, waitFor } from '@testing-library/react'
+import { act, render } from '@testing-library/react'
 import { Suspense } from 'react'
 import {
   afterAll,
@@ -44,7 +44,7 @@ describe('Suspense Timer Tests', () => {
   let fetchCount: { count: number }
 
   beforeAll(() => {
-    vi.useFakeTimers({ shouldAdvanceTime: true })
+    vi.useFakeTimers()
   })
 
   afterAll(() => {
@@ -71,7 +71,8 @@ describe('Suspense Timer Tests', () => {
 
     const rendered = renderWithSuspense(queryClient, <TestComponent />)
 
-    await waitFor(() => rendered.getByText('data: data'))
+    await act(() => vi.advanceTimersByTime(0))
+    rendered.getByText('data: data')
 
     rendered.rerender(
       <QueryClientProvider client={queryClient}>
@@ -81,9 +82,7 @@ describe('Suspense Timer Tests', () => {
       </QueryClientProvider>,
     )
 
-    act(() => {
-      vi.advanceTimersByTime(100)
-    })
+    await act(() => vi.advanceTimersByTime(100))
 
     expect(fetchCount.count).toBe(1)
   })
@@ -97,7 +96,8 @@ describe('Suspense Timer Tests', () => {
 
     const rendered = renderWithSuspense(queryClient, <TestComponent />)
 
-    await waitFor(() => rendered.getByText('data: data'))
+    await act(() => vi.advanceTimersByTime(0))
+    rendered.getByText('data: data')
 
     rendered.rerender(
       <QueryClientProvider client={queryClient}>
@@ -107,9 +107,7 @@ describe('Suspense Timer Tests', () => {
       </QueryClientProvider>,
     )
 
-    act(() => {
-      vi.advanceTimersByTime(100)
-    })
+    await act(() => vi.advanceTimersByTime(100))
 
     expect(fetchCount.count).toBe(1)
   })
@@ -123,7 +121,8 @@ describe('Suspense Timer Tests', () => {
 
     const rendered = renderWithSuspense(queryClient, <TestComponent />)
 
-    await waitFor(() => rendered.getByText('data: data'))
+    await act(() => vi.advanceTimersByTime(0))
+    rendered.getByText('data: data')
 
     rendered.rerender(
       <QueryClientProvider client={queryClient}>
@@ -133,9 +132,7 @@ describe('Suspense Timer Tests', () => {
       </QueryClientProvider>,
     )
 
-    act(() => {
-      vi.advanceTimersByTime(1500)
-    })
+    await act(() => vi.advanceTimersByTime(1500))
 
     expect(fetchCount.count).toBe(1)
   })
@@ -149,7 +146,8 @@ describe('Suspense Timer Tests', () => {
 
     const rendered = renderWithSuspense(queryClient, <TestComponent />)
 
-    await waitFor(() => rendered.getByText('data: data'))
+    await act(() => vi.advanceTimersByTime(0))
+    rendered.getByText('data: data')
 
     rendered.rerender(
       <QueryClientProvider client={queryClient}>
@@ -159,9 +157,7 @@ describe('Suspense Timer Tests', () => {
       </QueryClientProvider>,
     )
 
-    act(() => {
-      vi.advanceTimersByTime(500)
-    })
+    await act(() => vi.advanceTimersByTime(500))
 
     expect(fetchCount.count).toBe(1)
   })
@@ -175,7 +171,8 @@ describe('Suspense Timer Tests', () => {
 
     const rendered = renderWithSuspense(queryClient, <TestComponent />)
 
-    await waitFor(() => rendered.getByText('data: data'))
+    await act(() => vi.advanceTimersByTime(0))
+    rendered.getByText('data: data')
 
     rendered.rerender(
       <QueryClientProvider client={queryClient}>
@@ -185,9 +182,7 @@ describe('Suspense Timer Tests', () => {
       </QueryClientProvider>,
     )
 
-    act(() => {
-      vi.advanceTimersByTime(2000)
-    })
+    await act(() => vi.advanceTimersByTime(2000))
 
     expect(fetchCount.count).toBe(1)
   })