test(svelte-query/useMutationState): switch to fake timers, and replace 'waitFor' with 'vi.waitFor' (#9388)

sukvvon · manudeli · web-flow · commit 32467aa330c6 · 2025-07-11T13:30:46.000+09:00
Co-authored-by: Jonghyeon Ko &lt;manudeli.ko@gmail.com&gt;
diff --git a/packages/svelte-query/tests/useMutationState/useMutationState.test.ts b/packages/svelte-query/tests/useMutationState/useMutationState.test.ts
@@ -1,8 +1,16 @@
-import { describe, expect, test, vi } from 'vitest'
-import { fireEvent, render, waitFor } from '@testing-library/svelte'
+import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest'
+import { fireEvent, render } from '@testing-library/svelte'
 import BaseExample from './BaseExample.svelte'
 
 describe('useMutationState', () => {
+  beforeEach(() => {
+    vi.useFakeTimers()
+  })
+
+  afterEach(() => {
+    vi.useRealTimers()
+  })
+
   test('Run few mutation functions and check from useMutationState', async () => {
     const successMutationFn = vi.fn()
 
@@ -26,14 +34,14 @@ describe('useMutationState', () => {
 
     fireEvent.click(rendered.getByTestId('success'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(successMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual('["success"]')
     })
 
     fireEvent.click(rendered.getByTestId('error'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(errorMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual(
         '["success","error"]',
@@ -67,14 +75,14 @@ describe('useMutationState', () => {
 
     fireEvent.click(rendered.getByTestId('success'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(successMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual('[]')
     })
 
     fireEvent.click(rendered.getByTestId('error'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(errorMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual('["error"]')
     })
@@ -106,14 +114,14 @@ describe('useMutationState', () => {
 
     fireEvent.click(rendered.getByTestId('success'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(successMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual('["success"]')
     })
 
     fireEvent.click(rendered.getByTestId('error'))
 
-    await waitFor(() => {
+    await vi.waitFor(() => {
       expect(errorMutationFn).toHaveBeenCalledTimes(1)
       expect(rendered.getByTestId('result').innerHTML).toEqual('["success"]')
     })
