From c329015dedd22f05b0d7d7b3871681e513a45c4a Mon Sep 17 00:00:00 2001 From: Jonghyeon Ko Date: Wed, 23 Apr 2025 19:14:27 +0900 Subject: [PATCH] test(react-query): use fake timers for fine-grained-persister.test.tsx --- .../__tests__/fine-grained-persister.test.tsx | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/packages/react-query/src/__tests__/fine-grained-persister.test.tsx b/packages/react-query/src/__tests__/fine-grained-persister.test.tsx index 7b78e8eea0..800e708559 100644 --- a/packages/react-query/src/__tests__/fine-grained-persister.test.tsx +++ b/packages/react-query/src/__tests__/fine-grained-persister.test.tsx @@ -1,5 +1,4 @@ -import { describe, expect, it, vi } from 'vitest' -import { waitFor } from '@testing-library/react' +import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest' import * as React from 'react' import { QueryCache, hashKey } from '@tanstack/query-core' import { @@ -10,6 +9,14 @@ import { useQuery } from '..' import { createQueryClient, queryKey, renderWithClient, sleep } from './utils' describe('fine grained persister', () => { + beforeEach(() => { + vi.useFakeTimers() + }) + + afterEach(() => { + vi.useRealTimers() + }) + const queryCache = new QueryCache() const queryClient = createQueryClient({ queryCache }) @@ -61,7 +68,7 @@ describe('fine grained persister', () => { const rendered = renderWithClient(queryClient, ) - await waitFor(() => rendered.getByText('Works from persister')) + await vi.waitFor(() => rendered.getByText('Works from persister')) expect(spy).not.toHaveBeenCalled() }) @@ -116,8 +123,8 @@ describe('fine grained persister', () => { const rendered = renderWithClient(queryClient, ) - await waitFor(() => rendered.getByText('Works from persister')) - await waitFor(() => rendered.getByText('Works from queryFn')) + await vi.waitFor(() => rendered.getByText('Works from persister')) + await vi.waitFor(() => rendered.getByText('Works from queryFn')) expect(spy).toHaveBeenCalledTimes(1) }) @@ -155,7 +162,7 @@ describe('fine grained persister', () => { const rendered = renderWithClient(queryClient, ) - await waitFor(() => rendered.getByText('Works from queryFn')) + await vi.waitFor(() => rendered.getByText('Works from queryFn')) expect(spy).toHaveBeenCalledTimes(1) const storedItem = await storage.getItem(`${PERSISTER_KEY_PREFIX}-${hash}`)