11// Test Live Shard tests cover test live shard script behavior.
2-import { spawnSync } from "node:child_process";
3-import { existsSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
2+import { spawn, spawnSync } from "node:child_process";
3+import { chmodSync, existsSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
44import { tmpdir } from "node:os";
55import path from "node:path";
6+import { setTimeout as delay } from "node:timers/promises";
67import { describe, expect, it } from "vitest";
78import {
89LIVE_TEST_SHARDS,
@@ -421,4 +422,119 @@ describe("scripts/test-live-shard", () => {
421422stdio: "inherit",
422423});
423424});
425+426+it.skipIf(process.platform === "win32")(
427+"cleans live shard descendants before forwarding parent SIGTERM",
428+async () => {
429+const root = mkdtempSync(path.join(tmpdir(), "openclaw-live-shard-signal-"));
430+const fakePnpmPath = path.join(root, "pnpm");
431+const childPidPath = path.join(root, "child.pid");
432+const descendantPidPath = path.join(root, "descendant.pid");
433+const signaledPath = path.join(root, "signaled");
434+let childPid = 0;
435+let descendantPid = 0;
436+let runner: ReturnType<typeof spawn> | undefined;
437+438+try {
439+writeFakePnpm(fakePnpmPath);
440+runner = spawn(
441+process.execPath,
442+["scripts/test-live-shard.mjs", "native-live-src-agents"],
443+{
444+env: {
445+ ...process.env,
446+OPENCLAW_FAKE_PNPM_DESCENDANT_PID_PATH: descendantPidPath,
447+OPENCLAW_FAKE_PNPM_PID_PATH: childPidPath,
448+OPENCLAW_FAKE_PNPM_SIGNALED_PATH: signaledPath,
449+npm_execpath: fakePnpmPath,
450+},
451+stdio: "ignore",
452+},
453+);
454+455+await waitFor(() => existsSync(childPidPath), 5_000);
456+await waitFor(() => existsSync(descendantPidPath), 5_000);
457+childPid = Number(readFileSync(childPidPath, "utf8"));
458+descendantPid = Number(readFileSync(descendantPidPath, "utf8"));
459+expect(Number.isInteger(childPid)).toBe(true);
460+expect(Number.isInteger(descendantPid)).toBe(true);
461+462+runner.kill("SIGTERM");
463+464+await expect(waitForClose(runner)).resolves.toEqual({ code: null, signal: "SIGTERM" });
465+await waitFor(() => existsSync(signaledPath), 5_000);
466+expect(readFileSync(signaledPath, "utf8")).toBe("SIGTERM");
467+await waitFor(() => !isProcessAlive(childPid), 5_000);
468+await waitFor(() => !isProcessAlive(descendantPid), 5_000);
469+} finally {
470+if (runner?.pid && isProcessAlive(runner.pid)) {
471+process.kill(runner.pid, "SIGKILL");
472+}
473+if (childPid && isProcessAlive(childPid)) {
474+process.kill(childPid, "SIGKILL");
475+}
476+if (descendantPid && isProcessAlive(descendantPid)) {
477+process.kill(descendantPid, "SIGKILL");
478+}
479+rmSync(root, { force: true, recursive: true });
480+}
481+},
482+);
424483});
484+485+function writeFakePnpm(filePath: string): void {
486+writeFileSync(
487+filePath,
488+[
489+"#!/usr/bin/env node",
490+'const { spawn } = require("node:child_process");',
491+'const fs = require("node:fs");',
492+"const child = spawn(process.execPath, [",
493+' "-e",',
494+" \"process.on('SIGTERM', () => {}); setInterval(() => {}, 1000);\",",
495+"], { stdio: 'ignore' });",
496+"fs.writeFileSync(process.env.OPENCLAW_FAKE_PNPM_DESCENDANT_PID_PATH, String(child.pid));",
497+"fs.writeFileSync(process.env.OPENCLAW_FAKE_PNPM_PID_PATH, String(process.pid));",
498+'process.on("SIGTERM", () => {',
499+' fs.writeFileSync(process.env.OPENCLAW_FAKE_PNPM_SIGNALED_PATH, "SIGTERM");',
500+" process.exit(0);",
501+"});",
502+"setInterval(() => {}, 1000);",
503+"",
504+].join("\n"),
505+);
506+chmodSync(filePath, 0o755);
507+}
508+509+async function waitFor(condition: () => boolean, timeoutMs: number): Promise<void> {
510+const startedAt = Date.now();
511+while (!condition()) {
512+if (Date.now() - startedAt > timeoutMs) {
513+throw new Error("timed out waiting for condition");
514+}
515+await delay(25);
516+}
517+}
518+519+async function waitForClose(
520+child: ReturnType<typeof spawn>,
521+timeoutMs = 5_000,
522+): Promise<{ code: number | null; signal: NodeJS.Signals | null }> {
523+return await Promise.race([
524+new Promise<{ code: number | null; signal: NodeJS.Signals | null }>((resolve) => {
525+child.once("close", (code, signal) => resolve({ code, signal }));
526+}),
527+delay(timeoutMs).then(() => {
528+throw new Error("timed out waiting for child close");
529+}),
530+]);
531+}
532+533+function isProcessAlive(pid: number): boolean {
534+try {
535+process.kill(pid, 0);
536+return true;
537+} catch {
538+return false;
539+}
540+}
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。