File size: 4,492 Bytes
e249c6d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | import { afterEach, describe, expect, it, vi } from "vitest";
import {
getInstallationTarget,
withInstallationTarget,
} from "../../infra/installation-target-context.js";
import { withEnvAsync } from "../../test-utils/env.js";
import { buildPreparedCliRunContext } from "../cli-runner.test-helpers.js";
import { executePreparedCliRun as executePreparedCliRunImpl } from "./execute.js";
import {
createManagedRun,
createSuccessfulProcessExit,
supervisorSpawnMock,
wrapPreparedCliRunWithTestAdmission,
} from "./execute.test-support.js";
const executePreparedCliRun = wrapPreparedCliRunWithTestAdmission(executePreparedCliRunImpl);
afterEach(() => supervisorSpawnMock.mockReset());
describe("CLI installation target", () => {
it.each(["process", "plugin", "node"] as const)(
"projects local child environment and fences %s placement",
async (kind) => {
const target = {
stateDir: "/fixture/diagnosed",
configPath: "/fixture/custom.json",
defaultWorkspaceDir: "/fixture/default-workspace",
};
const context = buildPreparedCliRunContext({
model: "fixture-model",
backend: {
command: "/bin/sh",
args: [],
output: "text",
systemPromptFileArg: undefined,
input: "stdin",
},
});
let childEnv: NodeJS.ProcessEnv | undefined;
const pluginExecute = vi.fn(async function* (execution: { env: NodeJS.ProcessEnv }) {
childEnv = execution.env;
yield { type: "result", subtype: "success", result: "done" };
});
if (kind === "plugin") {
context.executionTarget = { kind, execute: pluginExecute };
context.preparedBackend.backend.output = "jsonl";
context.preparedBackend.backend.jsonlDialect = "claude-stream-json";
} else if (kind === "node") {
context.executionTarget = { kind, placement: { nodeId: "fixture-node" } };
}
supervisorSpawnMock.mockResolvedValue(
createManagedRun({
...createSuccessfulProcessExit(),
durationMs: 1,
stdout: "done",
}),
);
await withEnvAsync(
{
OPENCLAW_STATE_DIR: "/fixture/scratch",
OPENCLAW_CONFIG_PATH: undefined,
OPENCLAW_WORKSPACE_DIR: "/fixture/execution-cwd",
},
async () => {
const run = withInstallationTarget(target, () => executePreparedCliRun(context));
expect(getInstallationTarget()).toBeUndefined();
if (kind === "node") {
await expect(run).rejects.toThrow("saved prompt");
expect(supervisorSpawnMock).not.toHaveBeenCalled();
expect(pluginExecute).not.toHaveBeenCalled();
return;
}
await expect(run).resolves.toMatchObject({ text: "done" });
const expectedEnv = {
OPENCLAW_STATE_DIR: target.stateDir,
OPENCLAW_CONFIG_PATH: target.configPath,
OPENCLAW_WORKSPACE_DIR: target.defaultWorkspaceDir,
};
if (kind === "process") {
expect(supervisorSpawnMock).toHaveBeenLastCalledWith(
expect.objectContaining({ env: expect.objectContaining(expectedEnv) }),
);
} else {
expect(childEnv).toMatchObject(expectedEnv);
}
expect(process.env.OPENCLAW_STATE_DIR).toBe("/fixture/scratch");
expect(process.env.OPENCLAW_CONFIG_PATH).toBeUndefined();
expect(process.env.OPENCLAW_WORKSPACE_DIR).toBe("/fixture/execution-cwd");
await executePreparedCliRun(context);
if (kind === "process") {
expect(supervisorSpawnMock).toHaveBeenLastCalledWith(
expect.objectContaining({
env: expect.objectContaining({
OPENCLAW_STATE_DIR: "/fixture/scratch",
OPENCLAW_WORKSPACE_DIR: "/fixture/execution-cwd",
}),
}),
);
expect(supervisorSpawnMock).not.toHaveBeenLastCalledWith(
expect.objectContaining({
env: expect.objectContaining({ OPENCLAW_CONFIG_PATH: expect.anything() }),
}),
);
} else {
expect(childEnv?.OPENCLAW_STATE_DIR).toBe("/fixture/scratch");
expect(childEnv?.OPENCLAW_CONFIG_PATH).toBeUndefined();
expect(childEnv?.OPENCLAW_WORKSPACE_DIR).toBe("/fixture/execution-cwd");
}
},
);
},
);
});
|