|
1 | 1 | import * as React from 'react'
|
2 | 2 | import { renderToString } from 'react-dom/server'
|
3 |
| -import { describe, expect, it, vi } from 'vitest' |
| 3 | +import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest' |
4 | 4 | import { QueryCache, QueryClientProvider, useInfiniteQuery, useQuery } from '..'
|
5 |
| -import { createQueryClient, queryKey, setIsServer, sleep } from './utils' |
| 5 | +import { createQueryClient, queryKey, setIsServer } from './utils' |
6 | 6 |
|
7 | 7 | describe('Server Side Rendering', () => {
|
8 | 8 | setIsServer(true)
|
9 | 9 |
|
| 10 | + beforeEach(() => { |
| 11 | + vi.useFakeTimers() |
| 12 | + }) |
| 13 | + |
| 14 | + afterEach(() => { |
| 15 | + vi.useRealTimers() |
| 16 | + }) |
| 17 | + |
10 | 18 | it('should not trigger fetch', () => {
|
11 | 19 | const queryCache = new QueryCache()
|
12 | 20 | const queryClient = createQueryClient({ queryCache })
|
@@ -54,8 +62,8 @@ describe('Server Side Rendering', () => {
|
54 | 62 | const queryCache = new QueryCache()
|
55 | 63 | const queryClient = createQueryClient({ queryCache })
|
56 | 64 | const key = queryKey()
|
57 |
| - const queryFn = vi.fn(() => { |
58 |
| - sleep(10) |
| 65 | + const queryFn = vi.fn(async () => { |
| 66 | + await vi.advanceTimersByTimeAsync(10) |
59 | 67 | return 'data'
|
60 | 68 | })
|
61 | 69 |
|
@@ -123,7 +131,7 @@ describe('Server Side Rendering', () => {
|
123 | 131 | const queryClient = createQueryClient({ queryCache })
|
124 | 132 | const key = queryKey()
|
125 | 133 | const queryFn = vi.fn(async () => {
|
126 |
| - await sleep(5) |
| 134 | + await vi.advanceTimersByTimeAsync(5) |
127 | 135 | return 'page 1'
|
128 | 136 | })
|
129 | 137 |
|
|
0 commit comments