1
- import { describe , expect , test , vi } from 'vitest'
1
+ import { afterEach , beforeEach , describe , expect , test , vi } from 'vitest'
2
2
import { sleep } from '@tanstack/query-test-utils'
3
3
import { useInfiniteQuery } from '../useInfiniteQuery'
4
4
import { infiniteQueryOptions } from '../infiniteQueryOptions'
5
5
6
6
vi . mock ( '../useQueryClient' )
7
7
8
8
describe ( 'useQuery' , ( ) => {
9
+ beforeEach ( ( ) => {
10
+ vi . useFakeTimers ( )
11
+ } )
12
+
13
+ afterEach ( ( ) => {
14
+ vi . useRealTimers ( )
15
+ } )
16
+
9
17
test ( 'should properly execute infinite query' , async ( ) => {
10
18
const { data, fetchNextPage, status } = useInfiniteQuery ( {
11
19
queryKey : [ 'infiniteQuery' ] ,
@@ -18,7 +26,7 @@ describe('useQuery', () => {
18
26
expect ( data . value ) . toStrictEqual ( undefined )
19
27
expect ( status . value ) . toStrictEqual ( 'pending' )
20
28
21
- await sleep ( 0 )
29
+ await vi . advanceTimersByTimeAsync ( 0 )
22
30
23
31
expect ( data . value ) . toStrictEqual ( {
24
32
pageParams : [ 0 ] ,
@@ -28,7 +36,7 @@ describe('useQuery', () => {
28
36
29
37
fetchNextPage ( )
30
38
31
- await sleep ( 0 )
39
+ await vi . advanceTimersByTimeAsync ( 0 )
32
40
33
41
expect ( data . value ) . toStrictEqual ( {
34
42
pageParams : [ 0 , 12 ] ,
@@ -50,7 +58,7 @@ describe('useQuery', () => {
50
58
expect ( data . value ) . toStrictEqual ( undefined )
51
59
expect ( status . value ) . toStrictEqual ( 'pending' )
52
60
53
- await sleep ( 0 )
61
+ await vi . advanceTimersByTimeAsync ( 0 )
54
62
55
63
expect ( data . value ) . toStrictEqual ( {
56
64
pageParams : [ 0 ] ,
@@ -60,7 +68,7 @@ describe('useQuery', () => {
60
68
61
69
fetchNextPage ( )
62
70
63
- await sleep ( 0 )
71
+ await vi . advanceTimersByTimeAsync ( 0 )
64
72
65
73
expect ( data . value ) . toStrictEqual ( {
66
74
pageParams : [ 0 , 12 ] ,
0 commit comments