97 lines
3.2 KiB
JavaScript
97 lines
3.2 KiB
JavaScript
import assert from "node:assert/strict"
|
|
import { spawnSync } from "node:child_process"
|
|
import { existsSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs"
|
|
import { tmpdir } from "node:os"
|
|
import { delimiter, dirname, join, resolve } from "node:path"
|
|
import { fileURLToPath } from "node:url"
|
|
import { describe, it } from "node:test"
|
|
|
|
const repoRoot = resolve(dirname(fileURLToPath(import.meta.url)), "..")
|
|
const launcher = join(repoRoot, "scripts", "pi-isolated.mjs")
|
|
|
|
function runLauncher({ exitStatus = 0 } = {}) {
|
|
const fakeBin = mkdtempSync(join(tmpdir(), "pi-commandcode-fake-bin-"))
|
|
const logPath = join(fakeBin, "calls.jsonl")
|
|
const fakePi = join(fakeBin, "pi")
|
|
|
|
writeFileSync(
|
|
fakePi,
|
|
`#!/bin/sh
|
|
node - "$@" <<'NODE'
|
|
const { appendFileSync } = require("node:fs")
|
|
appendFileSync(process.env.FAKE_PI_LOG, JSON.stringify({
|
|
args: process.argv.slice(2),
|
|
agentDir: process.env.PI_CODING_AGENT_DIR,
|
|
sessionDir: process.env.PI_CODING_AGENT_SESSION_DIR,
|
|
skipVersionCheck: process.env.PI_SKIP_VERSION_CHECK,
|
|
home: process.env.HOME,
|
|
userProfile: process.env.USERPROFILE,
|
|
inheritedApiKey: process.env.COMMANDCODE_API_KEY ?? null,
|
|
}) + "\\n")
|
|
NODE
|
|
if [ "$1" = "install" ]; then exit 0; fi
|
|
exit ${exitStatus}
|
|
`,
|
|
{ mode: 0o700 },
|
|
)
|
|
|
|
try {
|
|
const result = spawnSync(process.execPath, [launcher, "--model", "claude-sonnet-5"], {
|
|
cwd: repoRoot,
|
|
env: {
|
|
...process.env,
|
|
PATH: `${fakeBin}${delimiter}${process.env.PATH ?? ""}`,
|
|
FAKE_PI_LOG: logPath,
|
|
COMMANDCODE_API_KEY: "must-not-leak",
|
|
},
|
|
encoding: "utf8",
|
|
})
|
|
const calls = readFileSync(logPath, "utf8")
|
|
.trim()
|
|
.split("\n")
|
|
.map((line) => JSON.parse(line))
|
|
return { result, calls }
|
|
} finally {
|
|
rmSync(fakeBin, { recursive: true, force: true })
|
|
}
|
|
}
|
|
|
|
describe("isolated pi launcher", () => {
|
|
it("installs the current checkout, forwards arguments, and removes its environment", () => {
|
|
const { result, calls } = runLauncher()
|
|
|
|
assert.equal(result.status, 0)
|
|
assert.equal(calls.length, 2)
|
|
assert.deepEqual(calls[0].args, ["install", repoRoot, "--no-approve"])
|
|
assert.deepEqual(calls[1].args, [
|
|
"--no-approve",
|
|
"--provider",
|
|
"commandcode",
|
|
"--model",
|
|
"gpt-5.6-luna",
|
|
"--model",
|
|
"claude-sonnet-5",
|
|
])
|
|
|
|
const [install, launch] = calls
|
|
assert.equal(install.agentDir, launch.agentDir)
|
|
assert.equal(install.sessionDir, launch.sessionDir)
|
|
assert.equal(launch.skipVersionCheck, "1")
|
|
assert.equal(launch.inheritedApiKey, null)
|
|
assert.ok(launch.agentDir.includes("pi-commandcode-isolated-"))
|
|
assert.equal(launch.home, dirname(launch.agentDir))
|
|
assert.equal(launch.userProfile, dirname(launch.agentDir))
|
|
assert.equal(dirname(launch.agentDir), dirname(launch.sessionDir))
|
|
assert.equal(existsSync(dirname(launch.agentDir)), false)
|
|
assert.match(result.stderr, /Removed the isolated pi environment/)
|
|
})
|
|
|
|
it("returns the pi exit status and still removes its environment", () => {
|
|
const { result, calls } = runLauncher({ exitStatus: 7 })
|
|
|
|
assert.equal(result.status, 7)
|
|
assert.equal(calls.length, 2)
|
|
assert.equal(existsSync(dirname(calls[1].agentDir)), false)
|
|
})
|
|
})
|