68 lines
3.5 KiB
TypeScript
68 lines
3.5 KiB
TypeScript
import { http, HttpResponse } from "msw";
|
|
import { server } from "../test/msw";
|
|
import {
|
|
asPiManagementApiError,
|
|
getPiManagementLogs,
|
|
getPiManagementOptions,
|
|
getPiManagementStatus,
|
|
runPiManagementTest,
|
|
savePiManagementConfig,
|
|
} from "./pi-management";
|
|
|
|
test("Pi management client calls only the sanctioned sanitized endpoints", async () => {
|
|
const calls: Array<{ method: string; path: string; body?: unknown }> = [];
|
|
server.use(
|
|
http.get("/api/pi-management/status", ({ request }) => {
|
|
calls.push({ method: request.method, path: new URL(request.url).pathname });
|
|
return HttpResponse.json({ ready: true, version: "0.80.3", credentials: "present", config: {}, checkedAt: "2026-08-05T10:00:00.000Z" });
|
|
}),
|
|
http.get("/api/pi-management/options", ({ request }) => {
|
|
calls.push({ method: request.method, path: new URL(request.url).pathname });
|
|
return HttpResponse.json({ providers: ["zai"], models: [{ provider: "zai", id: "glm-5.2" }], reasoning: ["low", "medium", "high"], checkedAt: "2026-08-05T10:00:00.000Z" });
|
|
}),
|
|
http.put("/api/pi-management/config", async ({ request }) => {
|
|
calls.push({ method: request.method, path: new URL(request.url).pathname, body: await request.json() });
|
|
return HttpResponse.json({ provider: "zai", model: "glm-5.2", reasoning: "high", updatedAt: "2026-08-05T10:00:00.000Z" });
|
|
}),
|
|
http.post("/api/pi-management/test", ({ request }) => {
|
|
calls.push({ method: request.method, path: new URL(request.url).pathname });
|
|
return HttpResponse.json({ ready: true, checkedAt: "2026-08-05T10:00:00.000Z" });
|
|
}),
|
|
http.get("/api/pi-management/logs", ({ request }) => {
|
|
calls.push({ method: request.method, path: new URL(request.url).pathname });
|
|
return HttpResponse.json({ lines: ["Pi smoke check succeeded"], checkedAt: "2026-08-05T10:00:00.000Z" });
|
|
}),
|
|
);
|
|
|
|
await expect(getPiManagementStatus()).resolves.toMatchObject({ version: "0.80.3", ready: true, credentials: "present" });
|
|
await expect(getPiManagementOptions()).resolves.toMatchObject({ providers: ["zai"] });
|
|
await expect(savePiManagementConfig({ provider: "zai", model: "glm-5.2", reasoning: "high" })).resolves.toMatchObject({ reasoning: "high" });
|
|
await expect(runPiManagementTest()).resolves.toMatchObject({ ready: true });
|
|
await expect(getPiManagementLogs()).resolves.toMatchObject({ lines: ["Pi smoke check succeeded"] });
|
|
|
|
expect(calls).toEqual([
|
|
{ method: "GET", path: "/api/pi-management/status" },
|
|
{ method: "GET", path: "/api/pi-management/options" },
|
|
{ method: "PUT", path: "/api/pi-management/config", body: { provider: "zai", model: "glm-5.2", reasoning: "high" } },
|
|
{ method: "POST", path: "/api/pi-management/test" },
|
|
{ method: "GET", path: "/api/pi-management/logs" },
|
|
]);
|
|
expect(calls.some((call) => /update|terminal|shell/i.test(call.path))).toBe(false);
|
|
});
|
|
|
|
test("Pi management client exposes the stable forbidden message without raw response text", async () => {
|
|
server.use(http.get("/api/pi-management/status", () =>
|
|
HttpResponse.json({ code: "pi_management_forbidden", error: "Pi management is not permitted", raw: "token=do-not-show" }, { status: 403 }),
|
|
));
|
|
|
|
await expect(getPiManagementStatus()).rejects.toSatisfy((error: unknown) => {
|
|
expect(asPiManagementApiError(error)).toEqual({
|
|
status: 403,
|
|
code: "pi_management_forbidden",
|
|
message: "Pi management is not permitted",
|
|
});
|
|
expect(asPiManagementApiError(error)?.message).not.toContain("token");
|
|
return true;
|
|
});
|
|
});
|