test(react-query): use fake timers for useQuery.promise.test.tsx (#8890)

Author: minseong0324

packages/react-query/src/__tests__/useQuery.promise.test.tsx

@@ -12,7 +12,7 @@ import {
   useQuery,
 } from '..'
 import { QueryCache } from '../index'
-import { createQueryClient, queryKey, sleep } from './utils'
+import { createQueryClient, queryKey } from './utils'
 
 describe('useQuery().promise', () => {
   const queryCache = new QueryCache()
@@ -21,11 +21,13 @@ describe('useQuery().promise', () => {
   })
 
   beforeAll(() => {
+    vi.useFakeTimers({ shouldAdvanceTime: true })
     queryClient.setDefaultOptions({
       queries: { experimental_prefetchInRender: true },
     })
   })
   afterAll(() => {
+    vi.useRealTimers()
     queryClient.setDefaultOptions({
       queries: { experimental_prefetchInRender: false },
     })
@@ -52,7 +54,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test'
         },
       })
@@ -95,7 +97,7 @@ describe('useQuery().promise', () => {
         queryKey: key,
         queryFn: async () => {
           callCount++
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test'
         },
         staleTime: 1000,
@@ -149,7 +151,7 @@ describe('useQuery().promise', () => {
         queryKey: key,
         queryFn: async () => {
           callCount++
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test'
         },
         staleTime: 1000,
@@ -227,7 +229,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test'
         },
         initialData: 'initial',
@@ -263,7 +265,7 @@ describe('useQuery().promise', () => {
     const key = queryKey()
     const renderStream = createRenderStream({ snapshotDOM: true })
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(1)
+      await vi.advanceTimersByTimeAsync(1)
       return 'test'
     })
 
@@ -328,7 +330,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test'
         },
         placeholderData: 'placeholder',
@@ -381,7 +383,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: [...key, count],
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test-' + count
         },
         placeholderData: keepPreviousData,
@@ -450,7 +452,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return { name: 'test' }
         },
         select: (data) => data.name,
@@ -505,7 +507,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           if (++queryCount > 1) {
             // second time this query mounts, it should not throw
             return 'data'
@@ -587,7 +589,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           throw new Error('Error test')
         },
         retry: false,
@@ -645,7 +647,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: key,
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return 'test1'
         },
       })
@@ -689,7 +691,7 @@ describe('useQuery().promise', () => {
     const key = queryKey()
     const renderStream = createRenderStream({ snapshotDOM: true })
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(10)
+      await vi.advanceTimersByTimeAsync(10)
       return 'test'
     })
 
@@ -746,7 +748,7 @@ describe('useQuery().promise', () => {
     let count = 0
     const renderStream = createRenderStream({ snapshotDOM: true })
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(10)
+      await vi.advanceTimersByTimeAsync(10)
       return 'test' + count++
     })
 
@@ -805,7 +807,7 @@ describe('useQuery().promise', () => {
     const key = queryKey()
     let count = 0
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(10)
+      await vi.advanceTimersByTimeAsync(10)
       return 'test' + count++
     })
 
@@ -880,7 +882,7 @@ describe('useQuery().promise', () => {
     const renderStream = createRenderStream({ snapshotDOM: true })
     const key = queryKey()
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(10)
+      await vi.advanceTimersByTimeAsync(10)
       return 'test'
     })
 
@@ -938,7 +940,7 @@ describe('useQuery().promise', () => {
     const options = (count: number) => ({
       queryKey: [...key, count],
       queryFn: async () => {
-        await sleep(10)
+        await vi.advanceTimersByTimeAsync(10)
         return 'test' + count
       },
     })
@@ -994,7 +996,7 @@ describe('useQuery().promise', () => {
     queryClient.setQueryData(key, 'initial')
 
     const queryFn = vi.fn().mockImplementation(async () => {
-      await sleep(1)
+      await vi.advanceTimersByTimeAsync(1)
       return 'test'
     })
 
@@ -1056,7 +1058,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: [key, count],
         queryFn: async () => {
-          await sleep(10)
+          await vi.advanceTimersByTimeAsync(10)
           return 'test' + count
         },
         staleTime: Infinity,
@@ -1136,7 +1138,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: [key, count],
         queryFn: async () => {
-          await sleep(10)
+          await vi.advanceTimersByTimeAsync(10)
           return 'test' + count
         },
         staleTime: Infinity,
@@ -1212,7 +1214,7 @@ describe('useQuery().promise', () => {
       const query = useQuery({
         queryKey: [key, count],
         queryFn: async () => {
-          await sleep(10)
+          await vi.advanceTimersByTimeAsync(10)
           return 'test' + count + modifier
         },
       })
@@ -1314,7 +1316,7 @@ describe('useQuery().promise', () => {
         staleTime: Infinity,
         queryKey: [key, input],
         queryFn: async () => {
-          await sleep(1)
+          await vi.advanceTimersByTimeAsync(1)
           return input + ' response'
         },
       })