diff --git a/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx b/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx index 10e6948405..6976d851a2 100644 --- a/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx +++ b/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx @@ -384,12 +384,11 @@ describe('PersistQueryClientProvider', () => { function Page() { const state = useQuery({ queryKey: key, - queryFn: () => - sleep(10).then(() => { - fetched = true - return 'fetched' - }), - + queryFn: async () => { + await sleep(10) + fetched = true + return 'fetched' + }, staleTime: Infinity, }) @@ -415,6 +414,8 @@ describe('PersistQueryClientProvider', () => { expect(rendered.getByText('data: null')).toBeInTheDocument() await act(() => vi.advanceTimersByTimeAsync(10)) expect(rendered.getByText('data: hydrated')).toBeInTheDocument() + await act(() => vi.advanceTimersByTimeAsync(11)) + expect(rendered.getByText('data: hydrated')).toBeInTheDocument() expect(states).toHaveLength(2)