File size: 3,304 Bytes
fc93158 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | 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";
const keyB = "OPENCLAW_ENV_TEST_B";
const snapshot = captureEnv([keyA, keyB]);
const prevA = process.env[keyA];
const prevB = process.env[keyB];
process.env[keyA] = "mutated";
delete process.env[keyB];
snapshot.restore();
expect(process.env[keyA]).toBe(prevA);
expect(process.env[keyB]).toBe(prevB);
});
it("captureFullEnv restores added keys and baseline values", () => {
const key = "OPENCLAW_ENV_TEST_ADDED";
const prevHome = process.env.HOME;
const snapshot = captureFullEnv();
process.env[key] = "1";
delete process.env.HOME;
snapshot.restore();
expect(process.env[key]).toBeUndefined();
expect(process.env.HOME).toBe(prevHome);
});
it("withEnv applies values only inside callback", () => {
const key = "OPENCLAW_ENV_TEST_SYNC";
const prev = process.env[key];
const seen = withEnv({ [key]: "inside" }, () => process.env[key]);
expect(seen).toBe("inside");
expect(process.env[key]).toBe(prev);
});
it("withEnv restores values when callback throws", () => {
const key = "OPENCLAW_ENV_TEST_SYNC_THROW";
const prev = process.env[key];
expect(() =>
withEnv({ [key]: "inside" }, () => {
expect(process.env[key]).toBe("inside");
throw new Error("boom");
}),
).toThrow("boom");
expect(process.env[key]).toBe(prev);
});
it("withEnv can delete a key only inside callback", () => {
const key = "OPENCLAW_ENV_TEST_SYNC_DELETE";
const prev = process.env[key];
process.env[key] = "outer";
const seen = withEnv({ [key]: undefined }, () => process.env[key]);
expect(seen).toBeUndefined();
expect(process.env[key]).toBe("outer");
restoreEnvKey(key, prev);
});
it("withEnvAsync restores values when callback throws", async () => {
const key = "OPENCLAW_ENV_TEST_ASYNC";
const prev = process.env[key];
await expect(
withEnvAsync({ [key]: "inside" }, async () => {
expect(process.env[key]).toBe("inside");
throw new Error("boom");
}),
).rejects.toThrow("boom");
expect(process.env[key]).toBe(prev);
});
it("withEnvAsync applies values only inside async callback", async () => {
const key = "OPENCLAW_ENV_TEST_ASYNC_OK";
const prev = process.env[key];
const seen = await withEnvAsync({ [key]: "inside" }, async () => process.env[key]);
expect(seen).toBe("inside");
expect(process.env[key]).toBe(prev);
});
it("withEnvAsync can delete a key only inside callback", async () => {
const key = "OPENCLAW_ENV_TEST_ASYNC_DELETE";
const prev = process.env[key];
process.env[key] = "outer";
const seen = await withEnvAsync({ [key]: undefined }, async () => process.env[key]);
expect(seen).toBeUndefined();
expect(process.env[key]).toBe("outer");
restoreEnvKey(key, prev);
});
});
|