mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-07 05:31:38 +00:00
refactor(agents): dedupe plugin hooks and test helpers
This commit is contained in:
@@ -1,6 +1,14 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { captureEnv, captureFullEnv, withEnv, withEnvAsync } from "./env.js";
|
||||
|
||||
function restoreEnvKey(key: string, previous: string | undefined): void {
|
||||
if (previous === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = previous;
|
||||
}
|
||||
}
|
||||
|
||||
describe("env test utils", () => {
|
||||
it("captureEnv restores mutated keys", () => {
|
||||
const keyA = "OPENCLAW_ENV_TEST_A";
|
||||
@@ -63,11 +71,7 @@ describe("env test utils", () => {
|
||||
|
||||
expect(seen).toBeUndefined();
|
||||
expect(process.env[key]).toBe("outer");
|
||||
if (prev === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = prev;
|
||||
}
|
||||
restoreEnvKey(key, prev);
|
||||
});
|
||||
|
||||
it("withEnvAsync restores values when callback throws", async () => {
|
||||
@@ -103,10 +107,6 @@ describe("env test utils", () => {
|
||||
|
||||
expect(seen).toBeUndefined();
|
||||
expect(process.env[key]).toBe("outer");
|
||||
if (prev === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = prev;
|
||||
}
|
||||
restoreEnvKey(key, prev);
|
||||
});
|
||||
});
|
||||
|
||||
@@ -17,6 +17,16 @@ export function captureEnv(keys: string[]) {
|
||||
};
|
||||
}
|
||||
|
||||
function applyEnvValues(env: Record<string, string | undefined>): void {
|
||||
for (const [key, value] of Object.entries(env)) {
|
||||
if (value === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export function captureFullEnv() {
|
||||
const snapshot: Record<string, string | undefined> = { ...process.env };
|
||||
|
||||
@@ -41,13 +51,7 @@ export function captureFullEnv() {
|
||||
export function withEnv<T>(env: Record<string, string | undefined>, fn: () => T): T {
|
||||
const snapshot = captureEnv(Object.keys(env));
|
||||
try {
|
||||
for (const [key, value] of Object.entries(env)) {
|
||||
if (value === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = value;
|
||||
}
|
||||
}
|
||||
applyEnvValues(env);
|
||||
return fn();
|
||||
} finally {
|
||||
snapshot.restore();
|
||||
@@ -60,13 +64,7 @@ export async function withEnvAsync<T>(
|
||||
): Promise<T> {
|
||||
const snapshot = captureEnv(Object.keys(env));
|
||||
try {
|
||||
for (const [key, value] of Object.entries(env)) {
|
||||
if (value === undefined) {
|
||||
delete process.env[key];
|
||||
} else {
|
||||
process.env[key] = value;
|
||||
}
|
||||
}
|
||||
applyEnvValues(env);
|
||||
return await fn();
|
||||
} finally {
|
||||
snapshot.restore();
|
||||
|
||||
Reference in New Issue
Block a user