diff options
| author | Adam Malczewski <[email protected]> | 2026-06-05 21:20:12 +0900 |
|---|---|---|
| committer | Adam Malczewski <[email protected]> | 2026-06-05 21:20:12 +0900 |
| commit | 7fb3269c698ae583ea7997ce206c4ae252fd3218 (patch) | |
| tree | 247d03408ecccd633290ea56b1b08811ebe460ec /packages/credential-store/src/registry.test.ts | |
| parent | 4283d1f8a0bc3953e65962a2364c903d0015f047 (diff) | |
| download | dispatch-7fb3269c698ae583ea7997ce206c4ae252fd3218.tar.gz dispatch-7fb3269c698ae583ea7997ce206c4ae252fd3218.zip | |
feat(backend): credential-store + model selection/catalog (GET /models) + per-turn cwd through orchestrator/transport/host-bin
Diffstat (limited to 'packages/credential-store/src/registry.test.ts')
| -rw-r--r-- | packages/credential-store/src/registry.test.ts | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/packages/credential-store/src/registry.test.ts b/packages/credential-store/src/registry.test.ts new file mode 100644 index 0000000..e67b5d4 --- /dev/null +++ b/packages/credential-store/src/registry.test.ts @@ -0,0 +1,81 @@ +import type { ProviderContract } from "@dispatch/kernel"; +import { describe, expect, it } from "vitest"; +import { createCredentialStore } from "./registry.js"; + +function fakeProvider( + id: string, + listModels?: () => Promise<readonly { id: string }[]>, +): ProviderContract { + return { + id, + stream: async function* () {}, + ...(listModels ? { listModels } : {}), + }; +} + +describe("createCredentialStore", () => { + const credentials = [{ name: "opencode", providerId: "openai-compat" }]; + const providers = new Map<string, ProviderContract>(); + + const store = createCredentialStore({ + credentials, + getProvider: (id) => providers.get(id), + }); + + describe("resolve", () => { + it("resolves a simple model name", () => { + providers.set("openai-compat", fakeProvider("openai-compat")); + const result = store.resolve("opencode/deepseek-v4-flash"); + expect(result).toEqual({ providerId: "openai-compat", model: "deepseek-v4-flash" }); + }); + + it("resolves a model name with slashes", () => { + const result = store.resolve("opencode/a/b"); + expect(result).toEqual({ providerId: "openai-compat", model: "a/b" }); + }); + + it("returns undefined for unknown credential name", () => { + const result = store.resolve("unknown/x"); + expect(result).toBeUndefined(); + }); + + it("returns undefined for no slash", () => { + const result = store.resolve("noslash"); + expect(result).toBeUndefined(); + }); + + it("returns undefined for empty model segment", () => { + const result = store.resolve("opencode/"); + expect(result).toBeUndefined(); + }); + }); + + describe("listCatalog", () => { + it("lists models from providers with listModels", async () => { + providers.set( + "openai-compat", + fakeProvider("openai-compat", async () => [{ id: "m1" }, { id: "m2" }]), + ); + + const catalog = await store.listCatalog(); + expect(catalog).toEqual(["opencode/m1", "opencode/m2"]); + }); + + it("skips credentials whose provider has no listModels", async () => { + providers.set("openai-compat", fakeProvider("openai-compat")); + + const catalog = await store.listCatalog(); + expect(catalog).toEqual([]); + }); + + it("skips credentials whose provider is missing", async () => { + const emptyStore = createCredentialStore({ + credentials: [{ name: "missing", providerId: "nonexistent" }], + getProvider: () => undefined, + }); + + const catalog = await emptyStore.listCatalog(); + expect(catalog).toEqual([]); + }); + }); +}); |
