1- import { describe , expect , test , vi } from 'vitest'
1+ import { afterAll , beforeAll , describe , expect , test , vi } from 'vitest'
22import { Query , QueryClient , hashKey } from '@tanstack/query-core'
33import {
44 PERSISTER_KEY_PREFIX ,
55 experimental_createPersister ,
66} from '../createPersister'
7- import { sleep } from './utils'
87import type { QueryFunctionContext , QueryKey } from '@tanstack/query-core'
98import type { StoragePersisterOptions } from '../createPersister'
109
@@ -60,6 +59,14 @@ function setupPersister(
6059}
6160
6261describe ( 'createPersister' , ( ) => {
62+ beforeAll ( ( ) => {
63+ vi . useFakeTimers ( )
64+ } )
65+
66+ afterAll ( ( ) => {
67+ vi . useRealTimers ( )
68+ } )
69+
6370 test ( 'should fetch if storage is not provided' , async ( ) => {
6471 const { context, persisterFn, query, queryFn } = setupPersister ( [ 'foo' ] , {
6572 storage : undefined ,
@@ -215,7 +222,7 @@ describe('createPersister', () => {
215222 query . fetch = vi . fn ( )
216223 expect ( query . state . dataUpdatedAt ) . toEqual ( 0 )
217224
218- await sleep ( 0 )
225+ await vi . advanceTimersByTimeAsync ( 0 )
219226
220227 expect ( queryFn ) . toHaveBeenCalledTimes ( 0 )
221228 expect ( query . fetch ) . toHaveBeenCalledTimes ( 0 )
@@ -243,7 +250,7 @@ describe('createPersister', () => {
243250 query . state . isInvalidated = true
244251 query . fetch = vi . fn ( )
245252
246- await sleep ( 0 )
253+ await vi . advanceTimersByTimeAsync ( 0 )
247254
248255 expect ( queryFn ) . toHaveBeenCalledTimes ( 0 )
249256 expect ( query . fetch ) . toHaveBeenCalledTimes ( 1 )
@@ -266,7 +273,7 @@ describe('createPersister', () => {
266273 await persisterFn ( queryFn , context , query )
267274 query . setData ( 'baz' )
268275
269- await sleep ( 0 )
276+ await vi . advanceTimersByTimeAsync ( 0 )
270277
271278 expect ( queryFn ) . toHaveBeenCalledOnce ( )
272279 expect ( queryFn ) . toHaveBeenCalledWith ( context )
@@ -308,7 +315,7 @@ describe('createPersister', () => {
308315 await persisterFn ( queryFn , context , query )
309316 query . fetch = vi . fn ( )
310317
311- await sleep ( 0 )
318+ await vi . advanceTimersByTimeAsync ( 0 )
312319
313320 expect ( queryFn ) . toHaveBeenCalledTimes ( 1 )
314321 expect ( query . fetch ) . toHaveBeenCalledTimes ( 0 )
@@ -337,7 +344,7 @@ describe('createPersister', () => {
337344 query . state . isInvalidated = true
338345 query . fetch = vi . fn ( )
339346
340- await sleep ( 0 )
347+ await vi . advanceTimersByTimeAsync ( 0 )
341348
342349 expect ( queryFn ) . toHaveBeenCalledTimes ( 0 )
343350 expect ( query . fetch ) . toHaveBeenCalledTimes ( 1 )
@@ -362,7 +369,7 @@ describe('createPersister', () => {
362369 await persisterFn ( queryFn , context , query )
363370 query . setData ( 'baz' )
364371
365- await sleep ( 0 )
372+ await vi . advanceTimersByTimeAsync ( 0 )
366373
367374 expect ( queryFn ) . toHaveBeenCalledOnce ( )
368375 expect ( queryFn ) . toHaveBeenCalledWith ( context )
0 commit comments