diff options
Diffstat (limited to 'packages/provider-openai-compat/src/extension.test.ts')
| -rw-r--r-- | packages/provider-openai-compat/src/extension.test.ts | 140 |
1 files changed, 70 insertions, 70 deletions
diff --git a/packages/provider-openai-compat/src/extension.test.ts b/packages/provider-openai-compat/src/extension.test.ts index 6dc409a..653b825 100644 --- a/packages/provider-openai-compat/src/extension.test.ts +++ b/packages/provider-openai-compat/src/extension.test.ts @@ -3,97 +3,97 @@ import { describe, expect, it, vi } from "vitest"; import { activate, manifest } from "./extension.js"; function makeFakeHost(overrides: { - getAuthProvider?: (id: string) => AuthContract | undefined; - configGet?: (key: string) => unknown; + getAuthProvider?: (id: string) => AuthContract | undefined; + configGet?: (key: string) => unknown; }): { host: HostAPI; defineProvider: ReturnType<typeof vi.fn> } { - const defineProvider = vi.fn(); - const warn = vi.fn(); - const info = vi.fn(); - const host = { - defineProvider, - config: { get: overrides.configGet ?? (() => undefined) }, - logger: { debug: vi.fn(), info, warn, error: vi.fn() }, - getAuthProvider: overrides.getAuthProvider ?? (() => undefined), - } as unknown as HostAPI; - return { host, defineProvider }; + const defineProvider = vi.fn(); + const warn = vi.fn(); + const info = vi.fn(); + const host = { + defineProvider, + config: { get: overrides.configGet ?? (() => undefined) }, + logger: { debug: vi.fn(), info, warn, error: vi.fn() }, + getAuthProvider: overrides.getAuthProvider ?? (() => undefined), + } as unknown as HostAPI; + return { host, defineProvider }; } describe("provider-openai-compat activation", () => { - it("registers provider when auth resolves ApiKeyCredentials", async () => { - const fakeCreds: ApiKeyCredentials = { type: "api-key", apiKey: "sk-test" }; - const fakeAuth: AuthContract = { - id: "apikey", - resolve: async () => fakeCreds, - }; + it("registers provider when auth resolves ApiKeyCredentials", async () => { + const fakeCreds: ApiKeyCredentials = { type: "api-key", apiKey: "sk-test" }; + const fakeAuth: AuthContract = { + id: "apikey", + resolve: async () => fakeCreds, + }; - const { host, defineProvider } = makeFakeHost({ - getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), - }); + const { host, defineProvider } = makeFakeHost({ + getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), + }); - await activate(host); + await activate(host); - expect(defineProvider).toHaveBeenCalledTimes(1); - expect(defineProvider.mock.calls[0]?.[0]?.id).toBe("openai-compat"); - }); + expect(defineProvider).toHaveBeenCalledTimes(1); + expect(defineProvider.mock.calls[0]?.[0]?.id).toBe("openai-compat"); + }); - it("does not register provider when getAuthProvider returns undefined", async () => { - const { host, defineProvider } = makeFakeHost({ - getAuthProvider: () => undefined, - }); + it("does not register provider when getAuthProvider returns undefined", async () => { + const { host, defineProvider } = makeFakeHost({ + getAuthProvider: () => undefined, + }); - await activate(host); + await activate(host); - expect(defineProvider).not.toHaveBeenCalled(); - }); + expect(defineProvider).not.toHaveBeenCalled(); + }); - it("does not register provider when resolve returns null", async () => { - const fakeAuth: AuthContract = { - id: "apikey", - resolve: async () => null, - }; + it("does not register provider when resolve returns null", async () => { + const fakeAuth: AuthContract = { + id: "apikey", + resolve: async () => null, + }; - const { host, defineProvider } = makeFakeHost({ - getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), - }); + const { host, defineProvider } = makeFakeHost({ + getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), + }); - await activate(host); + await activate(host); - expect(defineProvider).not.toHaveBeenCalled(); - }); + expect(defineProvider).not.toHaveBeenCalled(); + }); - it("does not register provider when credentials are not api-key type", async () => { - const fakeAuth: AuthContract = { - id: "apikey", - resolve: async () => ({ type: "bearer-token", token: "tok-123" }), - }; + it("does not register provider when credentials are not api-key type", async () => { + const fakeAuth: AuthContract = { + id: "apikey", + resolve: async () => ({ type: "bearer-token", token: "tok-123" }), + }; - const { host, defineProvider } = makeFakeHost({ - getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), - }); + const { host, defineProvider } = makeFakeHost({ + getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), + }); - await activate(host); + await activate(host); - expect(defineProvider).not.toHaveBeenCalled(); - }); + expect(defineProvider).not.toHaveBeenCalled(); + }); - it("uses default model when config returns undefined", async () => { - const fakeCreds: ApiKeyCredentials = { type: "api-key", apiKey: "sk-test" }; - const fakeAuth: AuthContract = { - id: "apikey", - resolve: async () => fakeCreds, - }; + it("uses default model when config returns undefined", async () => { + const fakeCreds: ApiKeyCredentials = { type: "api-key", apiKey: "sk-test" }; + const fakeAuth: AuthContract = { + id: "apikey", + resolve: async () => fakeCreds, + }; - const { host, defineProvider } = makeFakeHost({ - getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), - configGet: () => undefined, - }); + const { host, defineProvider } = makeFakeHost({ + getAuthProvider: (id) => (id === "apikey" ? fakeAuth : undefined), + configGet: () => undefined, + }); - await activate(host); + await activate(host); - expect(defineProvider).toHaveBeenCalledTimes(1); - }); + expect(defineProvider).toHaveBeenCalledTimes(1); + }); - it("declares dependsOn auth-apikey", () => { - expect(manifest.dependsOn).toEqual(["auth-apikey"]); - }); + it("declares dependsOn auth-apikey", () => { + expect(manifest.dependsOn).toEqual(["auth-apikey"]); + }); }); |