From ca66f1426aa5b9e5a249c6d8eeff1043d81d88cd Mon Sep 17 00:00:00 2001 From: Shakker Date: Mon, 11 May 2026 03:16:36 +0100 Subject: [PATCH] test: tighten vercel catalog assertions --- .../provider-catalog.test.ts | 39 +++++++++---------- 1 file changed, 19 insertions(+), 20 deletions(-) diff --git a/extensions/vercel-ai-gateway/provider-catalog.test.ts b/extensions/vercel-ai-gateway/provider-catalog.test.ts index d51c8c2a58c..7cc09bb06b9 100644 --- a/extensions/vercel-ai-gateway/provider-catalog.test.ts +++ b/extensions/vercel-ai-gateway/provider-catalog.test.ts @@ -5,36 +5,35 @@ import { buildVercelAiGatewayProvider, } from "./provider-catalog.js"; +const STATIC_MODEL_IDS = [ + "anthropic/claude-opus-4.6", + "openai/gpt-5.4", + "openai/gpt-5.4-pro", + "moonshotai/kimi-k2.6", +]; + describe("vercel ai gateway provider catalog", () => { it("builds the bundled Vercel AI Gateway defaults", async () => { const provider = await buildVercelAiGatewayProvider(); - expect(provider.baseUrl).toBe(VERCEL_AI_GATEWAY_BASE_URL); - expect(provider.api).toBe("anthropic-messages"); - expect(provider.models?.map((model) => model.id)).toEqual( - expect.arrayContaining([ - "anthropic/claude-opus-4.6", - "openai/gpt-5.4", - "openai/gpt-5.4-pro", - "moonshotai/kimi-k2.6", - ]), - ); + expect(provider).toStrictEqual({ + baseUrl: VERCEL_AI_GATEWAY_BASE_URL, + api: "anthropic-messages", + models: getStaticVercelAiGatewayModelCatalog(), + }); }); it("exposes the static fallback model catalog", () => { - expect(getStaticVercelAiGatewayModelCatalog().map((model) => model.id)).toEqual( - expect.arrayContaining([ - "anthropic/claude-opus-4.6", - "openai/gpt-5.4", - "openai/gpt-5.4-pro", - "moonshotai/kimi-k2.6", - ]), + expect(getStaticVercelAiGatewayModelCatalog().map((model) => model.id)).toStrictEqual( + STATIC_MODEL_IDS, ); }); it("builds an offline static provider catalog", () => { - expect(buildStaticVercelAiGatewayProvider().models?.map((model) => model.id)).toEqual( - expect.arrayContaining(["moonshotai/kimi-k2.6"]), - ); + expect(buildStaticVercelAiGatewayProvider()).toStrictEqual({ + baseUrl: VERCEL_AI_GATEWAY_BASE_URL, + api: "anthropic-messages", + models: getStaticVercelAiGatewayModelCatalog(), + }); }); });