@@ -10,7 +10,7 @@ import {
1010 hydrate ,
1111 useQuery ,
1212} from '..'
13- import { createQueryClient , setIsServer , sleep } from './utils'
13+ import { createQueryClient , setIsServer } from './utils'
1414
1515const ReactHydrate = ( element : React . ReactElement , container : Element ) => {
1616 let root : any
@@ -23,7 +23,7 @@ const ReactHydrate = (element: React.ReactElement, container: Element) => {
2323}
2424
2525async function fetchData < TData > ( value : TData , ms ?: number ) : Promise < TData > {
26- await sleep ( ms || 1 )
26+ await vi . advanceTimersByTimeAsync ( ms || 1 )
2727 return value
2828}
2929
@@ -36,12 +36,15 @@ describe('Server side rendering with de/rehydration', () => {
3636 beforeAll ( ( ) => {
3737 // @ts -expect-error we expect IS_REACT_ACT_ENVIRONMENT to exist
3838 previousIsReactActEnvironment = globalThis . IS_REACT_ACT_ENVIRONMENT = true
39+ vi . useFakeTimers ( )
3940 } )
4041
4142 afterAll ( ( ) => {
4243 // @ts -expect-error we expect IS_REACT_ACT_ENVIRONMENT to exist
4344 globalThis . IS_REACT_ACT_ENVIRONMENT = previousIsReactActEnvironment
45+ vi . useRealTimers ( )
4446 } )
47+
4548 it ( 'should not mismatch on success' , async ( ) => {
4649 const consoleMock = vi . spyOn ( console , 'error' )
4750 consoleMock . mockImplementation ( ( ) => undefined )
@@ -185,7 +188,7 @@ describe('Server side rendering with de/rehydration', () => {
185188 expect ( consoleMock ) . toHaveBeenCalledTimes ( 0 )
186189 expect ( fetchDataError ) . toHaveBeenCalledTimes ( 2 )
187190 expect ( el . innerHTML ) . toBe ( expectedMarkup )
188- await sleep ( 50 )
191+ await vi . advanceTimersByTimeAsync ( 50 )
189192 expect ( fetchDataError ) . toHaveBeenCalledTimes ( 2 )
190193 expect ( el . innerHTML ) . toBe (
191194 'ErrorComponent - status:error fetching:false data:undefined' ,
@@ -253,7 +256,7 @@ describe('Server side rendering with de/rehydration', () => {
253256 expect ( consoleMock ) . toHaveBeenCalledTimes ( 0 )
254257 expect ( fetchDataSuccess ) . toHaveBeenCalledTimes ( 1 )
255258 expect ( el . innerHTML ) . toBe ( expectedMarkup )
256- await sleep ( 50 )
259+ await vi . advanceTimersByTimeAsync ( 50 )
257260 expect ( fetchDataSuccess ) . toHaveBeenCalledTimes ( 1 )
258261 expect ( el . innerHTML ) . toBe (
259262 'SuccessComponent - status:success fetching:false data:success!' ,
0 commit comments