mirror of
https://github.com/openclaw/openclaw.git
synced 2026-02-19 18:39:20 -05:00
test: share temp home env harness
This commit is contained in:
@@ -1,38 +1,14 @@
|
||||
import fs from "node:fs/promises";
|
||||
import os from "node:os";
|
||||
import path from "node:path";
|
||||
import { captureEnv } from "../test-utils/env.js";
|
||||
import { createTempHomeEnv } from "../test-utils/temp-home.js";
|
||||
|
||||
export async function withTempHome<T>(
|
||||
prefix: string,
|
||||
fn: (home: string) => Promise<T>,
|
||||
): Promise<T> {
|
||||
const home = await fs.mkdtemp(path.join(os.tmpdir(), prefix));
|
||||
await fs.mkdir(path.join(home, ".openclaw"), { recursive: true });
|
||||
|
||||
const snapshot = captureEnv([
|
||||
"HOME",
|
||||
"USERPROFILE",
|
||||
"HOMEDRIVE",
|
||||
"HOMEPATH",
|
||||
"OPENCLAW_STATE_DIR",
|
||||
]);
|
||||
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] || "\\";
|
||||
}
|
||||
}
|
||||
const tempHome = await createTempHomeEnv(prefix);
|
||||
|
||||
try {
|
||||
return await fn(home);
|
||||
return await fn(tempHome.home);
|
||||
} finally {
|
||||
snapshot.restore();
|
||||
await fs.rm(home, { recursive: true, force: true });
|
||||
await tempHome.restore();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,44 +1,25 @@
|
||||
import fs from "node:fs/promises";
|
||||
import os from "node:os";
|
||||
import path from "node:path";
|
||||
import JSZip from "jszip";
|
||||
import sharp from "sharp";
|
||||
import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";
|
||||
import { isPathWithinBase } from "../../test/helpers/paths.js";
|
||||
import { captureEnv } from "../test-utils/env.js";
|
||||
import { createTempHomeEnv, type TempHomeEnv } from "../test-utils/temp-home.js";
|
||||
|
||||
describe("media store", () => {
|
||||
let store: typeof import("./store.js");
|
||||
let home = "";
|
||||
let envSnapshot: ReturnType<typeof captureEnv>;
|
||||
let tempHome: TempHomeEnv;
|
||||
|
||||
beforeAll(async () => {
|
||||
envSnapshot = captureEnv([
|
||||
"HOME",
|
||||
"USERPROFILE",
|
||||
"HOMEDRIVE",
|
||||
"HOMEPATH",
|
||||
"OPENCLAW_STATE_DIR",
|
||||
]);
|
||||
home = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-test-home-"));
|
||||
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] || "\\";
|
||||
}
|
||||
}
|
||||
await fs.mkdir(path.join(home, ".openclaw"), { recursive: true });
|
||||
tempHome = await createTempHomeEnv("openclaw-test-home-");
|
||||
home = tempHome.home;
|
||||
store = await import("./store.js");
|
||||
});
|
||||
|
||||
afterAll(async () => {
|
||||
envSnapshot.restore();
|
||||
try {
|
||||
await fs.rm(home, { recursive: true, force: true });
|
||||
await tempHome.restore();
|
||||
} catch {
|
||||
// ignore cleanup failures in tests
|
||||
}
|
||||
|
||||
26
src/test-utils/temp-home.test.ts
Normal file
26
src/test-utils/temp-home.test.ts
Normal 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();
|
||||
});
|
||||
});
|
||||
43
src/test-utils/temp-home.ts
Normal file
43
src/test-utils/temp-home.ts
Normal 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 });
|
||||
},
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user