File size: 2,559 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
const mocks = vi.hoisted(() => ({
getProviderConnectionById: vi.fn(),
getApiKeys: vi.fn(),
getConsistentMachineId: vi.fn(),
}));
vi.mock("@/lib/localDb", () => ({
getProviderConnectionById: mocks.getProviderConnectionById,
getApiKeys: mocks.getApiKeys,
}));
vi.mock("@/shared/utils/machineId", () => ({
getConsistentMachineId: mocks.getConsistentMachineId,
}));
vi.mock("next/server", () => ({
NextResponse: {
json(body, init = {}) {
return new Response(JSON.stringify(body), {
status: init.status || 200,
headers: { "Content-Type": "application/json" },
});
},
},
}));
const originalFetch = global.fetch;
describe("provider test-models route kind routing", () => {
beforeEach(() => {
vi.clearAllMocks();
mocks.getProviderConnectionById.mockResolvedValue({
id: "conn-hf",
provider: "huggingface",
});
mocks.getApiKeys.mockResolvedValue([{ key: "sk-internal", isActive: true }]);
mocks.getConsistentMachineId.mockResolvedValue("cli-token");
global.fetch = vi.fn((url) => {
if (String(url).includes("/api/v1/images/generations")) {
return Promise.resolve(new Response(JSON.stringify({
created: 1,
data: [{ b64_json: "abc" }],
}), {
status: 200,
headers: { "Content-Type": "application/json" },
}));
}
return Promise.resolve(new Response(JSON.stringify({
choices: [{ message: { role: "assistant", content: "ok" } }],
}), {
status: 200,
headers: { "Content-Type": "application/json" },
}));
});
});
afterEach(() => {
global.fetch = originalFetch;
});
it("routes huggingface image models to /api/v1/images/generations", async () => {
const { POST } = await import("../../src/app/api/providers/[id]/test-models/route.js");
const req = new Request("http://localhost/api/providers/conn-hf/test-models", {
method: "POST",
headers: { "Content-Type": "application/json" },
});
const res = await POST(req, { params: Promise.resolve({ id: "conn-hf" }) });
const body = await res.json();
expect(body.provider).toBe("huggingface");
expect(body.results.some((r) => r.modelId === "black-forest-labs/FLUX.1-schnell" && r.ok)).toBe(true);
expect(global.fetch).toHaveBeenCalledWith(
expect.stringContaining("/api/v1/images/generations"),
expect.objectContaining({
method: "POST",
})
);
});
});
|