




























@@ -3,6 +3,7 @@
33import os from "node:os";
44import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
55import { makeNetworkInterfacesSnapshot } from "../test-helpers/network-interfaces.js";
6+import { captureEnv, deleteTestEnvValue, setTestEnvValue } from "../test-utils/env.js";
67import {
78__resetContainerCacheForTest,
89defaultGatewayBindMode,
@@ -24,22 +25,12 @@ import {
2425const flyMachineEnvKeys = ["FLY_MACHINE_ID", "FLY_APP_NAME"] as const;
25262627function clearFlyMachineEnvForTest(): () => void {
27-const previousEnv = new Map<(typeof flyMachineEnvKeys)[number], string | undefined>();
28+const envSnapshot = captureEnv([...flyMachineEnvKeys]);
2829for (const key of flyMachineEnvKeys) {
29-previousEnv.set(key, process.env[key]);
30-delete process.env[key];
30+deleteTestEnvValue(key);
3131}
323233-return () => {
34-for (const key of flyMachineEnvKeys) {
35-const value = previousEnv.get(key);
36-if (value === undefined) {
37-delete process.env[key];
38-} else {
39-process.env[key] = value;
40-}
41-}
42-};
33+return () => envSnapshot.restore();
4334}
44354536function useClearedFlyMachineEnv() {
@@ -587,8 +578,8 @@ describe("isContainerEnvironment", () => {
587578});
588579vi.spyOn(fs, "readFileSync").mockReturnValue("10:cpuset:/\n9:perf_event:/\n8:memory:/\n0::/\n");
589580590-process.env.FLY_MACHINE_ID = "3d8d5459a03038";
591-process.env.FLY_APP_NAME = "openclaw-test";
581+setTestEnvValue("FLY_MACHINE_ID", "3d8d5459a03038");
582+setTestEnvValue("FLY_APP_NAME", "openclaw-test");
592583expect(isContainerEnvironment()).toBe(true);
593584});
594585此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。