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'
3
2
import * as React from 'react'
4
3
import { QueryCache , hashKey } from '@tanstack/query-core'
5
4
import {
@@ -10,6 +9,14 @@ import { useQuery } from '..'
10
9
import { createQueryClient , queryKey , renderWithClient , sleep } from './utils'
11
10
12
11
describe ( 'fine grained persister' , ( ) => {
12
+ beforeEach ( ( ) => {
13
+ vi . useFakeTimers ( )
14
+ } )
15
+
16
+ afterEach ( ( ) => {
17
+ vi . useRealTimers ( )
18
+ } )
19
+
13
20
const queryCache = new QueryCache ( )
14
21
const queryClient = createQueryClient ( { queryCache } )
15
22
@@ -61,7 +68,7 @@ describe('fine grained persister', () => {
61
68
62
69
const rendered = renderWithClient ( queryClient , < Test /> )
63
70
64
- await waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
71
+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from persister' ) )
65
72
expect ( spy ) . not . toHaveBeenCalled ( )
66
73
} )
67
74
@@ -116,8 +123,8 @@ describe('fine grained persister', () => {
116
123
117
124
const rendered = renderWithClient ( queryClient , < Test /> )
118
125
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' ) )
121
128
expect ( spy ) . toHaveBeenCalledTimes ( 1 )
122
129
} )
123
130
@@ -155,7 +162,7 @@ describe('fine grained persister', () => {
155
162
156
163
const rendered = renderWithClient ( queryClient , < Test /> )
157
164
158
- await waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
165
+ await vi . waitFor ( ( ) => rendered . getByText ( 'Works from queryFn' ) )
159
166
expect ( spy ) . toHaveBeenCalledTimes ( 1 )
160
167
161
168
const storedItem = await storage . getItem ( `${ PERSISTER_KEY_PREFIX } -${ hash } ` )
0 commit comments