refactor(test): centralize temporary state-dir env setup

src/test-helpers/state-dir-env.test.ts

@@ -0,0 +1,45 @@
+import fs from "node:fs/promises";
+import path from "node:path";
+import { describe, expect, it } from "vitest";
+import {
+  restoreStateDirEnv,
+  setStateDirEnv,
+  snapshotStateDirEnv,
+  withStateDirEnv,
+} from "./state-dir-env.js";
+
+describe("state-dir-env helpers", () => {
+  it("set/snapshot/restore round-trips OPENCLAW_STATE_DIR", () => {
+    const prevOpenClaw = process.env.OPENCLAW_STATE_DIR;
+    const prevLegacy = process.env.CLAWDBOT_STATE_DIR;
+    const snapshot = snapshotStateDirEnv();
+
+    setStateDirEnv("/tmp/openclaw-state-dir-test");
+    expect(process.env.OPENCLAW_STATE_DIR).toBe("/tmp/openclaw-state-dir-test");
+    expect(process.env.CLAWDBOT_STATE_DIR).toBeUndefined();
+
+    restoreStateDirEnv(snapshot);
+    expect(process.env.OPENCLAW_STATE_DIR).toBe(prevOpenClaw);
+    expect(process.env.CLAWDBOT_STATE_DIR).toBe(prevLegacy);
+  });
+
+  it("withStateDirEnv sets env for callback and cleans up temp root", async () => {
+    const prevOpenClaw = process.env.OPENCLAW_STATE_DIR;
+    const prevLegacy = process.env.CLAWDBOT_STATE_DIR;
+
+    let capturedTempRoot = "";
+    let capturedStateDir = "";
+    await withStateDirEnv("openclaw-state-dir-env-", async ({ tempRoot, stateDir }) => {
+      capturedTempRoot = tempRoot;
+      capturedStateDir = stateDir;
+      expect(process.env.OPENCLAW_STATE_DIR).toBe(stateDir);
+      expect(process.env.CLAWDBOT_STATE_DIR).toBeUndefined();
+      await fs.writeFile(path.join(stateDir, "probe.txt"), "ok", "utf8");
+    });
+
+    expect(process.env.OPENCLAW_STATE_DIR).toBe(prevOpenClaw);
+    expect(process.env.CLAWDBOT_STATE_DIR).toBe(prevLegacy);
+    await expect(fs.stat(capturedStateDir)).rejects.toThrow();
+    await expect(fs.stat(capturedTempRoot)).rejects.toThrow();
+  });
+});

src/test-helpers/state-dir-env.ts

@@ -1,3 +1,6 @@
+import fs from "node:fs/promises";
+import os from "node:os";
+import path from "node:path";
 import { captureEnv } from "../test-utils/env.js";
 
 export function snapshotStateDirEnv() {
@@ -12,3 +15,20 @@ export function setStateDirEnv(stateDir: string): void {
   process.env.OPENCLAW_STATE_DIR = stateDir;
   delete process.env.CLAWDBOT_STATE_DIR;
 }
+
+export async function withStateDirEnv<T>(
+  prefix: string,
+  fn: (ctx: { tempRoot: string; stateDir: string }) => Promise<T>,
+): Promise<T> {
+  const snapshot = snapshotStateDirEnv();
+  const tempRoot = await fs.mkdtemp(path.join(os.tmpdir(), prefix));
+  const stateDir = path.join(tempRoot, "state");
+  await fs.mkdir(stateDir, { recursive: true });
+  setStateDirEnv(stateDir);
+  try {
+    return await fn({ tempRoot, stateDir });
+  } finally {
+    restoreStateDirEnv(snapshot);
+    await fs.rm(tempRoot, { recursive: true, force: true });
+  }
+}