test: todo

Author: posva

src/data-loaders/defineColadaLoader.spec.ts

@@ -1,7 +1,7 @@
 /**
  * @vitest-environment happy-dom
  */
-import { App, defineComponent } from 'vue'
+import { App, defineComponent, markRaw } from 'vue'
 import { defineColadaLoader } from './defineColadaLoader'
 import {
   describe,
@@ -15,6 +15,7 @@ import {
 import {
   DataLoaderPlugin,
   DataLoaderPluginOptions,
+  getCurrentContext,
   setCurrentContext,
   UseDataLoader,
 } from 'unplugin-vue-router/runtime'
@@ -23,7 +24,12 @@ import { getRouter } from 'vue-router-mock'
 import { enableAutoUnmount, mount } from '@vue/test-utils'
 import RouterViewMock from '../../tests/data-loaders/RouterViewMock.vue'
 import { setActivePinia, createPinia, getActivePinia } from 'pinia'
-import { useQuery, PiniaColada } from '@pinia/colada'
+import {
+  useQuery,
+  PiniaColada,
+  useQueryCache,
+  reviveTreeMap,
+} from '@pinia/colada'
 import { RouteLocationNormalizedLoaded } from 'vue-router'
 
 describe(
@@ -205,5 +211,56 @@ describe(
       expect(query).toHaveBeenCalledTimes(2)
       expect(loaderData.value).toBe('1')
     })
+
+    // NOTE: this test should fail if the `setCurrentContext(currentContext)` is not called in the `if (isInitial)` branch
+    it.todo('restores the context after using a loader', async () => {
+      const query = vi.fn().mockResolvedValue('data')
+
+      const useData = defineColadaLoader({
+        query,
+        key: () => ['id'],
+      })
+
+      let useDataResult: ReturnType<typeof useData> | undefined
+      const component = defineComponent({
+        setup() {
+          useDataResult = useData()
+          expect(getCurrentContext()).toEqual([])
+
+          const { data, error, isLoading } = useDataResult
+          return { data, error, isLoading }
+        },
+        template: `<p/>`,
+      })
+
+      const router = getRouter()
+      router.addRoute({
+        name: '_test',
+        path: '/fetch',
+        meta: {
+          loaders: [useData],
+        },
+        component,
+      })
+
+      const pinia = createPinia()
+
+      const treeMap = reviveTreeMap([
+        ['id', ['data', null, Date.now()], undefined],
+      ])
+      pinia.state.value[useQueryCache.$id] = { caches: markRaw(treeMap) }
+
+      const wrapper = mount(RouterViewMock, {
+        global: {
+          plugins: [[DataLoaderPlugin, { router }], pinia, PiniaColada],
+        },
+      })
+
+      await router.push('/fetch')
+
+      expect(useDataResult?.data.value).toBe('data')
+
+      expect(getCurrentContext()).toEqual([])
+    })
   }
 )