claude-code-hub/tests/unit/app/v1/url.test.ts

import { describe, expect, test, vi } from "vitest";

vi.mock("@/lib/logger", () => ({
  logger: {
    debug: vi.fn(),
    error: vi.fn(),
    info: vi.fn(),
    warn: vi.fn(),
  },
}));

import { buildProxyUrl } from "@/app/v1/_lib/url";

describe("buildProxyUrl", () => {
  test("标准拼接：baseUrl 无路径时使用 requestPath + search", () => {
    const out = buildProxyUrl(
      "https://api.example.com",
      new URL("https://dummy.com/v1/messages?x=1")
    );

    expect(out).toBe("https://api.example.com/v1/messages?x=1");
  });

  test("避免重复拼接：baseUrl 已包含 /responses 时不追加 /v1/responses", () => {
    const out = buildProxyUrl(
      "https://example.com/openai/responses",
      new URL("https://dummy.com/v1/responses?x=1")
    );

    expect(out).toBe("https://example.com/openai/responses?x=1");
  });

  test("避免重复拼接：baseUrl 已包含 /embeddings 时不追加 /v1/embeddings", () => {
    const out = buildProxyUrl(
      "https://example.com/openai/embeddings",
      new URL("https://dummy.com/v1/embeddings?x=1")
    );

    expect(out).toBe("https://example.com/openai/embeddings?x=1");
  });

  test("子路径不丢失：baseUrl=/v1/messages + request=/v1/messages/count_tokens", () => {
    const out = buildProxyUrl(
      "https://api.example.com/v1/messages",
      new URL("https://dummy.com/v1/messages/count_tokens")
    );

    expect(out).toBe("https://api.example.com/v1/messages/count_tokens");
  });

  test("带前缀路径的 baseUrl：/openai/messages + /v1/messages/count_tokens", () => {
    const out = buildProxyUrl(
      "https://example.com/openai/messages",
      new URL("https://dummy.com/v1/messages/count_tokens")
    );

    expect(out).toBe("https://example.com/openai/messages/count_tokens");
  });

  test("query 以 requestUrl 为准（覆盖 baseUrl 自带 query）", () => {
    const out = buildProxyUrl(
      "https://api.example.com/v1/messages?from=base",
      new URL("https://dummy.com/v1/messages?from=request")
    );

    expect(out).toBe("https://api.example.com/v1/messages?from=request");
  });

  test("baseUrl 以 /models 结尾时去除请求中的版本前缀", () => {
    const out = buildProxyUrl(
      "https://api.example.com/gemini/models",
      new URL("https://dummy.com/v1beta/models/gemini-1.5-pro:streamGenerateContent")
    );

    expect(out).toBe("https://api.example.com/gemini/models/gemini-1.5-pro:streamGenerateContent");
  });

  test("支持 v1internal 版本前缀", () => {
    const out = buildProxyUrl(
      "https://example.com/gemini/models",
      new URL("https://dummy.com/v1internal/models/gemini-2.5-flash:generateContent")
    );

    expect(out).toBe("https://example.com/gemini/models/gemini-2.5-flash:generateContent");
  });

  test("支持未来的版本前缀如 v2", () => {
    const out = buildProxyUrl(
      "https://example.com/api/models",
      new URL("https://dummy.com/v2/models/some-model:action")
    );

    expect(out).toBe("https://example.com/api/models/some-model:action");
  });
});