feat(tests): add isolated pi launchers

2026-08-07 09:55:02 +02:00
parent 690c640d5b
commit b100d553c1
9 changed files with 346 additions and 4 deletions
@@ -0,0 +1,96 @@
+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)
+  })
+})