snapshot of https://github.com/TanStack/query.git (main, 2026-09-16; blobs > 131072B omitted)

packages/vue-query/src/__tests__/useMutationState.test.ts

@@ -0,0 +1,81 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import { ref } from 'vue-demi'
+import { queryKey, sleep } from '@tanstack/query-test-utils'
+import { useMutation } from '../useMutation'
+import { useMutationState } from '../useMutationState'
+import { useQueryClient } from '../useQueryClient'
+
+vi.mock('../useQueryClient')
+
+describe('useMutationState', () => {
+  beforeEach(() => {
+    vi.useFakeTimers()
+  })
+
+  afterEach(() => {
+    vi.useRealTimers()
+  })
+
+  it('should return variables after calling mutate 1', () => {
+    const key = queryKey()
+    const variables = 'foo123'
+
+    const { mutate } = useMutation({
+      mutationKey: key,
+      mutationFn: (params: string) => sleep(0).then(() => params),
+    })
+
+    mutate(variables)
+
+    const mutationState = useMutationState({
+      filters: { mutationKey: key, status: 'pending' },
+      select: (mutation) => mutation.state.variables,
+    })
+
+    expect(mutationState.value).toEqual([variables])
+  })
+
+  it('should return variables after calling mutate 2', () => {
+    const queryClient = useQueryClient()
+    queryClient.clear()
+    const key = queryKey()
+    const variables = 'bar234'
+
+    const { mutate } = useMutation({
+      mutationKey: key,
+      mutationFn: (params: string) => sleep(0).then(() => params),
+    })
+
+    mutate(variables)
+
+    const mutationState = useMutationState()
+
+    expect(mutationState.value[0]?.variables).toEqual(variables)
+  })
+
+  it('should work with options getter and be reactive', async () => {
+    const key = queryKey()
+    const keyRef = ref('useMutationStateGetter2')
+    const variables = 'foo123'
+
+    const { mutate } = useMutation({
+      mutationKey: key,
+      mutationFn: (params: string) => sleep(10).then(() => params),
+    })
+
+    mutate(variables)
+
+    const mutationState = useMutationState(() => ({
+      filters: { mutationKey: [keyRef.value], status: 'pending' },
+      select: (mutation) => mutation.state.variables,
+    }))
+
+    expect(mutationState.value).toEqual([])
+
+    keyRef.value = key[0]!
+
+    await vi.advanceTimersByTimeAsync(0)
+
+    expect(mutationState.value).toEqual([variables])
+  })
+})