From c7ab83e8ab4945b16cb5c51d243f16b73f9ef44a Mon Sep 17 00:00:00 2001 From: Shakker Date: Sat, 9 May 2026 21:37:19 +0100 Subject: [PATCH] test: speed up cron model override tests --- .../isolated-agent.model-overrides.test.ts | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/src/cron/isolated-agent.model-overrides.test.ts b/src/cron/isolated-agent.model-overrides.test.ts index 1f7f87c7171..eb78871e35f 100644 --- a/src/cron/isolated-agent.model-overrides.test.ts +++ b/src/cron/isolated-agent.model-overrides.test.ts @@ -2,6 +2,10 @@ import "./isolated-agent.mocks.js"; import { beforeEach, describe, expect, it, vi } from "vitest"; import { loadModelCatalog } from "../agents/model-catalog.js"; import { runEmbeddedPiAgent } from "../agents/pi-embedded.js"; +import { BASE_THINKING_LEVELS } from "../auto-reply/thinking.shared.js"; +import type { PluginProviderRegistration } from "../plugins/registry.js"; +import { resetPluginRuntimeStateForTest, setActivePluginRegistry } from "../plugins/runtime.js"; +import { createTestRegistry } from "../test-utils/channel-plugins.js"; import { DEFAULT_AGENT_TURN_PAYLOAD, DEFAULT_MESSAGE, @@ -14,8 +18,30 @@ import { } from "./isolated-agent.turn-test-helpers.js"; import * as isolatedAgentRunRuntime from "./isolated-agent/run.runtime.js"; +function installThinkingTestProviders() { + const registry = createTestRegistry(); + registry.providers = ["anthropic", "openai", "openrouter"].map( + (providerId): PluginProviderRegistration => ({ + pluginId: providerId, + source: "test", + provider: { + id: providerId, + label: providerId, + auth: [], + resolveThinkingProfile: () => ({ + levels: BASE_THINKING_LEVELS.map((id) => ({ id })), + defaultLevel: "off", + }), + }, + }), + ); + setActivePluginRegistry(registry); +} + describe("runCronIsolatedAgentTurn model overrides", () => { beforeEach(() => { + resetPluginRuntimeStateForTest(); + installThinkingTestProviders(); vi.spyOn(isolatedAgentRunRuntime, "resolveThinkingDefault").mockReturnValue("off"); vi.mocked(runEmbeddedPiAgent).mockClear(); vi.mocked(loadModelCatalog).mockResolvedValue([]);