|
1 | 1 | import { beforeEach, describe, expect, it, vi } from "vitest"; |
2 | 2 |
|
3 | 3 | const mocks = vi.hoisted(() => ({ |
4 | | - resolveManifestBuiltInModelSuppression: vi.fn(), |
5 | 4 | buildManifestBuiltInModelSuppressionResolver: vi.fn(), |
| 5 | + resolveManifestBuiltInModelSuppression: vi.fn(), |
6 | 6 | })); |
7 | 7 |
|
8 | 8 | vi.mock("../plugins/manifest-model-suppression.js", () => ({ |
9 | | - resolveManifestBuiltInModelSuppression: mocks.resolveManifestBuiltInModelSuppression, |
10 | 9 | buildManifestBuiltInModelSuppressionResolver: mocks.buildManifestBuiltInModelSuppressionResolver, |
| 10 | + resolveManifestBuiltInModelSuppression: mocks.resolveManifestBuiltInModelSuppression, |
11 | 11 | })); |
12 | 12 |
|
13 | | -import { buildShouldSuppressBuiltInModel, shouldSuppressBuiltInModel } from "./model-suppression.js"; |
| 13 | +import { |
| 14 | + buildShouldSuppressBuiltInModel, |
| 15 | + shouldSuppressBuiltInModel, |
| 16 | +} from "./model-suppression.js"; |
14 | 17 |
|
15 | 18 | describe("model suppression", () => { |
16 | 19 | beforeEach(() => { |
| 20 | + mocks.buildManifestBuiltInModelSuppressionResolver.mockReset(); |
17 | 21 | mocks.resolveManifestBuiltInModelSuppression.mockReset(); |
18 | 22 | }); |
19 | 23 |
|
@@ -51,18 +55,42 @@ describe("model suppression", () => { |
51 | 55 | mocks.buildManifestBuiltInModelSuppressionResolver.mockReset(); |
52 | 56 | }); |
53 | 57 |
|
54 | | - it("normalizes provider aliases before checking suppressions", () => { |
55 | | - const resolverMock = vi.fn().mockReturnValue({ suppress: true }); |
56 | | - mocks.buildManifestBuiltInModelSuppressionResolver.mockReturnValueOnce(resolverMock); |
| 58 | + it("creates a reusable manifest resolver with normalized provider and model ids", () => { |
| 59 | + const resolver = vi |
| 60 | + .fn() |
| 61 | + .mockReturnValueOnce({ suppress: true, errorMessage: "manifest suppression" }) |
| 62 | + .mockReturnValueOnce(undefined); |
| 63 | + const config = {}; |
| 64 | + mocks.buildManifestBuiltInModelSuppressionResolver.mockReturnValueOnce(resolver); |
| 65 | + |
| 66 | + const shouldSuppress = buildShouldSuppressBuiltInModel({ config }); |
| 67 | + |
| 68 | + expect(shouldSuppress({ provider: "bedrock", id: "Claude-3" })).toBe(true); |
| 69 | + expect(shouldSuppress({ provider: "aws-bedrock", id: "claude-4" })).toBe(false); |
| 70 | + expect(mocks.buildManifestBuiltInModelSuppressionResolver).toHaveBeenCalledOnce(); |
| 71 | + expect(mocks.buildManifestBuiltInModelSuppressionResolver).toHaveBeenCalledWith({ |
| 72 | + config, |
| 73 | + env: process.env, |
| 74 | + }); |
| 75 | + expect(resolver).toHaveBeenNthCalledWith(1, { |
| 76 | + provider: "amazon-bedrock", |
| 77 | + id: "claude-3", |
| 78 | + }); |
| 79 | + expect(resolver).toHaveBeenNthCalledWith(2, { |
| 80 | + provider: "amazon-bedrock", |
| 81 | + id: "claude-4", |
| 82 | + }); |
| 83 | + }); |
| 84 | + |
| 85 | + it("does not call the manifest resolver for empty provider or model ids", () => { |
| 86 | + const resolver = vi.fn(); |
| 87 | + mocks.buildManifestBuiltInModelSuppressionResolver.mockReturnValueOnce(resolver); |
57 | 88 |
|
58 | | - const predicate = buildShouldSuppressBuiltInModel({ config: {} }); |
59 | | - |
60 | | - expect(predicate({ provider: "bedrock", id: "anthropic.claude-3-5-sonnet" })).toBe(true); |
| 89 | + const shouldSuppress = buildShouldSuppressBuiltInModel({}); |
61 | 90 |
|
62 | | - expect(resolverMock).toHaveBeenCalledOnce(); |
63 | | - expect(resolverMock).toHaveBeenCalledWith( |
64 | | - expect.objectContaining({ provider: "amazon-bedrock", id: "anthropic.claude-3-5-sonnet" }) |
65 | | - ); |
| 91 | + expect(shouldSuppress({ provider: "openai", id: "" })).toBe(false); |
| 92 | + expect(shouldSuppress({ provider: "", id: "gpt-5.5" })).toBe(false); |
| 93 | + expect(resolver).not.toHaveBeenCalled(); |
66 | 94 | }); |
67 | 95 | }); |
68 | 96 | }); |
0 commit comments