openclaw/test/scripts/e2e-temp-state-dir.test.ts

import { spawn } from "node:child_process";
import { existsSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import path from "node:path";
import { setTimeout as delay } from "node:timers/promises";
import { pathToFileURL } from "node:url";
import { describe, expect, it } from "vitest";
import { createE2eStateDir } from "../../scripts/e2e/lib/temp-state-dir.ts";

async function waitForFile(filePath: string) {
  for (let attempt = 0; attempt < 100; attempt += 1) {
    if (existsSync(filePath)) {
      return;
    }
    await delay(20);
  }
  throw new Error(`Timed out waiting for ${filePath}`);
}

function waitForExit(child: ReturnType<typeof spawn>) {
  return new Promise<{ code: number | null; signal: NodeJS.Signals | null }>((resolve, reject) => {
    child.once("error", reject);
    child.once("exit", (code, signal) => resolve({ code, signal }));
  });
}

describe("E2E temp state dirs", () => {
  it("cleans generated state dirs", async () => {
    const state = await createE2eStateDir("openclaw-e2e-temp-state-test-", {
      OPENCLAW_STATE_DIR: "",
    });

    expect(state.created).toBe(true);
    expect(existsSync(state.stateDir)).toBe(true);
    state.cleanup();
    expect(existsSync(state.stateDir)).toBe(false);
  });

  it("leaves caller-provided state dirs alone", async () => {
    const root = mkdtempSync(path.join(tmpdir(), "openclaw-e2e-temp-state-existing-"));
    try {
      const state = await createE2eStateDir("openclaw-e2e-temp-state-test-", {
        OPENCLAW_STATE_DIR: root,
      });

      expect(state).toMatchObject({ created: false, stateDir: root });
      state.cleanup();
      expect(existsSync(root)).toBe(true);
    } finally {
      rmSync(root, { force: true, recursive: true });
    }
  });

  it("cleans generated state dirs on termination signals", async () => {
    const root = mkdtempSync(path.join(tmpdir(), "openclaw-e2e-temp-state-signal-"));
    try {
      const statePathFile = path.join(root, "state-path");
      const scriptPath = path.join(root, "probe.mjs");
      const helperUrl = pathToFileURL(path.resolve("scripts/e2e/lib/temp-state-dir.ts")).href;
      writeFileSync(
        scriptPath,
        `import { writeFileSync } from "node:fs";
import { createE2eStateDir } from ${JSON.stringify(helperUrl)};

const state = await createE2eStateDir("openclaw-e2e-temp-state-signal-", {
  OPENCLAW_STATE_DIR: "",
});
state.registerExitCleanup();
writeFileSync(${JSON.stringify(statePathFile)}, state.stateDir);
setInterval(() => {}, 1000);
`,
      );

      const child = spawn(process.execPath, [scriptPath], {
        stdio: "ignore",
      });
      try {
        await waitForFile(statePathFile);
        const stateDir = readFileSync(statePathFile, "utf8").trim();
        expect(existsSync(stateDir)).toBe(true);

        child.kill("SIGTERM");
        const exit = await waitForExit(child);
        expect(exit).toEqual({ code: null, signal: "SIGTERM" });
        expect(existsSync(stateDir)).toBe(false);
      } finally {
        if (child.exitCode === null && child.signalCode === null) {
          child.kill("SIGKILL");
        }
      }
    } finally {
      rmSync(root, { force: true, recursive: true });
    }
  });
});