Skip to content

Commit 0849660

Browse files
authored
test(react-query): use fake timers for ssr-hydration.test.tsx (#9058)
1 parent c987abc commit 0849660

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

packages/react-query/src/__tests__/ssr-hydration.test.tsx

+7-4
Original file line numberDiff line numberDiff line change
@@ -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

1515
const ReactHydrate = (element: React.ReactElement, container: Element) => {
1616
let root: any
@@ -23,7 +23,7 @@ const ReactHydrate = (element: React.ReactElement, container: Element) => {
2323
}
2424

2525
async 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

Comments
 (0)