test: trim more agent partial mocks

This commit is contained in:
Peter Steinberger 2026-04-03 20:06:23 +01:00
parent 5f17362667
commit 2d2fe2bf47
No known key found for this signature in database
4 changed files with 14 additions and 8 deletions

View File

@ -11,8 +11,10 @@ const mocks = vi.hoisted(() => ({
operations: undefined as CapturedEditOperations | undefined,
}));
vi.mock("@mariozechner/pi-coding-agent", async (importOriginal) => {
const actual = await importOriginal<typeof import("@mariozechner/pi-coding-agent")>();
vi.mock("@mariozechner/pi-coding-agent", async () => {
const actual = await vi.importActual<typeof import("@mariozechner/pi-coding-agent")>(
"@mariozechner/pi-coding-agent",
);
return {
...actual,
createEditTool: (_cwd: string, options?: { operations?: CapturedEditOperations }) => {

View File

@ -8,8 +8,9 @@ import { createHostSandboxFsBridge } from "./test-helpers/host-sandbox-fs-bridge
import { expectReadWriteEditTools, getTextContent } from "./test-helpers/pi-tools-fs-helpers.js";
import { createPiToolsSandboxContext } from "./test-helpers/pi-tools-sandbox-context.js";
vi.mock("../infra/shell-env.js", async (importOriginal) => {
const mod = await importOriginal<typeof import("../infra/shell-env.js")>();
vi.mock("../infra/shell-env.js", async () => {
const mod =
await vi.importActual<typeof import("../infra/shell-env.js")>("../infra/shell-env.js");
return { ...mod, getShellPathFromLoginShell: () => null };
});
async function withTempDir<T>(prefix: string, fn: (dir: string) => Promise<T>) {

View File

@ -10,8 +10,9 @@ let cleanStaleLockFiles: typeof import("./session-write-lock.js").cleanStaleLock
let resetSessionWriteLockStateForTest: typeof import("./session-write-lock.js").resetSessionWriteLockStateForTest;
let resolveSessionLockMaxHoldFromTimeout: typeof import("./session-write-lock.js").resolveSessionLockMaxHoldFromTimeout;
vi.mock("../shared/pid-alive.js", async (importOriginal) => {
const original = await importOriginal<typeof import("../shared/pid-alive.js")>();
vi.mock("../shared/pid-alive.js", async () => {
const original =
await vi.importActual<typeof import("../shared/pid-alive.js")>("../shared/pid-alive.js");
return {
...original,
// Keep liveness checks real; only pin process start time for PID recycle coverage.

View File

@ -19,8 +19,10 @@ vi.mock("../process/exec.js", () => ({
runCommandWithTimeout: (...args: unknown[]) => runCommandWithTimeoutMock(...args),
}));
vi.mock("../security/skill-scanner.js", async (importOriginal) => ({
...(await importOriginal<typeof import("../security/skill-scanner.js")>()),
vi.mock("../security/skill-scanner.js", async () => ({
...(await vi.importActual<typeof import("../security/skill-scanner.js")>(
"../security/skill-scanner.js",
)),
scanDirectoryWithSummary: (...args: unknown[]) => scanDirectoryWithSummaryMock(...args),
}));