| import { describe, expect, it } from "vitest"; |
| import { createComputeClient } from "@homepilot/compute-client"; |
| import type { ApiClient } from "@homepilot/api-client"; |
|
|
| |
| |
| |
| |
| |
|
|
| type Call = { method: string; path: string; body?: unknown }; |
|
|
| function fakeApi(responses: Record<string, unknown>): { api: ApiClient; calls: Call[] } { |
| const calls: Call[] = []; |
| const reply = (path: string) => responses[path] ?? {}; |
| const api: ApiClient = { |
| baseUrl: "http://test", |
| get: async (p) => { calls.push({ method: "GET", path: p }); return reply(p) as never; }, |
| post: async (p, b) => { calls.push({ method: "POST", path: p, body: b }); return reply(p) as never; }, |
| put: async (p, b) => { calls.push({ method: "PUT", path: p, body: b }); return reply(p) as never; }, |
| del: async (p) => { calls.push({ method: "DELETE", path: p }); return {} as never; }, |
| }; |
| return { api, calls }; |
| } |
|
|
| describe("compute-client admin", () => { |
| it("lists and camelizes sources", async () => { |
| const { api } = fakeApi({ |
| "/compute/admin/sources": { sources: [{ id: "s", name: "S", kind: "openai_compatible", base_url: "http://x", credential_ref: "r", enabled: true, execution_type: "direct" }] }, |
| }); |
| const sources = await createComputeClient(api).listComputeSources(); |
| expect(sources[0].baseUrl).toBe("http://x"); |
| expect(sources[0].credentialRef).toBe("r"); |
| expect(sources[0].executionType).toBe("direct"); |
| }); |
|
|
| it("sends per-model routes as snake_case", async () => { |
| const { api, calls } = fakeApi({ "/compute/admin/routes": { route: {} } }); |
| await createComputeClient(api).upsertRoute({ |
| modality: "chat", modelId: "m", primaryTarget: "device:colab", |
| fallbackTargets: ["local"], strategy: "pinned", privacy: "private_nodes", |
| }); |
| const body = calls.find((c) => c.method === "POST")!.body as Record<string, unknown>; |
| expect(body.model_id).toBe("m"); |
| expect(body.primary_target).toBe("device:colab"); |
| expect(body.fallback_targets).toEqual(["local"]); |
| }); |
|
|
| it("recursively snake-cases nested modality defaults in settings", async () => { |
| const { api, calls } = fakeApi({ "/compute/admin/settings": { settings: {} } }); |
| await createComputeClient(api).putComputeSettings({ |
| computeMode: "auto", |
| modalityDefaults: { |
| image: { modality: "image", strategy: "prefer_remote", fallbackTargets: ["local"], privacy: "private_nodes" }, |
| }, |
| }); |
| const body = calls.find((c) => c.method === "PUT")!.body as Record<string, unknown>; |
| expect(body.compute_mode).toBe("auto"); |
| const md = body.modality_defaults as Record<string, Record<string, unknown>>; |
| expect(md.image.fallback_targets).toEqual(["local"]); |
| }); |
|
|
| it("resolveRoute passes model_id and returns the plan", async () => { |
| const { api, calls } = fakeApi({ |
| "/compute/admin/resolve?modality=chat&model_id=m": { modality: "chat", model_id: "m", matched: "route", compute_mode: "local", targets: ["device:colab", "local"] }, |
| }); |
| const plan = await createComputeClient(api).resolveRoute("chat", "m"); |
| expect(plan.matched).toBe("route"); |
| expect(plan.targets).toEqual(["device:colab", "local"]); |
| expect(calls[0].path).toContain("model_id=m"); |
| }); |
|
|
| it("tests a source and reads local resources", async () => { |
| const { api, calls } = fakeApi({ |
| "/compute/admin/sources/oa/test": { ok: true, reason: "Reachable." }, |
| "/v1/system/resources": { gpus: [{ available: true, name: "RTX 4090", vram_total_mb: 24576, used_percent: 30 }], cpu: { percent: 12 }, ram: { percent: 40 } }, |
| }); |
| const client = createComputeClient(api); |
| const t = await client.testComputeSource("oa"); |
| expect(t.ok).toBe(true); |
| expect(calls[0]).toMatchObject({ method: "POST", path: "/compute/admin/sources/oa/test" }); |
| const res = await client.getLocalResources(); |
| expect(res.gpus[0].vramTotalMb).toBe(24576); |
| expect(res.gpus[0].usedPercent).toBe(30); |
| }); |
|
|
| it("syncs from cloud via POST /compute/admin/sync", async () => { |
| const { api, calls } = fakeApi({ |
| "/compute/admin/sync": { ok: true, linked: true, devices: 2, models: 5, source_id: "ollabridge-cloud" }, |
| }); |
| const r = await createComputeClient(api).syncFromCloud(); |
| expect(r).toMatchObject({ linked: true, devices: 2, models: 5 }); |
| expect(calls[0]).toMatchObject({ method: "POST", path: "/compute/admin/sync" }); |
| }); |
|
|
| it("deletes sources and routes at the right paths", async () => { |
| const { api, calls } = fakeApi({}); |
| const client = createComputeClient(api); |
| await client.deleteComputeSource("s 1"); |
| await client.deleteRoute("chat", "m/x"); |
| expect(calls[0]).toMatchObject({ method: "DELETE", path: "/compute/admin/sources/s%201" }); |
| expect(calls[1]).toMatchObject({ method: "DELETE", path: "/compute/admin/routes/chat/m%2Fx" }); |
| }); |
| }); |
|
|
