test: 更新用例以适配新的安装流程

Author: l246804

tests/index.test.ts

@@ -1,8 +1,9 @@
 import type { RouterPlugin } from '../src/plugin'
 import { describe, expect, it, vi } from 'vitest'
 import { computed, createApp, ref, watch } from 'vue'
-import { createMemoryHistory } from 'vue-router'
-import { asVuePlugin, createRouter } from '../src/index'
+import { createRouter as _createRouter, createMemoryHistory } from 'vue-router'
+import { createRouter, withInstall } from '../src/index'
+import { getInternals } from '../src/internals'
 
 describe.concurrent('vue Router Plugin System', () => {
   // 测试插件基础功能
@@ -107,27 +108,114 @@ describe.concurrent('vue Router Plugin System', () => {
     expect(logSpy).toHaveBeenCalledTimes(2)
   })
 
-  // 测试 asVuePlugin
-  it('test asVuePlugin', () => {
-    const counter = ref(0)
-    const logSpy = vi.fn()
-    const mockPlugin: RouterPlugin = () => {
-      const doubled = computed(() => counter.value * 2)
-      watch(doubled, logSpy, { immediate: true, flush: 'sync' })
-      counter.value = 1
-    }
+  // 多个 runWithApp 的刷新顺序
+  it('should flush runWithApp handlers in registration order', () => {
+    const calls: string[] = []
+    const p1: RouterPlugin = ctx => ctx.runWithApp(() => calls.push('a'))
+    const p2: RouterPlugin = ctx => ctx.runWithApp(() => calls.push('b'))
+    const p3: RouterPlugin = ctx => ctx.runWithApp(() => calls.push('c'))
 
     const router = createRouter({
       history: createMemoryHistory(),
       routes: [],
+      plugins: [p1, p2, p3],
     })
 
     const app = createApp(() => null)
-    app.use(router).use(asVuePlugin(mockPlugin))
+    app.use(router)
+
+    expect(calls).toEqual(['a', 'b', 'c'])
+  })
+
+  // 使用 withInstall：通过 router 直接安装（app.use 前，runWithApp 应延迟）
+  it('withInstall: install with router before app.use should defer runWithApp', () => {
+    const runWithAppSpy = vi.fn()
+    const Plugin = withInstall((ctx) => {
+      ctx.runWithApp(() => runWithAppSpy())
+    })
+
+    const router = _createRouter({
+      history: createMemoryHistory(),
+      routes: [],
+    })
+
+    // install via router directly
+    Plugin.install(router)
+
+    // before app.use, runWithApp should not run
+    expect(runWithAppSpy).not.toHaveBeenCalled()
+
+    const app = createApp(() => null)
+    app.use(router)
+    expect(runWithAppSpy).toHaveBeenCalledOnce()
+  })
+
+  // 使用 withInstall：通过 router 在 app.use 后安装（应立即执行 runWithApp）
+  it('withInstall: install with router after app.use should run runWithApp immediately', () => {
+    const runWithAppSpy = vi.fn()
+    const Plugin = withInstall((ctx) => {
+      ctx.runWithApp(() => runWithAppSpy())
+    })
+
+    const router = createRouter({
+      history: createMemoryHistory(),
+      routes: [],
+    })
+
+    const app = createApp(() => null)
+    app.use(router)
+
+    Plugin.install(router)
+    expect(runWithAppSpy).toHaveBeenCalledOnce()
+  })
+
+  // 使用 withInstall：通过 app.use 安装时，若未先注册 router，应抛错
+  it('withInstall: installing with app before router should throw', () => {
+    const Plugin = withInstall(() => {})
+    const app = createApp(() => null)
+    expect(() => app.use(Plugin as any)).toThrowError(
+      /Please install vue-router first/,
+    )
+  })
+
+  // 使用 withInstall：onUninstall 应在 app.unmount 时触发
+  it('withInstall: should register uninstall hook', () => {
+    const uninstallSpy = vi.fn()
+    const Plugin = withInstall((ctx) => {
+      ctx.onUninstall(uninstallSpy)
+    })
+
+    const router = createRouter({
+      history: createMemoryHistory(),
+      routes: [],
+    })
+
+    const app = createApp(() => null)
+    app.use(router)
+
+    Plugin.install(router)
+
+    expect(uninstallSpy).not.toHaveBeenCalled()
     app.unmount()
+    expect(uninstallSpy).toHaveBeenCalledOnce()
+  })
 
-    expect(logSpy).toHaveBeenCalledTimes(2)
-    counter.value = 2
-    expect(logSpy).toHaveBeenCalledTimes(2)
+  // overrideRouterInstall 幂等：仅包裹一次 install，并有内部标记
+  it('overrideRouterInstall is idempotent (wrapped once with internal flag)', () => {
+    const router = _createRouter({ history: createMemoryHistory(), routes: [] })
+    const originalInstall = router.install
+
+    const P1 = withInstall(() => {})
+    P1.install(router)
+    const wrappedOnce = router.install
+    expect(wrappedOnce).not.toBe(originalInstall)
+
+    const P2 = withInstall(() => {})
+    P2.install(router)
+    const wrappedTwice = router.install
+    expect(wrappedTwice).toBe(wrappedOnce)
+
+    // internal flag should be true
+    expect(getInternals(router).isInstallOverridden).toBe(true)
   })
 })