test(query-core/query): add 'advanceTimersByTimeAsync' and remove 'vi.waitFor' (#9319)

Co-authored-by: Jonghyeon Ko <manudeli.ko@gmail.com>

Author: sukvvon

packages/query-core/src/__tests__/query.test.tsx

@@ -501,15 +501,14 @@ describe('query', () => {
     })
     const unsubscribe1 = observer.subscribe(() => undefined)
     unsubscribe1()
-    await vi.waitFor(() =>
-      expect(queryCache.find({ queryKey: key })).toBeUndefined(),
-    )
+
+    await vi.advanceTimersByTimeAsync(0)
+    expect(queryCache.find({ queryKey: key })).toBeUndefined()
     const unsubscribe2 = observer.subscribe(() => undefined)
     unsubscribe2()
 
-    await vi.waitFor(() =>
-      expect(queryCache.find({ queryKey: key })).toBeUndefined(),
-    )
+    await vi.advanceTimersByTimeAsync(0)
+    expect(queryCache.find({ queryKey: key })).toBeUndefined()
     expect(count).toBe(1)
   })
 
@@ -524,9 +523,9 @@ describe('query', () => {
     const unsubscribe = observer.subscribe(() => undefined)
     expect(queryCache.find({ queryKey: key })).toBeDefined()
     unsubscribe()
-    await vi.waitFor(() =>
-      expect(queryCache.find({ queryKey: key })).toBeUndefined(),
-    )
+
+    await vi.advanceTimersByTimeAsync(0)
+    expect(queryCache.find({ queryKey: key })).toBeUndefined()
   })
 
   test('should be garbage collected later when unsubscribed and query is fetching', async () => {
@@ -544,9 +543,8 @@ describe('query', () => {
     // unsubscribe should not remove even though gcTime has elapsed b/c query is still fetching
     expect(queryCache.find({ queryKey: key })).toBeDefined()
     // should be removed after an additional staleTime wait
-    await vi.waitFor(() =>
-      expect(queryCache.find({ queryKey: key })).toBeUndefined(),
-    )
+    await vi.advanceTimersByTimeAsync(30)
+    expect(queryCache.find({ queryKey: key })).toBeUndefined()
   })
 
   test('should not be garbage collected unless there are no subscribers', async () => {
@@ -887,12 +885,11 @@ describe('query', () => {
 
     queryClient.setQueryData(key, 'data')
 
-    await vi.waitFor(() =>
-      expect(fn).toHaveBeenCalledWith(
-        expect.objectContaining({
-          type: 'removed',
-        }),
-      ),
+    await vi.advanceTimersByTimeAsync(10)
+    expect(fn).toHaveBeenCalledWith(
+      expect.objectContaining({
+        type: 'removed',
+      }),
     )
 
     expect(queryClient.getQueryCache().findAll()).toHaveLength(0)