@@ -10,7 +10,7 @@ import {
10
10
hydrate ,
11
11
useQuery ,
12
12
} from '..'
13
- import { createQueryClient , setIsServer , sleep } from './utils'
13
+ import { createQueryClient , setIsServer } from './utils'
14
14
15
15
const ReactHydrate = ( element : React . ReactElement , container : Element ) => {
16
16
let root : any
@@ -23,7 +23,7 @@ const ReactHydrate = (element: React.ReactElement, container: Element) => {
23
23
}
24
24
25
25
async function fetchData < TData > ( value : TData , ms ?: number ) : Promise < TData > {
26
- await sleep ( ms || 1 )
26
+ await vi . advanceTimersByTimeAsync ( ms || 1 )
27
27
return value
28
28
}
29
29
@@ -36,12 +36,15 @@ describe('Server side rendering with de/rehydration', () => {
36
36
beforeAll ( ( ) => {
37
37
// @ts -expect-error we expect IS_REACT_ACT_ENVIRONMENT to exist
38
38
previousIsReactActEnvironment = globalThis . IS_REACT_ACT_ENVIRONMENT = true
39
+ vi . useFakeTimers ( )
39
40
} )
40
41
41
42
afterAll ( ( ) => {
42
43
// @ts -expect-error we expect IS_REACT_ACT_ENVIRONMENT to exist
43
44
globalThis . IS_REACT_ACT_ENVIRONMENT = previousIsReactActEnvironment
45
+ vi . useRealTimers ( )
44
46
} )
47
+
45
48
it ( 'should not mismatch on success' , async ( ) => {
46
49
const consoleMock = vi . spyOn ( console , 'error' )
47
50
consoleMock . mockImplementation ( ( ) => undefined )
@@ -185,7 +188,7 @@ describe('Server side rendering with de/rehydration', () => {
185
188
expect ( consoleMock ) . toHaveBeenCalledTimes ( 0 )
186
189
expect ( fetchDataError ) . toHaveBeenCalledTimes ( 2 )
187
190
expect ( el . innerHTML ) . toBe ( expectedMarkup )
188
- await sleep ( 50 )
191
+ await vi . advanceTimersByTimeAsync ( 50 )
189
192
expect ( fetchDataError ) . toHaveBeenCalledTimes ( 2 )
190
193
expect ( el . innerHTML ) . toBe (
191
194
'ErrorComponent - status:error fetching:false data:undefined' ,
@@ -253,7 +256,7 @@ describe('Server side rendering with de/rehydration', () => {
253
256
expect ( consoleMock ) . toHaveBeenCalledTimes ( 0 )
254
257
expect ( fetchDataSuccess ) . toHaveBeenCalledTimes ( 1 )
255
258
expect ( el . innerHTML ) . toBe ( expectedMarkup )
256
- await sleep ( 50 )
259
+ await vi . advanceTimersByTimeAsync ( 50 )
257
260
expect ( fetchDataSuccess ) . toHaveBeenCalledTimes ( 1 )
258
261
expect ( el . innerHTML ) . toBe (
259
262
'SuccessComponent - status:success fetching:false data:success!' ,
0 commit comments