test(react-query/usePrefetchQuery): remove 'vi.waitFor' and add 'advanceTimersByTimeAsync'

Author: sukvvon

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

@@ -1,6 +1,6 @@
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 import React from 'react'
-import { fireEvent } from '@testing-library/react'
+import { act, fireEvent } from '@testing-library/react'
 import { ErrorBoundary } from 'react-error-boundary'
 import { queryKey } from '@tanstack/query-test-utils'
 import {
@@ -72,7 +72,8 @@ describe('usePrefetchQuery', () => {
 
     const rendered = renderWithClient(queryClient, <App />)
 
-    await vi.waitFor(() => rendered.getByText('data: prefetchQuery'))
+    await act(() => vi.advanceTimersByTimeAsync(11))
+    rendered.getByText('data: prefetchQuery')
     expect(queryOpts.queryFn).toHaveBeenCalledTimes(1)
   })
 
@@ -97,9 +98,7 @@ describe('usePrefetchQuery', () => {
     const rendered = renderWithClient(queryClient, <App />)
 
     expect(rendered.queryByText('fetching: true')).not.toBeInTheDocument()
-    await vi.waitFor(() =>
-      rendered.getByText('data: The usePrefetchQuery hook is smart!'),
-    )
+    rendered.getByText('data: The usePrefetchQuery hook is smart!')
     expect(queryOpts.queryFn).not.toHaveBeenCalled()
   })
 
@@ -135,7 +134,7 @@ describe('usePrefetchQuery', () => {
     queryFn.mockClear()
     const rendered = renderWithClient(queryClient, <App />)
 
-    await vi.waitFor(() => rendered.getByText('Oops!'))
+    rendered.getByText('Oops!')
     expect(rendered.queryByText('data: Not an error')).not.toBeInTheDocument()
     expect(queryOpts.queryFn).not.toHaveBeenCalled()
 
@@ -166,7 +165,8 @@ describe('usePrefetchQuery', () => {
     }
 
     const rendered = renderWithClient(queryClient, <App />)
-    await vi.waitFor(() => rendered.getByText('data: prefetchedQuery'))
+    await act(() => vi.advanceTimersByTimeAsync(11))
+    rendered.getByText('data: prefetchedQuery')
     expect(queryOpts.queryFn).toHaveBeenCalledTimes(1)
   })
 
@@ -212,11 +212,10 @@ describe('usePrefetchQuery', () => {
 
     const rendered = renderWithClient(queryClient, <App />)
 
-    await vi.waitFor(() => rendered.getByText('Oops!'))
+    rendered.getByText('Oops!')
     fireEvent.click(rendered.getByText('Try again'))
-    await vi.waitFor(() =>
-      rendered.getByText('data: This is fine :dog: :fire:'),
-    )
+    await act(() => vi.advanceTimersByTimeAsync(11))
+    rendered.getByText('data: This is fine :dog: :fire:')
     expect(queryOpts.queryFn).toHaveBeenCalledTimes(1)
     consoleMock.mockRestore()
   })
@@ -265,14 +264,11 @@ describe('usePrefetchQuery', () => {
     expect(
       queryClient.getQueryState(thirdQueryOpts.queryKey)?.fetchStatus,
     ).toBe('fetching')
-    await vi.waitFor(() => rendered.getByText('Loading...'))
-    await vi.waitFor(() => rendered.getByText('data: Prefetch is nice!'))
-    await vi.waitFor(() =>
-      rendered.getByText('data: Prefetch is really nice!!'),
-    )
-    await vi.waitFor(() =>
-      rendered.getByText('data: Prefetch does not create waterfalls!!'),
-    )
+    rendered.getByText('Loading...')
+    await act(() => vi.advanceTimersByTimeAsync(11))
+    rendered.getByText('data: Prefetch is nice!')
+    rendered.getByText('data: Prefetch is really nice!!')
+    rendered.getByText('data: Prefetch does not create waterfalls!!')
     expect(Fallback).toHaveBeenCalledTimes(1)
     expect(firstQueryOpts.queryFn).toHaveBeenCalledTimes(1)
     expect(secondQueryOpts.queryFn).toHaveBeenCalledTimes(1)