fix: adapt useMutationWithToast to react-query v5 callback signatures

This commit is contained in:
Kirill
2026-05-27 21:30:42 +05:00
parent 7642edc154
commit d254c3c813
2 changed files with 23 additions and 21 deletions
@@ -21,47 +21,49 @@ describe('useMutationWithToast', () => {
})
it('shows success notification on success with successMessage', async () => {
const mutationFn = vi.fn().mockResolvedValue({ ok: true })
const mutationFn = (): Promise<{ ok: boolean }> => Promise.resolve({ ok: true })
const { result } = renderHook(() => useMutationWithToast({ mutationFn, successMessage: 'Done!' }), {
wrapper: createWrapper(),
})
result.current.mutate()
;(result.current.mutate as () => void)()
await waitFor(() => expect(result.current.isSuccess).toBe(true))
expect(addNotification).toHaveBeenCalledWith({ type: 'success', message: 'Done!' })
})
it('does NOT show success notification without successMessage', async () => {
const mutationFn = vi.fn().mockResolvedValue({ ok: true })
const mutationFn = (): Promise<{ ok: boolean }> => Promise.resolve({ ok: true })
const { result } = renderHook(() => useMutationWithToast({ mutationFn }), { wrapper: createWrapper() })
result.current.mutate()
;(result.current.mutate as () => void)()
await waitFor(() => expect(result.current.isSuccess).toBe(true))
expect(addNotification).not.toHaveBeenCalled()
})
it('shows error notification on mutation error', async () => {
const mutationFn = vi.fn().mockRejectedValue(new Error('Boom'))
const err = new Error('Boom')
const mutationFn = (): Promise<never> => Promise.reject(err)
const { result } = renderHook(() => useMutationWithToast({ mutationFn }), { wrapper: createWrapper() })
result.current.mutate()
;(result.current.mutate as () => void)()
await waitFor(() => expect(result.current.isError).toBe(true))
expect(addNotification).toHaveBeenCalledWith({ type: 'error', message: 'Boom' })
})
it('calls user-provided onSuccess callback', async () => {
const onSuccess = vi.fn()
const mutationFn = vi.fn().mockResolvedValue({ ok: true })
const onSuccess: (data: { ok: boolean }, variables: void, onMutateResult: unknown, mutationContext: unknown) => void = vi.fn()
const mutationFn = (): Promise<{ ok: boolean }> => Promise.resolve({ ok: true })
const { result } = renderHook(() => useMutationWithToast({ mutationFn, onSuccess, successMessage: 'OK' }), {
wrapper: createWrapper(),
})
result.current.mutate()
;(result.current.mutate as () => void)()
await waitFor(() => expect(result.current.isSuccess).toBe(true))
expect(onSuccess).toHaveBeenCalled()
})
it('calls user-provided onError callback', async () => {
const onError = vi.fn()
const mutationFn = vi.fn().mockRejectedValue(new Error('fail'))
const onError: (error: Error, variables: void, onMutateResult: unknown, mutationContext: unknown) => void = vi.fn()
const err = new Error('fail')
const mutationFn = (): Promise<never> => Promise.reject(err)
const { result } = renderHook(() => useMutationWithToast({ mutationFn, onError }), { wrapper: createWrapper() })
result.current.mutate()
;(result.current.mutate as () => void)()
await waitFor(() => expect(result.current.isError).toBe(true))
expect(onError).toHaveBeenCalled()
})