101 lines
3.2 KiB
TypeScript
101 lines
3.2 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import type { ExecApprovalsResolved } from "../infra/exec-approvals.js";
|
|
import { sanitizeBinaryOutput } from "./shell-utils.js";
|
|
|
|
const isWin = process.platform === "win32";
|
|
|
|
vi.mock("../infra/shell-env.js", async (importOriginal) => {
|
|
const mod = await importOriginal<typeof import("../infra/shell-env.js")>();
|
|
return {
|
|
...mod,
|
|
getShellPathFromLoginShell: vi.fn(() => "/custom/bin:/opt/bin"),
|
|
resolveShellEnvFallbackTimeoutMs: vi.fn(() => 1234),
|
|
};
|
|
});
|
|
|
|
vi.mock("../infra/exec-approvals.js", async (importOriginal) => {
|
|
const mod = await importOriginal<typeof import("../infra/exec-approvals.js")>();
|
|
const approvals: ExecApprovalsResolved = {
|
|
path: "/tmp/exec-approvals.json",
|
|
socketPath: "/tmp/exec-approvals.sock",
|
|
token: "token",
|
|
defaults: {
|
|
security: "full",
|
|
ask: "off",
|
|
askFallback: "full",
|
|
autoAllowSkills: false,
|
|
},
|
|
agent: {
|
|
security: "full",
|
|
ask: "off",
|
|
askFallback: "full",
|
|
autoAllowSkills: false,
|
|
},
|
|
allowlist: [],
|
|
file: {
|
|
version: 1,
|
|
socket: { path: "/tmp/exec-approvals.sock", token: "token" },
|
|
defaults: {
|
|
security: "full",
|
|
ask: "off",
|
|
askFallback: "full",
|
|
autoAllowSkills: false,
|
|
},
|
|
agents: {},
|
|
},
|
|
};
|
|
return { ...mod, resolveExecApprovals: () => approvals };
|
|
});
|
|
|
|
const normalizeText = (value?: string) =>
|
|
sanitizeBinaryOutput(value ?? "")
|
|
.replace(/\r\n/g, "\n")
|
|
.replace(/\r/g, "\n")
|
|
.trim();
|
|
|
|
describe("exec PATH login shell merge", () => {
|
|
const originalPath = process.env.PATH;
|
|
|
|
afterEach(() => {
|
|
process.env.PATH = originalPath;
|
|
});
|
|
|
|
it("merges login-shell PATH for host=gateway", async () => {
|
|
if (isWin) return;
|
|
process.env.PATH = "/usr/bin";
|
|
|
|
const { createExecTool } = await import("./bash-tools.exec.js");
|
|
const { getShellPathFromLoginShell } = await import("../infra/shell-env.js");
|
|
const shellPathMock = vi.mocked(getShellPathFromLoginShell);
|
|
shellPathMock.mockClear();
|
|
shellPathMock.mockReturnValue("/custom/bin:/opt/bin");
|
|
|
|
const tool = createExecTool({ host: "gateway", security: "full", ask: "off" });
|
|
const result = await tool.execute("call1", { command: "echo $PATH" });
|
|
const text = normalizeText(result.content.find((c) => c.type === "text")?.text);
|
|
|
|
expect(text).toBe("/custom/bin:/opt/bin:/usr/bin");
|
|
expect(shellPathMock).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it("skips login-shell PATH when env.PATH is provided", async () => {
|
|
if (isWin) return;
|
|
process.env.PATH = "/usr/bin";
|
|
|
|
const { createExecTool } = await import("./bash-tools.exec.js");
|
|
const { getShellPathFromLoginShell } = await import("../infra/shell-env.js");
|
|
const shellPathMock = vi.mocked(getShellPathFromLoginShell);
|
|
shellPathMock.mockClear();
|
|
|
|
const tool = createExecTool({ host: "gateway", security: "full", ask: "off" });
|
|
const result = await tool.execute("call1", {
|
|
command: "echo $PATH",
|
|
env: { PATH: "/explicit/bin" },
|
|
});
|
|
const text = normalizeText(result.content.find((c) => c.type === "text")?.text);
|
|
|
|
expect(text).toBe("/explicit/bin");
|
|
expect(shellPathMock).not.toHaveBeenCalled();
|
|
});
|
|
});
|