test: query and mutation

src/tree-map.ts

@@ -93,6 +93,7 @@ export class TreeMapNode<T = unknown> {
 // -------------------------------------
 // --- Below are debugging internals ---
 // -------------------------------------
+/* v8 ignore start */
 
 /**
  * Calculates the size of the node and all its children. Used in tests.
@@ -111,6 +112,13 @@ export function entryNodeSize(node: TreeMapNode): number {
   )
 }
 
+/**
+ * Logs the tree to the console. Used in tests.
+ * @internal
+ *
+ * @param tree - tree to log
+ * @param log - function to log the tree
+ */
 export function logTree(
   tree: TreeMapNode,
   // eslint-disable-next-line no-console
@@ -160,3 +168,4 @@ function printTreeMap(
 
   return treeStr
 }
+/* v8 ignore stop */

src/use-mutation.spec.ts

@@ -144,12 +144,37 @@ describe('useMutation', () => {
     )
   })
 
-  it('invokes the "onSuccess" hook', async () => {
-    const onSuccess = vi.fn()
+  it('skips setting the error if "onError" throws', async () => {
+    const onError = vi.fn().mockRejectedValueOnce(new Error('onError'))
     const { wrapper } = mountSimple({
-      onSuccess,
+      mutation: async () => {
+        throw new Error('mutation')
+      },
+      onError,
     })
 
+    wrapper.vm.mutate()
+    await runTimers()
+    expect(onError).toHaveBeenCalled()
+    // couldn't be set
+    expect(wrapper.vm.error).toEqual(null)
+  })
+
+  it('awaits the "onMutate" hook before mutation', async () => {
+    const onMutate = vi.fn(() => delay(10))
+    const { wrapper, mutation } = mountSimple({ onMutate })
+
+    wrapper.vm.mutate()
+    expect(onMutate).toHaveBeenCalled()
+    expect(mutation).not.toHaveBeenCalled()
+    await runTimers()
+    expect(mutation).toHaveBeenCalled()
+  })
+
+  it('invokes the "onSuccess" hook', async () => {
+    const onSuccess = vi.fn()
+    const { wrapper } = mountSimple({ onSuccess })
+
     wrapper.vm.mutate()
     await runTimers()
     expect(onSuccess).toHaveBeenCalledWith(
@@ -160,6 +185,38 @@ describe('useMutation', () => {
     )
   })
 
+  it('skips setting the data if "onSuccess" throws', async () => {
+    const onSuccess = vi.fn().mockRejectedValueOnce(new Error('onSuccess'))
+    const { wrapper, mutation } = mountSimple({ onSuccess })
+
+    wrapper.vm.mutate()
+    await runTimers()
+    expect(onSuccess).toHaveBeenCalled()
+    expect(mutation).toHaveBeenCalled()
+    // since it threw
+    expect(wrapper.vm.data).toBeUndefined()
+  })
+
+  it('sets the error if "onSuccess" throws', async () => {
+    const onSuccess = vi.fn().mockRejectedValueOnce(new Error('onSuccess'))
+    const { wrapper } = mountSimple({ onSuccess })
+
+    wrapper.vm.mutate()
+    await runTimers()
+    expect(onSuccess).toHaveBeenCalled()
+    expect(wrapper.vm.error).toEqual(new Error('onSuccess'))
+  })
+
+  it('sets the error if "onMutate" throws', async () => {
+    const onMutate = vi.fn().mockRejectedValueOnce(new Error('onMutate'))
+    const { wrapper } = mountSimple({ onMutate })
+
+    wrapper.vm.mutate()
+    await runTimers()
+    expect(onMutate).toHaveBeenCalled()
+    expect(wrapper.vm.error).toEqual(new Error('onMutate'))
+ })
+
   describe('invokes the "onSettled" hook', () => {
     it('on success', async () => {
       const onSettled = vi.fn()
@@ -198,4 +255,16 @@ describe('useMutation', () => {
       )
     })
   })
+
+  it('can reset the mutation', async () => {
+    const { wrapper } = mountSimple()
+
+    wrapper.vm.mutate()
+    await runTimers()
+    expect(wrapper.vm.data).toBe(42)
+    wrapper.vm.reset()
+    expect(wrapper.vm.data).toBeUndefined()
+    expect(wrapper.vm.error).toBeNull()
+    expect(wrapper.vm.status).toBe('pending')
+  })
 })

src/use-query.spec.ts

@@ -500,5 +500,40 @@ describe('useQuery', () => {
       expect(query).toHaveBeenCalledTimes(0)
       expect(wrapper.vm.data).toBe(2)
     })
+
+    it('sets the error if the initial data is an error', async () => {
+      const pinia = createPinia()
+
+      const caches = shallowReactive(
+        new TreeMapNode<UseQueryEntry>(
+          ['key'],
+          // fresh data
+          createQueryEntry(undefined, new Error('fail'), Date.now()),
+        ),
+      )
+      pinia.state.value[QUERY_STORE_ID] = { caches }
+      const { wrapper } = mountSimple({ refetchOnMount: false }, { plugins: [pinia] })
+
+      expect(wrapper.vm.status).toBe('error')
+      expect(wrapper.vm.error).toEqual(new Error('fail'))
+      expect(wrapper.vm.data).toBe(undefined)
+    })
+
+    it('sets the initial error even with initialData', async () => {
+      const pinia = createPinia()
+
+      const caches = shallowReactive(
+        new TreeMapNode<UseQueryEntry>(
+          ['key'],
+          // fresh data
+          createQueryEntry(undefined, new Error('fail'), Date.now()),
+        ),
+      )
+      pinia.state.value[QUERY_STORE_ID] = { caches }
+      const { wrapper } = mountSimple({ refetchOnMount: false, initialData: () => 42 }, { plugins: [pinia] })
+
+      expect(wrapper.vm.status).toBe('error')
+      expect(wrapper.vm.error).toEqual(new Error('fail'))
+    })
   })
 })

test/utils.ts

@@ -3,7 +3,7 @@ import type { Mock } from 'vitest'
 import { vi } from 'vitest'
 import { nextTick } from 'vue'
 
-export const delay = (ms: number) => new Promise((r) => setTimeout(r, ms))
+export const delay = (ms: number) => new Promise<void>((r) => setTimeout(r, ms))
 
 export type GlobalMountOptions = NonNullable<(Parameters<typeof mount>)[1]>['global']
 

vitest.config.ts

@@ -8,7 +8,7 @@ export default defineConfig({
     environment: 'happy-dom',
     coverage: {
       provider: 'v8',
-      reporter: ['text', 'lcov'],
+      reporter: ['text', 'lcovonly', 'html'],
       all: true,
       include: ['src'],
       exclude: ['src/index.ts', 'src/**/*.test-d.ts'],