@botbuddy/cli 1.5.0 → 1.5.1

package/src/run.test.mjs

@@ -1,173 +0,0 @@
-import test from "node:test";
-import assert from "node:assert/strict";
-import { mkdtempSync, readFileSync, renameSync, unlinkSync } from "node:fs";
-import { join } from "node:path";
-import { tmpdir } from "node:os";
-
-import { EventEmitter } from "node:events";
-import { EXIT, parseRunArgs, terminalStatus, commandHash, contextPath, launchRun, retryTerminalUpdate, runWorker, cancelRun } from "./run.mjs";
-
-test("run: rejects an unowned workload before it can start", () => {
-  const parsed = parseRunArgs(["--environment", "local", "--", "node", "-e", "process.exit(0)"]);
-  assert.ok(parsed.errors.some((e) => e.includes("session-id")));
-  assert.equal(parsed.command[0], "node");
-});
-
-test("run: accepts and forwards a structured rerun reason", async () => {
-  const parsed = parseRunArgs(["--session-id", "session-1", "--environment", "local", "--rerun-reason", '{"code":"source_changed","detail":"retry after fix"}', "--", "node", "-e", "process.exit(0)"]);
-  assert.deepEqual(parsed.opts.rerunReason, { code: "source_changed", detail: "retry after fix" });
-
-  const calls = [];
-  const result = await launchRun(["--session-id", "session-1", "--environment", "local", "--rerun-reason", '{"code":"source_changed","detail":"retry after fix"}', "--", "node", "-e", "process.exit(0)"], {
-    call: async (name, args) => {
-      calls.push({ name, args });
-      return { ok: true, data: { accepted: true, run_credential: "x".repeat(64), receipt_upload: "receipt://tenant/session/run" } };
-    },
-    spawnImpl: () => Object.assign(new EventEmitter(), { unref() {} }),
-  });
-  assert.equal(result.exitCode, EXIT.OK);
-  assert.deepEqual(calls[0].args.rerun_reason, { code: "source_changed", detail: "retry after fix" });
-  unlinkSync(contextPath(result.receipt.run_id));
-});
-
-test("run: rejects an invalid rerun reason before registering a run", () => {
-  const parsed = parseRunArgs(["--session-id", "session-1", "--environment", "local", "--rerun-reason", "not-json", "--", "node", "-e", "process.exit(0)"]);
-  assert.ok(parsed.errors.some((error) => error.includes("rerun-reason")));
-});
-
-test("run: derives complete terminal outcomes", () => {
-  assert.equal(terminalStatus({ exitCode: 0, signal: null }), "success");
-  assert.equal(terminalStatus({ exitCode: 1, signal: null }), "failure");
-  assert.equal(terminalStatus({ exitCode: null, signal: "SIGTERM" }), "cancelled");
-  assert.equal(terminalStatus({ exitCode: null, signal: "SIGTERM", timedOut: true }), "timed_out");
-  assert.equal(terminalStatus({ exitCode: null, signal: null, ownerLost: true }), "owner_lost");
-});
-
-test("run: the detached worker records one receipt after its launcher is gone", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-"));
-  const receiptPath = join(dir, "receipt.json");
-  const contextPath = join(dir, "context.json");
-  const updates = [];
-  const context = {
-    context_path: contextPath, run_id: "run-survives-launcher", run_credential: "x".repeat(64), receipt_upload: "receipt://botbuddy/session/run-survives-launcher",
-    command: [process.execPath, "-e", "setTimeout(() => process.exit(0), 30)"], cwd: dir, timeout_seconds: 2, receipt_path: receiptPath,
-  };
-  await import("node:fs/promises").then(({ writeFile }) => writeFile(contextPath, JSON.stringify(context)));
-  await runWorker(contextPath, { call: async (name, args) => { updates.push({ name, args }); return { ok: true, data: {} }; } });
-  const receipt = JSON.parse(readFileSync(receiptPath, "utf8"));
-  assert.equal(receipt.status, "success");
-  assert.equal(updates.length, 1);
-  assert.equal(updates[0].name, "update_command_run");
-  assert.equal(updates[0].args.status, "succeeded");
-  assert.equal(updates[0].args.run_id, context.run_id);
-});
-
-test("run: terminal callbacks retain and retry their sealed context until accepted", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-retry-"));
-  const context = { context_path: join(dir, "context.json") };
-  const update = { run_id: "retry-run", run_credential: "x".repeat(64), status: "succeeded" };
-  const attempts = [];
-  await retryTerminalUpdate(context, update, async () => {
-    attempts.push(true);
-    return attempts.length === 1 ? { ok: false, error: "offline" } : { ok: true, data: { accepted: true } };
-  }, { sleep: async () => {} });
-  assert.equal(attempts.length, 2);
-  const retained = JSON.parse(readFileSync(context.context_path, "utf8"));
-  assert.equal(retained.terminal_update.attempts, 1);
-});
-
-test("run: bounds captured output while preserving its total size", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-output-"));
-  const receiptPath = join(dir, "receipt.json");
-  const contextPath = join(dir, "context.json");
-  const context = {
-    context_path: contextPath, run_id: "bounded-output", run_credential: "x".repeat(64), receipt_upload: "receipt://botbuddy/session/bounded-output",
-    command: [process.execPath, "-e", "process.stdout.write('x'.repeat(50000))"], cwd: dir, timeout_seconds: 2, receipt_path: receiptPath,
-  };
-  await import("node:fs/promises").then(({ writeFile }) => writeFile(contextPath, JSON.stringify(context)));
-  await runWorker(contextPath, { call: async () => ({ ok: true, data: { accepted: true } }) });
-  const receipt = JSON.parse(readFileSync(receiptPath, "utf8"));
-  assert.equal(receipt.output_characters, 50000);
-  assert.ok(receipt.bounded_log.length <= 8192);
-});
-
-test("run: escalates a timed-out SIGTERM-resistant process group to SIGKILL", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-timeout-"));
-  const receiptPath = join(dir, "receipt.json");
-  const contextPath = join(dir, "context.json");
-  const context = {
-    context_path: contextPath, run_id: "timeout-escalation", run_credential: "x".repeat(64), receipt_upload: "receipt://botbuddy/session/timeout-escalation",
-    command: [process.execPath, "-e", "process.on('SIGTERM', () => {}); setInterval(() => {}, 1_000)"], cwd: dir, timeout_seconds: 0.03, timeout_grace_ms: 20, receipt_path: receiptPath,
-  };
-  await import("node:fs/promises").then(({ writeFile }) => writeFile(contextPath, JSON.stringify(context)));
-  await runWorker(contextPath, { call: async () => ({ ok: true, data: { accepted: true } }) });
-  const receipt = JSON.parse(readFileSync(receiptPath, "utf8"));
-  assert.equal(receipt.status, "timed_out");
-});
-
-test("run: asynchronous workload spawn errors become terminal owner-loss receipts", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-spawn-error-"));
-  const receiptPath = join(dir, "receipt.json");
-  const contextPath = join(dir, "context.json");
-  const context = {
-    context_path: contextPath, run_id: "async-spawn-error", run_credential: "x".repeat(64), receipt_upload: "receipt://botbuddy/session/async-spawn-error",
-    command: ["missing"], cwd: dir, timeout_seconds: 2, receipt_path: receiptPath,
-  };
-  await import("node:fs/promises").then(({ writeFile }) => writeFile(contextPath, JSON.stringify(context)));
-  const child = Object.assign(new EventEmitter(), { pid: 123, stdout: null, stderr: null });
-  // Emit the spawn error only AFTER runWorker has synchronously attached its
-  // 'error' listener (waitForChild runs before the first await). A fixed timer
-  // could otherwise fire while runWorker is still in its initial `await
-  // readFile`, making 'error' an unhandled event that throws under load.
-  await runWorker(contextPath, {
-    spawnImpl: () => { queueMicrotask(() => child.emit("error", new Error("ENOENT"))); return child; },
-    call: async () => ({ ok: true, data: { accepted: true } }),
-  });
-  const receipt = JSON.parse(readFileSync(receiptPath, "utf8"));
-  assert.equal(receipt.status, "owner_lost");
-  assert.match(receipt.reason, /ENOENT/);
-});
-
-test("run: command hashes are stable and command-sensitive", () => {
-  assert.equal(commandHash(["pnpm", "test"]), commandHash(["pnpm", "test"]));
-  assert.notEqual(commandHash(["pnpm", "test"]), commandHash(["pnpm", "test:e2e"]));
-  assert.equal(EXIT.BACKEND, 5);
-});
-
-test("run: cancellation terminates the detached workload process group and records one cancelled receipt", async () => {
-  const dir = mkdtempSync(join(tmpdir(), "bb-run-cancel-"));
-  const runId = `cancel-${Date.now()}`;
-  const receiptPath = join(dir, "receipt.json");
-  const sourceContext = join(dir, "context.json");
-  const updates = [];
-  const canonical = join(process.env.HOME ?? "", ".botbuddy", "command-runs", `${runId}.context.json`);
-  const context = {
-    context_path: canonical, run_id: runId, run_credential: "x".repeat(64), receipt_upload: `receipt://botbuddy/session/${runId}`,
-    command: [process.execPath, "-e", "setTimeout(() => process.exit(0), 10_000)"], cwd: dir, timeout_seconds: 20, receipt_path: receiptPath,
-  };
-  await import("node:fs/promises").then(({ writeFile }) => writeFile(sourceContext, JSON.stringify(context)));
-  // cancelRun intentionally uses the canonical per-user runner directory; link
-  // this deterministic fixture into it without giving the test a live server.
-  await import("node:fs/promises").then(({ mkdir }) => mkdir(join(process.env.HOME ?? "", ".botbuddy", "command-runs"), { recursive: true }));
-  renameSync(sourceContext, canonical);
-  const worker = runWorker(canonical, { call: async (name, args) => { updates.push({ name, args }); return { ok: true, data: {} }; } });
-  for (let attempt = 0; attempt < 20; attempt++) {
-    // The worker rewrites this file to stamp workload_pid; tolerate a transient
-    // read (an unfinished write, though writeJson is atomic) by retrying.
-    let current;
-    try {
-      current = JSON.parse(readFileSync(canonical, "utf8"));
-    } catch {
-      await new Promise((resolve) => setTimeout(resolve, 25));
-      continue;
-    }
-    if (Number.isInteger(current.workload_pid)) break;
-    await new Promise((resolve) => setTimeout(resolve, 25));
-  }
-  const cancelled = await cancelRun(runId, { call: async () => ({ ok: true, data: {} }) });
-  assert.equal(cancelled.receipt.outcome, "cancellation_requested", JSON.stringify(cancelled.receipt));
-  await worker;
-  const receipt = JSON.parse(readFileSync(receiptPath, "utf8"));
-  assert.equal(receipt.status, "cancelled");
-  assert.equal(updates.filter((u) => u.name === "update_command_run").length, 1);
-});