openclaw/src/auto-reply/reply.directive.directive-b...

18 lines
681 B
TypeScript

import { vi, type Mock } from "vitest";
export const runEmbeddedPiAgentMock: Mock = vi.fn();
export const loadModelCatalogMock: Mock = vi.fn();
vi.mock("../agents/pi-embedded.js", () => ({
abortEmbeddedPiRun: vi.fn().mockReturnValue(false),
runEmbeddedPiAgent: (...args: unknown[]) => runEmbeddedPiAgentMock(...args),
queueEmbeddedPiMessage: vi.fn().mockReturnValue(false),
resolveEmbeddedSessionLane: (key: string) => `session:${key.trim() || "main"}`,
isEmbeddedPiRunActive: vi.fn().mockReturnValue(false),
isEmbeddedPiRunStreaming: vi.fn().mockReturnValue(false),
}));
vi.mock("../agents/model-catalog.js", () => ({
loadModelCatalog: loadModelCatalogMock,
}));