1- import { describe , expect , it , vi } from 'vitest'
2- import { waitFor } from '@testing-library/react'
1+ import { afterEach , beforeEach , describe , expect , it , vi } from 'vitest'
32import * as React from 'react'
43import { QueryCache , hashKey } from '@tanstack/query-core'
54import {
@@ -10,6 +9,14 @@ import { useQuery } from '..'
109import { createQueryClient , queryKey , renderWithClient , sleep } from './utils'
1110
1211describe ( 'fine grained persister' , ( ) => {
12+ beforeEach ( ( ) => {
13+ vi . useFakeTimers ( )
14+ } )
15+
16+ afterEach ( ( ) => {
17+ vi . useRealTimers ( )
18+ } )
19+
1320 const queryCache = new QueryCache ( )
1421 const queryClient = createQueryClient ( { queryCache } )
1522
@@ -61,7 +68,7 @@ describe('fine grained persister', () => {
6168
6269 const rendered = renderWithClient ( queryClient , < Test /> )
6370
64- await waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
71+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
6572 expect ( spy ) . not . toHaveBeenCalled ( )
6673 } )
6774
@@ -116,8 +123,8 @@ describe('fine grained persister', () => {
116123
117124 const rendered = renderWithClient ( queryClient , < Test /> )
118125
119- await waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
120- await waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
126+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
127+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
121128 expect ( spy ) . toHaveBeenCalledTimes ( 1 )
122129 } )
123130
@@ -155,7 +162,7 @@ describe('fine grained persister', () => {
155162
156163 const rendered = renderWithClient ( queryClient , < Test /> )
157164
158- await waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
165+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
159166 expect ( spy ) . toHaveBeenCalledTimes ( 1 )
160167
161168 const storedItem = await storage . getItem ( `${ PERSISTER_KEY_PREFIX } -${ hash } ` )
0 commit comments