claude-code-hub/tests/unit/lib/provider-endpoints/probe.test.ts

import { afterEach, describe, expect, test, vi } from "vitest";
import type { ProviderEndpoint } from "@/types/provider";

function makeEndpoint(overrides: Partial<ProviderEndpoint>): ProviderEndpoint {
  return {
    id: 1,
    vendorId: 1,
    providerType: "claude",
    url: "https://example.com",
    label: null,
    sortOrder: 0,
    isEnabled: true,
    lastProbedAt: null,
    lastProbeOk: null,
    lastProbeStatusCode: null,
    lastProbeLatencyMs: null,
    lastProbeErrorType: null,
    lastProbeErrorMessage: null,
    createdAt: new Date(0),
    updatedAt: new Date(0),
    deletedAt: null,
    ...overrides,
  };
}

function createCircuitBreakerMock(overrides: Partial<Record<string, unknown>> = {}) {
  return {
    getEndpointCircuitStateSync: vi.fn(() => "closed"),
    resetEndpointCircuit: vi.fn(async () => {}),
    recordEndpointFailure: vi.fn(async () => {}),
    ...overrides,
  };
}

afterEach(() => {
  vi.unstubAllGlobals();
  vi.useRealTimers();
  delete process.env.ENDPOINT_PROBE_METHOD;
});

describe("provider-endpoints: probe", () => {
  test("probeEndpointUrl: HEAD 成功时直接返回，不触发 GET", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
      updateProviderEndpointProbeSnapshot: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const fetchMock = vi.fn(async (_url: string, init?: RequestInit) => {
      if (init?.method === "HEAD") {
        return new Response(null, { status: 204 });
      }
      throw new Error("unexpected");
    });
    vi.stubGlobal("fetch", fetchMock);

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 1234);

    expect(result).toEqual(
      expect.objectContaining({ ok: true, method: "HEAD", statusCode: 204, errorType: null })
    );
    expect(fetchMock).toHaveBeenCalledTimes(1);
  });

  test("probeEndpointUrl: HEAD 网络错误时回退 GET", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
      updateProviderEndpointProbeSnapshot: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const fetchMock = vi.fn(async (_url: string, init?: RequestInit) => {
      if (init?.method === "HEAD") {
        throw new Error("boom");
      }
      if (init?.method === "GET") {
        return new Response(null, { status: 200 });
      }
      throw new Error("unexpected");
    });
    vi.stubGlobal("fetch", fetchMock);

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 1234);

    expect(result).toEqual(
      expect.objectContaining({ ok: true, method: "GET", statusCode: 200, errorType: null })
    );
    expect(fetchMock).toHaveBeenCalledTimes(2);
  });

  test("probeEndpointUrl: 5xx 返回 ok=false 且标注 http_5xx", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
      updateProviderEndpointProbeSnapshot: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 503 }))
    );

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 1234);

    expect(result.ok).toBe(false);
    expect(result.method).toBe("HEAD");
    expect(result.statusCode).toBe(503);
    expect(result.errorType).toBe("http_5xx");
    expect(result.errorMessage).toBe("HTTP 503");
  });

  test("probeEndpointUrl: 4xx 仍视为 ok=true", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
      updateProviderEndpointProbeSnapshot: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 404 }))
    );

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 1234);

    expect(result.ok).toBe(true);
    expect(result.statusCode).toBe(404);
    expect(result.errorType).toBeNull();
  });

  test("probeEndpointUrl: AbortError 归类为 timeout", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
      updateProviderEndpointProbeSnapshot: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const fetchMock = vi.fn(async () => {
      const err = new Error("");
      err.name = "AbortError";
      throw err;
    });
    vi.stubGlobal("fetch", fetchMock);

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 1);

    expect(result.ok).toBe(false);
    expect(result.method).toBe("GET");
    expect(result.statusCode).toBeNull();
    expect(result.errorType).toBe("timeout");
    expect(result.errorMessage).toBe("timeout");
  });

  test("probeProviderEndpointAndRecord: endpoint 不存在时返回 null", async () => {
    vi.resetModules();

    const recordMock = vi.fn(async () => {});
    const snapshotMock = vi.fn(async () => {});
    const findMock = vi.fn(async () => null);

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    const recordFailureMock = vi.fn(async () => {});

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: findMock,
      recordProviderEndpointProbeResult: recordMock,
      updateProviderEndpointProbeSnapshot: snapshotMock,
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () =>
      createCircuitBreakerMock({ recordEndpointFailure: recordFailureMock })
    );

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 200 }))
    );

    const { probeProviderEndpointAndRecord } = await import("@/lib/provider-endpoints/probe");
    const result = await probeProviderEndpointAndRecord({ endpointId: 123, source: "manual" });

    expect(result).toBeNull();
    expect(recordMock).not.toHaveBeenCalled();
    expect(snapshotMock).not.toHaveBeenCalled();
    expect(recordFailureMock).not.toHaveBeenCalled();
  });

  test("probeProviderEndpointAndRecord: 记录入库字段包含 source/ok/statusCode/latency/probedAt", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.useFakeTimers();
    vi.setSystemTime(new Date("2026-01-01T00:00:00.000Z"));

    vi.resetModules();

    const recordMock = vi.fn(async () => {});
    const snapshotMock = vi.fn(async () => {});
    const findMock = vi.fn(async () => makeEndpoint({ id: 123, url: "https://example.com" }));

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    const recordFailureMock = vi.fn(async () => {});

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: findMock,
      recordProviderEndpointProbeResult: recordMock,
      updateProviderEndpointProbeSnapshot: snapshotMock,
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () =>
      createCircuitBreakerMock({ recordEndpointFailure: recordFailureMock })
    );

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 200 }))
    );

    const { probeProviderEndpointAndRecord } = await import("@/lib/provider-endpoints/probe");
    const result = await probeProviderEndpointAndRecord({
      endpointId: 123,
      source: "manual",
      timeoutMs: 1111,
    });

    expect(result).toEqual(expect.objectContaining({ ok: true, statusCode: 200, errorType: null }));

    expect(recordMock).toHaveBeenCalledTimes(1);
    const payload = recordMock.mock.calls[0]?.[0];
    expect(payload).toEqual(
      expect.objectContaining({
        endpointId: 123,
        source: "manual",
        ok: true,
        statusCode: 200,
        errorType: null,
        errorMessage: null,
      })
    );

    const probedAt = (payload as { probedAt: Date }).probedAt;
    expect(probedAt).toBeInstanceOf(Date);
    expect(probedAt.toISOString()).toBe("2026-01-01T00:00:00.000Z");

    expect(snapshotMock).not.toHaveBeenCalled();
    expect(recordFailureMock).not.toHaveBeenCalled();
  });

  test("probeProviderEndpointAndRecord: scheduled 成功总是写入探测日志记录", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.useFakeTimers();
    vi.setSystemTime(new Date("2026-01-01T00:00:30.000Z"));

    vi.resetModules();

    const recordMock = vi.fn(async () => {});
    const recordFailureMock = vi.fn(async () => {});

    const endpoint = makeEndpoint({
      id: 1,
      url: "https://example.com",
      lastProbeOk: true,
      lastProbedAt: new Date("2026-01-01T00:00:00.000Z"),
    });

    vi.doMock("@/lib/logger", () => ({
      logger: {
        debug: vi.fn(),
        info: vi.fn(),
        warn: vi.fn(),
        trace: vi.fn(),
        error: vi.fn(),
        fatal: vi.fn(),
      },
    }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(async () => endpoint),
      recordProviderEndpointProbeResult: recordMock,
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () =>
      createCircuitBreakerMock({ recordEndpointFailure: recordFailureMock })
    );

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 200 }))
    );

    const { probeProviderEndpointAndRecord } = await import("@/lib/provider-endpoints/probe");
    const result = await probeProviderEndpointAndRecord({ endpointId: 1, source: "scheduled" });

    expect(result).toEqual(expect.objectContaining({ ok: true, statusCode: 200 }));
    expect(recordMock).toHaveBeenCalledTimes(1);
    expect(recordFailureMock).not.toHaveBeenCalled();
  });

  test("probeProviderEndpointAndRecord: 失败会计入端点熔断计数（scheduled 与 manual）", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "HEAD";
    vi.resetModules();

    const recordMock = vi.fn(async () => {});
    const recordFailureMock = vi.fn(async () => {});

    vi.doMock("@/lib/logger", () => ({
      logger: {
        debug: vi.fn(),
        info: vi.fn(),
        warn: vi.fn(),
        trace: vi.fn(),
        error: vi.fn(),
        fatal: vi.fn(),
      },
    }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(async () =>
        makeEndpoint({ id: 123, url: "https://example.com" })
      ),
      recordProviderEndpointProbeResult: recordMock,
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () =>
      createCircuitBreakerMock({ recordEndpointFailure: recordFailureMock })
    );

    vi.stubGlobal(
      "fetch",
      vi.fn(async () => new Response(null, { status: 503 }))
    );

    const { probeProviderEndpointAndRecord } = await import("@/lib/provider-endpoints/probe");

    await probeProviderEndpointAndRecord({ endpointId: 123, source: "scheduled" });
    await probeProviderEndpointAndRecord({ endpointId: 123, source: "manual" });

    expect(recordFailureMock).toHaveBeenCalledTimes(2);
    expect(recordMock).toHaveBeenCalledTimes(2);
  });

  test("probeEndpointUrl: TCP mode connects to host:port without HTTP request", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "TCP";
    vi.resetModules();

    const logger = {
      debug: vi.fn(),
      info: vi.fn(),
      warn: vi.fn(),
      trace: vi.fn(),
      error: vi.fn(),
      fatal: vi.fn(),
    };

    vi.doMock("@/lib/logger", () => ({ logger }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    // Mock net.createConnection to simulate successful TCP connection
    const mockSocket = {
      destroy: vi.fn(),
      on: vi.fn(),
    };

    vi.doMock("node:net", () => ({
      default: {
        createConnection: vi.fn((_opts: unknown, cb: () => void) => {
          // Simulate immediate successful connection
          setTimeout(() => cb(), 0);
          return mockSocket;
        }),
      },
    }));

    const fetchMock = vi.fn();
    vi.stubGlobal("fetch", fetchMock);

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://api.example.com:8443/v1", 5000);

    expect(result.ok).toBe(true);
    expect(result.method).toBe("TCP");
    expect(result.statusCode).toBeNull();
    expect(result.errorType).toBeNull();
    expect(result.latencyMs).toBeTypeOf("number");
    // fetch should never be called in TCP mode
    expect(fetchMock).not.toHaveBeenCalled();
  });

  test("probeEndpointUrl: TCP mode defaults to port 80 for http URLs", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "TCP";
    vi.resetModules();

    vi.doMock("@/lib/logger", () => ({
      logger: {
        debug: vi.fn(),
        info: vi.fn(),
        warn: vi.fn(),
        trace: vi.fn(),
        error: vi.fn(),
        fatal: vi.fn(),
      },
    }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const mockSocket = {
      destroy: vi.fn(),
      on: vi.fn(),
    };

    vi.doMock("node:net", () => ({
      default: {
        createConnection: vi.fn((_opts: unknown, cb: () => void) => {
          setTimeout(() => cb(), 0);
          return mockSocket;
        }),
      },
    }));

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("http://api.example.com/v1/messages", 5000);

    // TCP connection succeeds, no HTTP status code
    expect(result.ok).toBe(true);
    expect(result.method).toBe("TCP");
    expect(result.statusCode).toBeNull();
  });

  test("probeEndpointUrl: TCP mode returns invalid_url for bad URLs", async () => {
    process.env.ENDPOINT_PROBE_METHOD = "TCP";
    vi.resetModules();

    vi.doMock("@/lib/logger", () => ({
      logger: {
        debug: vi.fn(),
        info: vi.fn(),
        warn: vi.fn(),
        trace: vi.fn(),
        error: vi.fn(),
        fatal: vi.fn(),
      },
    }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("not-a-valid-url", 5000);

    expect(result.ok).toBe(false);
    expect(result.method).toBe("TCP");
    expect(result.errorType).toBe("invalid_url");
  });

  test("probeEndpointUrl: defaults to TCP when ENDPOINT_PROBE_METHOD is not set", async () => {
    delete process.env.ENDPOINT_PROBE_METHOD;
    vi.resetModules();

    vi.doMock("@/lib/logger", () => ({
      logger: {
        debug: vi.fn(),
        info: vi.fn(),
        warn: vi.fn(),
        trace: vi.fn(),
        error: vi.fn(),
        fatal: vi.fn(),
      },
    }));
    vi.doMock("@/repository", () => ({
      findProviderEndpointById: vi.fn(),
      recordProviderEndpointProbeResult: vi.fn(),
    }));
    vi.doMock("@/lib/endpoint-circuit-breaker", () => createCircuitBreakerMock());

    const mockSocket = {
      destroy: vi.fn(),
      on: vi.fn(),
    };

    vi.doMock("node:net", () => ({
      default: {
        createConnection: vi.fn((_opts: unknown, cb: () => void) => {
          setTimeout(() => cb(), 0);
          return mockSocket;
        }),
      },
    }));

    const fetchMock = vi.fn();
    vi.stubGlobal("fetch", fetchMock);

    const { probeEndpointUrl } = await import("@/lib/provider-endpoints/probe");
    const result = await probeEndpointUrl("https://example.com", 5000);

    expect(result.method).toBe("TCP");
    expect(fetchMock).not.toHaveBeenCalled();
  });
});