fix: include wrapWebContent in external content mock

This commit is contained in:
Marc J Saint-jour 2026-03-12 19:26:15 -04:00
parent e2c2dcdc77
commit f7a08ecc96

View File

@ -64,7 +64,6 @@ vi.mock("../../agents/skills/refresh.js", () => ({
}));
vi.mock("../../agents/workspace.js", () => ({
DEFAULT_IDENTITY_FILENAME: "IDENTITY.md",
ensureAgentWorkspace: vi.fn().mockResolvedValue({ dir: "/tmp/workspace" }),
}));
@ -172,12 +171,18 @@ vi.mock("../../logger.js", () => ({
logWarn: (...args: unknown[]) => logWarnMock(...args),
}));
vi.mock("../../security/external-content.js", () => ({
buildSafeExternalPrompt: vi.fn().mockReturnValue("safe prompt"),
detectSuspiciousPatterns: vi.fn().mockReturnValue([]),
getHookType: vi.fn().mockReturnValue("unknown"),
isExternalHookSession: vi.fn().mockReturnValue(false),
}));
vi.mock("../../security/external-content.js", async () => {
const actual = await vi.importActual<typeof import("../../security/external-content.js")>(
"../../security/external-content.js",
);
return {
...actual,
buildSafeExternalPrompt: vi.fn().mockReturnValue("safe prompt"),
detectSuspiciousPatterns: vi.fn().mockReturnValue([]),
getHookType: vi.fn().mockReturnValue("unknown"),
isExternalHookSession: vi.fn().mockReturnValue(false),
};
});
vi.mock("../delivery.js", () => ({
resolveCronDeliveryPlan: resolveCronDeliveryPlanMock,