Merge branch 'master' of https://github.com/tannerlinsley/react-query

src/react/tests/useMutation.test.js

@@ -85,4 +85,115 @@ describe('useMutation', () => {
 
     console.error.mockRestore()
   })
+
+  it('should be able to call `onSuccess` and `onSettled` after each successful mutate', async () => {
+    let count = 0
+    const onSuccessMock = jest.fn()
+    const onSettledMock = jest.fn()
+
+    function Page() {
+      const [mutate] = useMutation(
+        async ({ count }) => Promise.resolve(count),
+        {
+          onSuccess: data => onSuccessMock(data),
+          onSettled: data => onSettledMock(data),
+        }
+      )
+
+      return (
+        <div>
+          <h1 data-testid="title">{count}</h1>
+          <button onClick={() => mutate({ count: ++count })}>mutate</button>
+        </div>
+      )
+    }
+
+    const { getByTestId, getByText } = render(<Page />)
+
+    expect(getByTestId('title').textContent).toBe('0')
+
+    fireEvent.click(getByText('mutate'))
+    fireEvent.click(getByText('mutate'))
+    fireEvent.click(getByText('mutate'))
+
+    await waitFor(() => getByTestId('title'))
+
+    expect(onSuccessMock).toHaveBeenCalledTimes(3)
+    expect(onSuccessMock).toHaveBeenCalledWith(1)
+    expect(onSuccessMock).toHaveBeenCalledWith(2)
+    expect(onSuccessMock).toHaveBeenCalledWith(3)
+
+    expect(onSettledMock).toHaveBeenCalledTimes(3)
+    expect(onSettledMock).toHaveBeenCalledWith(1)
+    expect(onSettledMock).toHaveBeenCalledWith(2)
+    expect(onSettledMock).toHaveBeenCalledWith(3)
+
+    expect(getByTestId('title').textContent).toBe('3')
+  })
+
+  it('should be able to call `onError` and `onSettled` after each failed mutate', async () => {
+    jest.spyOn(console, 'error')
+    console.error.mockImplementation(() => {})
+    const onErrorMock = jest.fn()
+    const onSettledMock = jest.fn()
+    let count = 0
+
+    function Page() {
+      const [mutate] = useMutation(
+        ({ count }) => {
+          const error = new Error(`Expected mock error. All is well! ${count}`)
+          error.stack = ''
+          return Promise.reject(error)
+        },
+        {
+          onError: error => onErrorMock(error.message),
+          onSettled: (_data, error) => onSettledMock(error.message),
+        },
+        {
+          throwOnError: false,
+        }
+      )
+
+      return (
+        <div>
+          <h1 data-testid="title">{count}</h1>
+          <button onClick={() => mutate({ count: ++count })}>mutate</button>
+        </div>
+      )
+    }
+
+    const { getByTestId, getByText } = render(<Page />)
+
+    expect(getByTestId('title').textContent).toBe('0')
+
+    fireEvent.click(getByText('mutate'))
+    fireEvent.click(getByText('mutate'))
+    fireEvent.click(getByText('mutate'))
+
+    await waitFor(() => getByTestId('title'))
+
+    expect(onErrorMock).toHaveBeenCalledTimes(3)
+    expect(onErrorMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 1'
+    )
+    expect(onErrorMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 2'
+    )
+    expect(onErrorMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 3'
+    )
+
+    expect(onSettledMock).toHaveBeenCalledTimes(3)
+    expect(onSettledMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 1'
+    )
+    expect(onSettledMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 2'
+    )
+    expect(onSettledMock).toHaveBeenCalledWith(
+      'Expected mock error. All is well! 3'
+    )
+
+    expect(getByTestId('title').textContent).toBe('3')
+  })
 })