openclaw/extensions/huggingface/models.proxy.test.ts

// Hugging Face proxy tests cover the live model discovery transport policy.
import { afterEach, describe, expect, it, vi } from "vitest";

const fetchWithSsrFGuardMock = vi.hoisted(() => vi.fn());

vi.mock("openclaw/plugin-sdk/ssrf-runtime", async (importOriginal) => ({
  ...(await importOriginal<typeof import("openclaw/plugin-sdk/ssrf-runtime")>()),
  fetchWithSsrFGuard: fetchWithSsrFGuardMock,
}));

import { discoverHuggingfaceModels } from "./models.js";

const ORIGINAL_VITEST = process.env.VITEST;
const ORIGINAL_NODE_ENV = process.env.NODE_ENV;

function restoreEnv(key: "VITEST" | "NODE_ENV", value: string | undefined) {
  if (value === undefined) {
    delete process.env[key];
  } else {
    process.env[key] = value;
  }
}

afterEach(() => {
  restoreEnv("VITEST", ORIGINAL_VITEST);
  restoreEnv("NODE_ENV", ORIGINAL_NODE_ENV);
  fetchWithSsrFGuardMock.mockReset();
  vi.restoreAllMocks();
});

describe("Hugging Face model discovery proxy policy", () => {
  it("allows the guarded official catalog request to use an eligible HTTP proxy", async () => {
    process.env.VITEST = "false";
    process.env.NODE_ENV = "development";
    const response = new Response("unavailable", { status: 503 });
    const release = vi.fn(async () => undefined);
    fetchWithSsrFGuardMock.mockResolvedValue({ response, release });

    await discoverHuggingfaceModels("hf_test_token");

    expect(fetchWithSsrFGuardMock).toHaveBeenCalledWith(
      expect.objectContaining({
        mode: "trusted_env_proxy",
        url: "https://router.huggingface.co/v1/models",
      }),
    );
    expect(release).toHaveBeenCalledTimes(1);
  });
});