@zhushanwen/pi-provider-live-sync 0.2.1 → 0.3.1

package/src/__tests__/model-sync.test.ts

@@ -1,7 +1,7 @@
 /**
  * provider-live-sync 单元测试（设计 §4.2「单元（extension）」全部断言）。
  *
- * 被测面 = 纯函数（`readObservation` / `evaluateSnapshot`），不依赖 pi 运行时：
+ * 被测对象 = 纯函数（`readObservation` / `evaluateSnapshot`），不依赖 pi 运行时：
  * 决策与基线推进是全部语义所在，时钟/timer 不进断言（timer 只做调度）。
  *
  * 覆盖映射（设计行 → 用例）：
@@ -19,12 +19,12 @@ import type { ExtensionAPI } from '@earendil-works/pi-coding-agent'
 import providerLiveSync, {
   evaluateSnapshot,
   readObservation,
-  POLL_INTERVAL_MS,
   WATCHED_FILES,
   type FileObservation,
   type Snapshot,
   type WatchedFile,
 } from '../index.ts'
+import { startWithRegistryCtx, tickOnce } from './tick-harness.ts'
 
 // tick 层用例：mock pi 宿主模块的 getAgentDir 指向真实 tmp fixture。纯函数用例不经
 // getAgentDir（readObservation 显式传目录），mock 对它们无影响。
@@ -110,7 +110,8 @@ describe('evaluateSnapshot —— D3 四态（models.json 缺失是 catalog-only
     const d = evaluateSnapshot(snap, obs({ 'models.json': { state: 'absent' }, 'auth.json': present('X') }))
     expect(d).toEqual({ action: 'skip-missing', missingFiles: ['models.json'] })
     // 持续缺失：抑制保持（每拍仍判定 skip-missing），但调用方按跃迁去重 → 只在第一拍记一行
-    // 日志（`missingLogged`），后续拍零输出（稳态无噪声，设计 §3.6 行 2/S9a②）。
+    // 日志（`missingLogged`），后续拍零输出（抑制持续语义见 src/index.ts 的
+    // Snapshot.suppressed 注释与 evaluateSnapshot 实装）。
     const d2 = evaluateSnapshot(snap, obs({ 'models.json': { state: 'absent' }, 'auth.json': present('X') }))
     expect(d2).toEqual({ action: 'skip-missing', missingFiles: ['models.json'] })
   })
@@ -242,40 +243,13 @@ describe('tick 层 —— 首基线置位时机与 refresh 触发', () => {
     rmSync(agentDir, { recursive: true, force: true, maxRetries: 5, retryDelay: 20 })
   })
 
+  /** 经 harness 启动扩展（pi 宿主面 = appendEntry spy；拍屏障说明见 tick-harness 头注释）。 */
   function startWithFakeCtx(): { refresh: ReturnType<typeof vi.fn>; appendEntry: ReturnType<typeof vi.fn> } {
-    const refresh = vi.fn(async () => ({ aborted: false, errors: new Map() }))
-    const getError = vi.fn(() => null)
-    const ctx = { modelRegistry: { refresh, getError } }
     const appendEntry = vi.fn()
-    const on = vi.fn()
-    providerLiveSync({ on, appendEntry } as unknown as ExtensionAPI)
-    const handler = on.mock.calls.find((call) => call[0] === 'session_start')?.[1] as
-      | ((event: unknown, sessionCtx: unknown) => void)
-      | undefined
-    expect(handler).toBeTypeOf('function')
-    handler({ type: 'session_start' }, ctx)
+    const { refresh } = startWithRegistryCtx({ appendEntry })
     return { refresh, appendEntry }
   }
 
-  /** 逐真实事件循环轮次（setImmediate 未被 fake）轮询等待谓词成立。 */
-  async function waitFor(predicate: () => boolean, maxTurns = 5_000): Promise<void> {
-    for (let i = 0; i < maxTurns && !predicate(); i += 1) {
-      await new Promise<void>((resolve) => setImmediate(resolve))
-    }
-  }
-
-  /**
-   * 推进一个轮询周期并等本拍 tick **完整落地**。确定性拍屏障：tick 完成后其
-   * `.finally(schedule)` 必然注册下一个 fake timer——等 `getTimerCount() >= 1`
-   * 即证明本拍 readFile 链（真实线程池 I/O）已结束。固定轮次 flush 不行：线程池
-   * 完成回调的到达轮次不定，偶发「tick 晚于下一拍的 writeFileSync 完成 → 新内容
-   * 被当首基线」假红。
-   */
-  async function tickOnce(): Promise<void> {
-    await vi.advanceTimersByTimeAsync(POLL_INTERVAL_MS)
-    await waitFor(() => vi.getTimerCount() >= 1)
-  }
-
   it('MF-1-1：首拍双 absent（catalog-only 全新安装）→ auth.json 首次出现 → 必须 refresh', async () => {
     const { refresh } = startWithFakeCtx()
     await tickOnce() // 首拍：双 absent 建基线，decision=none（首拍本就不刷新）

package/src/__tests__/pi-runtime-testkit.ts

@@ -0,0 +1,57 @@
+/**
+ * pi 运行时契约测试共用夹具（pi-semantics-catalog-injection 与 sdk-contract 的同款
+ * runtime 装配、注册入参与可见集合断言）。不 mock pi 本体——契约测试的价值即
+ * 「实装版 pi 的真实组合行为」，夹具只收敛重复形态，断言对象仍由各用例持有。
+ */
+import { expect } from 'vitest'
+import { join } from 'node:path'
+import { ModelRuntime } from '@earendil-works/pi-coding-agent'
+import { getBuiltinModels } from '@earendil-works/pi-ai/providers/all'
+
+/** pi-ai 内置基线模型（getBuiltinModels 的元素类型）。 */
+export type BuiltinModel = ReturnType<typeof getBuiltinModels>[number]
+
+/** 每用例独立 runtime：内存 ModelsStore + 不发起网络刷新 + tmp 凭据路径（同款隔离）。 */
+async function newRuntime(authDir: string): Promise<ModelRuntime> {
+  return ModelRuntime.create({
+    modelsPath: null,
+    refreshOnCreate: false,
+    authPath: join(authDir, 'auth.json'),
+  })
+}
+
+/** 全量内置基线的注册入参（逐模型拷贝，注册侧不持共享引用）。 */
+export function fullBaselineConfig(provider: string): { models: BuiltinModel[] } {
+  return { models: getBuiltinModels(provider).map((m) => ({ ...m })) }
+}
+
+/** 单模型注册入参（「整体替换 / 可见集合收缩」类用例）。 */
+function singleModelConfig(model: BuiltinModel): { models: BuiltinModel[] } {
+  return { models: [{ ...model }] }
+}
+
+/**
+ * 新 runtime、注册全量内置基线并**自证可见集合 = 基线**（契约用例共同开场；断言单源
+ * 在此执行，防开场即坏），返回 baseline 供后续断言引用。
+ */
+export async function runtimeWithFullBaseline(
+  provider: string,
+  authDir: string,
+): Promise<{ rt: ModelRuntime; baseline: BuiltinModel[] }> {
+  const baseline = getBuiltinModels(provider)
+  const rt = await newRuntime(authDir)
+  rt.registerProvider(provider, fullBaselineConfig(provider))
+  expectVisibleIds(rt, provider, baseline.map((m) => m.id))
+  return { rt, baseline }
+}
+
+/** 换注单模型并断言可见集合收缩为该模型（pi 重注册整体替换 / sdk 映射替换共用断言）。 */
+export function swapToSingleExpectShrink(rt: ModelRuntime, provider: string, model: BuiltinModel): void {
+  rt.registerProvider(provider, singleModelConfig(model))
+  expectVisibleIds(rt, provider, [model.id])
+}
+
+/** 该 provider 可见模型 id 顺序断言（getModels 即最终可见集合）。 */
+export function expectVisibleIds(rt: ModelRuntime, provider: string, ids: string[]): void {
+  expect(rt.getModels(provider).map((m) => m.id)).toEqual(ids)
+}