Skip to content

Commit 027be32

Browse files
authored
test(react-query): use fake timers for fine-grained-persister.test.tsx (#9056)
1 parent 0849660 commit 027be32

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

packages/react-query/src/__tests__/fine-grained-persister.test.tsx

+13-6
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1-
import { describe, expect, it, vi } from 'vitest'
2-
import { waitFor } from '@testing-library/react'
1+
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
32
import * as React from 'react'
43
import { QueryCache, hashKey } from '@tanstack/query-core'
54
import {
@@ -10,6 +9,14 @@ import { useQuery } from '..'
109
import { createQueryClient, queryKey, renderWithClient, sleep } from './utils'
1110

1211
describe('fine grained persister', () => {
12+
beforeEach(() => {
13+
vi.useFakeTimers()
14+
})
15+
16+
afterEach(() => {
17+
vi.useRealTimers()
18+
})
19+
1320
const queryCache = new QueryCache()
1421
const queryClient = createQueryClient({ queryCache })
1522

@@ -61,7 +68,7 @@ describe('fine grained persister', () => {
6168

6269
const rendered = renderWithClient(queryClient, <Test />)
6370

64-
await waitFor(() => rendered.getByText('Works from persister'))
71+
await vi.waitFor(() => rendered.getByText('Works from persister'))
6572
expect(spy).not.toHaveBeenCalled()
6673
})
6774

@@ -116,8 +123,8 @@ describe('fine grained persister', () => {
116123

117124
const rendered = renderWithClient(queryClient, <Test />)
118125

119-
await waitFor(() => rendered.getByText('Works from persister'))
120-
await waitFor(() => rendered.getByText('Works from queryFn'))
126+
await vi.waitFor(() => rendered.getByText('Works from persister'))
127+
await vi.waitFor(() => rendered.getByText('Works from queryFn'))
121128
expect(spy).toHaveBeenCalledTimes(1)
122129
})
123130

@@ -155,7 +162,7 @@ describe('fine grained persister', () => {
155162

156163
const rendered = renderWithClient(queryClient, <Test />)
157164

158-
await waitFor(() => rendered.getByText('Works from queryFn'))
165+
await vi.waitFor(() => rendered.getByText('Works from queryFn'))
159166
expect(spy).toHaveBeenCalledTimes(1)
160167

161168
const storedItem = await storage.getItem(`${PERSISTER_KEY_PREFIX}-${hash}`)

0 commit comments

Comments
 (0)