UHD_Media_Tool/packages/contracts/test/pi.test.ts

import { describe, expect, it } from "vitest";
import * as C from "../src/index.ts";

// W7-A Pi 上游合同:跨字段约束的正反例、设置兼容。

describe("PiUpstreamCreate", () => {
  const ok = (v: unknown) => C.PiUpstreamCreate.safeParse(v).success;

  it("正例:api_key / oauth / none / 自定义端点", () => {
    expect(ok({ name: "OpenAI 主号", provider: "openai", authKind: "api_key", apiKey: "sk-1" })).toBe(true);
    expect(ok({ name: "Codex", provider: "openai-codex", authKind: "oauth", thinkingLevel: "max" })).toBe(true);
    expect(
      ok({
        name: "Bedrock",
        provider: "amazon-bedrock",
        authKind: "none",
        extra: { env: { AWS_REGION: "us-east-1" } },
      }),
    ).toBe(true);
    expect(
      ok({
        name: "本地 vLLM",
        provider: "openai-compatible",
        authKind: "none",
        baseUrl: "http://llm.local/v1",
        defaultModel: "qwen3",
        extra: { api: "openai-completions", compat: { supportsDeveloperRole: false } },
      }),
    ).toBe(true);
    expect(ok({ name: "fake", provider: "fake", authKind: "none" })).toBe(true);
  });

  it("反例:未知 provider、oauth 带 apiKey / baseUrl、自定义端点缺 baseUrl 或模型、extra 多余字段、名字为空", () => {
    expect(ok({ name: "x", provider: "google-gemini-cli", authKind: "oauth" })).toBe(false);
    expect(ok({ name: "x", provider: "anthropic", authKind: "oauth", apiKey: "sk" })).toBe(false);
    expect(ok({ name: "x", provider: "anthropic", authKind: "oauth", baseUrl: "https://p.local" })).toBe(false);
    expect(ok({ name: "x", provider: "openai-compatible", authKind: "api_key", defaultModel: "m" })).toBe(false);
    expect(ok({ name: "x", provider: "openai-compatible", authKind: "api_key", baseUrl: "http://a/v1" })).toBe(false);
    expect(ok({ name: "x", provider: "openai", authKind: "api_key", extra: { secret: "sk" } })).toBe(false);
    expect(ok({ name: "  ", provider: "openai", authKind: "api_key" })).toBe(false);
    expect(ok({ name: "x", provider: "openai", authKind: "token" })).toBe(false);
  });

  it("provider 枚举 = 42 个内置 + openai-compatible + fake", () => {
    expect(C.PI_BUILTIN_PROVIDER_IDS).toHaveLength(42);
    expect(C.PiProviderId.options).toHaveLength(44);
  });
});

describe("OAuth 请求", () => {
  it("start/complete 的边界", () => {
    expect(C.PiOAuthStartRequest.safeParse({}).success).toBe(true);
    expect(C.PiOAuthStartRequest.safeParse({ mode: "popup" }).success).toBe(false);
    expect(C.PiOAuthCompleteRequest.safeParse({ sessionId: "s" }).success).toBe(true);
    expect(C.PiOAuthCompleteRequest.safeParse({ sessionId: "s", waitMs: 120_000 }).success).toBe(false);
    expect(C.PiOAuthCompleteRequest.safeParse({ pastedUrlOrCode: "x" }).success).toBe(false);
  });
});

describe("settings.pi 兼容", () => {
  it("缺省 defaultUpstreamId = null;PATCH 不能直接改 defaultUpstreamId,旧字段被丢弃", () => {
    expect(C.DEFAULT_SETTINGS.pi.defaultUpstreamId).toBeNull();
    expect(C.GlobalSettings.safeParse(C.DEFAULT_SETTINGS).success).toBe(true);
    const patch = C.GlobalSettingsUpdate.parse({
      pi: {
        defaultUpstreamId: "7d3f7f5e-8a8c-4b8e-9f0a-1b2c3d4e5f60",
        provider: "fake",
        apiKey: "sk-x",
        model: "m",
        contextBudget: 64_000,
      },
    });
    expect(patch.pi).toEqual({ contextBudget: 64_000 });
    const current = { ...C.DEFAULT_SETTINGS, pi: { ...C.DEFAULT_SETTINGS.pi, defaultUpstreamId: "keep-me" } };
    expect(C.mergeSettings(current as never, patch).pi.defaultUpstreamId).toBe("keep-me");
  });
});