Plugins: fix catalog contract mocks

This commit is contained in:
Vincent Koc 2026-03-16 18:02:46 -07:00
parent 4194bba575
commit 0a93e22b37
1 changed files with 8 additions and 0 deletions

View File

@ -11,16 +11,20 @@ function uniqueProviders() {
const resolvePluginProvidersMock = vi.fn();
const resolveOwningPluginIdsForProviderMock = vi.fn();
const resolveNonBundledProviderPluginIdsMock = vi.fn();
vi.mock("../providers.js", () => ({
resolvePluginProviders: (...args: unknown[]) => resolvePluginProvidersMock(...args),
resolveOwningPluginIdsForProvider: (...args: unknown[]) =>
resolveOwningPluginIdsForProviderMock(...args),
resolveNonBundledProviderPluginIds: (...args: unknown[]) =>
resolveNonBundledProviderPluginIdsMock(...args),
}));
const {
augmentModelCatalogWithProviderPlugins,
buildProviderMissingAuthMessageWithPlugin,
resetProviderRuntimeHookCacheForTest,
resolveProviderBuiltInModelSuppression,
} = await import("../provider-runtime.js");
@ -28,10 +32,14 @@ describe("provider catalog contract", () => {
beforeEach(() => {
const providers = uniqueProviders();
const providerIds = [...new Set(providerContractRegistry.map((entry) => entry.pluginId))];
resetProviderRuntimeHookCacheForTest();
resolveOwningPluginIdsForProviderMock.mockReset();
resolveOwningPluginIdsForProviderMock.mockReturnValue(providerIds);
resolveNonBundledProviderPluginIdsMock.mockReset();
resolveNonBundledProviderPluginIdsMock.mockReturnValue([]);
resolvePluginProvidersMock.mockReset();
resolvePluginProvidersMock.mockImplementation((params?: { onlyPluginIds?: string[] }) => {
const onlyPluginIds = params?.onlyPluginIds;