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..3eae33c02f 100644 --- a/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx +++ b/packages/react-query-persist-client/src/__tests__/PersistQueryClientProvider.test.tsx @@ -41,8 +41,9 @@ const createMockErrorPersister = ( // noop }, async restoreClient() { - await sleep(10) - throw error + return sleep(10).then(() => { + throw error + }) }, removeClient, }, @@ -415,6 +416,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)