test: share temp home env harness

This commit is contained in:
Peter Steinberger
2026-02-19 14:39:01 +00:00
parent edf92f1cb0
commit 0213a09211
4 changed files with 78 additions and 52 deletions

View File

@@ -0,0 +1,26 @@
import fs from "node:fs/promises";
import { describe, expect, it } from "vitest";
import { createTempHomeEnv } from "./temp-home.js";
describe("createTempHomeEnv", () => {
it("sets home env vars and restores them on cleanup", async () => {
const previousHome = process.env.HOME;
const previousUserProfile = process.env.USERPROFILE;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
const tempHome = await createTempHomeEnv("openclaw-temp-home-");
expect(process.env.HOME).toBe(tempHome.home);
expect(process.env.USERPROFILE).toBe(tempHome.home);
expect(process.env.OPENCLAW_STATE_DIR).toBe(`${tempHome.home}/.openclaw`);
await expect(fs.stat(tempHome.home)).resolves.toMatchObject({
isDirectory: expect.any(Function),
});
await tempHome.restore();
expect(process.env.HOME).toBe(previousHome);
expect(process.env.USERPROFILE).toBe(previousUserProfile);
expect(process.env.OPENCLAW_STATE_DIR).toBe(previousStateDir);
await expect(fs.stat(tempHome.home)).rejects.toThrow();
});
});

View File

@@ -0,0 +1,43 @@
import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { captureEnv } from "./env.js";
const HOME_ENV_KEYS = [
"HOME",
"USERPROFILE",
"HOMEDRIVE",
"HOMEPATH",
"OPENCLAW_STATE_DIR",
] as const;
export type TempHomeEnv = {
home: string;
restore: () => Promise<void>;
};
export async function createTempHomeEnv(prefix: string): Promise<TempHomeEnv> {
const home = await fs.mkdtemp(path.join(os.tmpdir(), prefix));
await fs.mkdir(path.join(home, ".openclaw"), { recursive: true });
const snapshot = captureEnv([...HOME_ENV_KEYS]);
process.env.HOME = home;
process.env.USERPROFILE = home;
process.env.OPENCLAW_STATE_DIR = path.join(home, ".openclaw");
if (process.platform === "win32") {
const match = home.match(/^([A-Za-z]:)(.*)$/);
if (match) {
process.env.HOMEDRIVE = match[1];
process.env.HOMEPATH = match[2] || "\\";
}
}
return {
home,
restore: async () => {
snapshot.restore();
await fs.rm(home, { recursive: true, force: true });
},
};
}