diff --git a/packages/react-query/src/__tests__/fine-grained-persister.test.tsx b/packages/react-query/src/__tests__/fine-grained-persister.test.tsx
index 7b78e8eea0..800e708559 100644
--- a/packages/react-query/src/__tests__/fine-grained-persister.test.tsx
+++ b/packages/react-query/src/__tests__/fine-grained-persister.test.tsx
@@ -1,5 +1,4 @@
-import { describe, expect, it, vi } from 'vitest'
-import { waitFor } from '@testing-library/react'
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
import * as React from 'react'
import { QueryCache, hashKey } from '@tanstack/query-core'
import {
@@ -10,6 +9,14 @@ import { useQuery } from '..'
import { createQueryClient, queryKey, renderWithClient, sleep } from './utils'
describe('fine grained persister', () => {
+ beforeEach(() => {
+ vi.useFakeTimers()
+ })
+
+ afterEach(() => {
+ vi.useRealTimers()
+ })
+
const queryCache = new QueryCache()
const queryClient = createQueryClient({ queryCache })
@@ -61,7 +68,7 @@ describe('fine grained persister', () => {
const rendered = renderWithClient(queryClient, )
- await waitFor(() => rendered.getByText('Works from persister'))
+ await vi.waitFor(() => rendered.getByText('Works from persister'))
expect(spy).not.toHaveBeenCalled()
})
@@ -116,8 +123,8 @@ describe('fine grained persister', () => {
const rendered = renderWithClient(queryClient, )
- await waitFor(() => rendered.getByText('Works from persister'))
- await waitFor(() => rendered.getByText('Works from queryFn'))
+ await vi.waitFor(() => rendered.getByText('Works from persister'))
+ await vi.waitFor(() => rendered.getByText('Works from queryFn'))
expect(spy).toHaveBeenCalledTimes(1)
})
@@ -155,7 +162,7 @@ describe('fine grained persister', () => {
const rendered = renderWithClient(queryClient, )
- await waitFor(() => rendered.getByText('Works from queryFn'))
+ await vi.waitFor(() => rendered.getByText('Works from queryFn'))
expect(spy).toHaveBeenCalledTimes(1)
const storedItem = await storage.getItem(`${PERSISTER_KEY_PREFIX}-${hash}`)