26 lines
1019 B
TypeScript
26 lines
1019 B
TypeScript
import { vi } from "vitest";
|
|
import { loadModelCatalog } from "../agents/model-catalog.js";
|
|
import { runEmbeddedPiAgent } from "../agents/pi-embedded.js";
|
|
import { runSubagentAnnounceFlow } from "../agents/subagent-announce.js";
|
|
import { telegramOutbound } from "../channels/plugins/outbound/telegram.js";
|
|
import { setActivePluginRegistry } from "../plugins/runtime.js";
|
|
import { createOutboundTestPlugin, createTestRegistry } from "../test-utils/channel-plugins.js";
|
|
|
|
export function setupIsolatedAgentTurnMocks(params?: { fast?: boolean }): void {
|
|
if (params?.fast) {
|
|
vi.stubEnv("OPENCLAW_TEST_FAST", "1");
|
|
}
|
|
vi.mocked(runEmbeddedPiAgent).mockReset();
|
|
vi.mocked(loadModelCatalog).mockResolvedValue([]);
|
|
vi.mocked(runSubagentAnnounceFlow).mockReset().mockResolvedValue(true);
|
|
setActivePluginRegistry(
|
|
createTestRegistry([
|
|
{
|
|
pluginId: "telegram",
|
|
plugin: createOutboundTestPlugin({ id: "telegram", outbound: telegramOutbound }),
|
|
source: "test",
|
|
},
|
|
]),
|
|
);
|
|
}
|