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
No known key found for this signature in database
2 changed files with 5 additions and 4 deletions

View File

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

View File

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