test: trim more agent tool partial mocks

This commit is contained in:
Peter Steinberger
2026-04-03 20:18:56 +01:00
parent 1fbf863f53
commit f5ec0e429f
2 changed files with 5 additions and 4 deletions

View File

@@ -20,8 +20,8 @@ import {
const completeMock = vi.hoisted(() => vi.fn()); const completeMock = vi.hoisted(() => vi.fn());
vi.mock("@mariozechner/pi-ai", async (importOriginal) => { vi.mock("@mariozechner/pi-ai", async () => {
const actual = await importOriginal<typeof import("@mariozechner/pi-ai")>(); const actual = await vi.importActual<typeof import("@mariozechner/pi-ai")>("@mariozechner/pi-ai");
return { return {
...actual, ...actual,
complete: completeMock, complete: completeMock,

View File

@@ -22,8 +22,9 @@ const loadConfigMock = vi.fn<() => SessionsToolTestConfig>(() => ({
tools: { agentToAgent: { enabled: false } }, tools: { agentToAgent: { enabled: false } },
})); }));
vi.mock("../../config/config.js", async (importOriginal) => { vi.mock("../../config/config.js", async () => {
const actual = await importOriginal<typeof import("../../config/config.js")>(); const actual =
await vi.importActual<typeof import("../../config/config.js")>("../../config/config.js");
return { return {
...actual, ...actual,
loadConfig: () => loadConfigMock() as never, loadConfig: () => loadConfigMock() as never,