mirror of https://github.com/openclaw/openclaw.git
81 lines
2.2 KiB
TypeScript
81 lines
2.2 KiB
TypeScript
import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
const { createPtyAdapterMock } = vi.hoisted(() => ({
|
|
createPtyAdapterMock: vi.fn(),
|
|
}));
|
|
|
|
vi.mock("../../agents/shell-utils.js", () => ({
|
|
getShellConfig: () => ({ shell: "sh", args: ["-c"] }),
|
|
}));
|
|
|
|
vi.mock("./adapters/pty.js", () => ({
|
|
createPtyAdapter: (...args: unknown[]) => createPtyAdapterMock(...args),
|
|
}));
|
|
|
|
function createStubPtyAdapter() {
|
|
return {
|
|
pid: 1234,
|
|
stdin: undefined,
|
|
onStdout: (_listener: (chunk: string) => void) => {
|
|
// no-op
|
|
},
|
|
onStderr: (_listener: (chunk: string) => void) => {
|
|
// no-op
|
|
},
|
|
wait: async () => ({ code: 0, signal: null }),
|
|
kill: (_signal?: NodeJS.Signals) => {
|
|
// no-op
|
|
},
|
|
dispose: () => {
|
|
// no-op
|
|
},
|
|
};
|
|
}
|
|
|
|
describe("process supervisor PTY command contract", () => {
|
|
let createProcessSupervisor: typeof import("./supervisor.js").createProcessSupervisor;
|
|
|
|
beforeAll(async () => {
|
|
({ createProcessSupervisor } = await import("./supervisor.js"));
|
|
});
|
|
|
|
beforeEach(() => {
|
|
createPtyAdapterMock.mockClear();
|
|
});
|
|
|
|
it("passes PTY command verbatim to shell args", async () => {
|
|
createPtyAdapterMock.mockResolvedValue(createStubPtyAdapter());
|
|
const supervisor = createProcessSupervisor();
|
|
const command = `printf '%s\\n' "a b" && printf '%s\\n' '$HOME'`;
|
|
|
|
const run = await supervisor.spawn({
|
|
sessionId: "s1",
|
|
backendId: "test",
|
|
mode: "pty",
|
|
ptyCommand: command,
|
|
timeoutMs: 1_000,
|
|
});
|
|
const exit = await run.wait();
|
|
|
|
expect(exit.reason).toBe("exit");
|
|
expect(createPtyAdapterMock).toHaveBeenCalledTimes(1);
|
|
const params = createPtyAdapterMock.mock.calls[0]?.[0] as { args?: string[] };
|
|
expect(params.args).toEqual(["-c", command]);
|
|
});
|
|
|
|
it("rejects empty PTY command", async () => {
|
|
createPtyAdapterMock.mockResolvedValue(createStubPtyAdapter());
|
|
const supervisor = createProcessSupervisor();
|
|
|
|
await expect(
|
|
supervisor.spawn({
|
|
sessionId: "s1",
|
|
backendId: "test",
|
|
mode: "pty",
|
|
ptyCommand: " ",
|
|
}),
|
|
).rejects.toThrow("PTY command cannot be empty");
|
|
expect(createPtyAdapterMock).not.toHaveBeenCalled();
|
|
});
|
|
});
|