LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-13 23:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(gateway): isolate Fly env container checks

Browse Source
This commit is contained in:
Lior Balmas
2026-05-10 14:15:34 +03:00
committed by Peter Steinberger
parent 0f8fc6bb61
commit b94919ab1b
1 changed files with 44 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/gateway/net.test.ts
View File

@@ -1,5 +1,5 @@
import os from "node:os";
import { afterEach, describe, expect, it, vi } from "vitest";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { makeNetworkInterfacesSnapshot } from "../test-helpers/network-interfaces.js";
import {
__resetContainerCacheForTest,
@@ -18,6 +18,40 @@ import {
resolveHostName,
} from "./net.js";
const flyMachineEnvKeys = ["FLY_MACHINE_ID", "FLY_APP_NAME"] as const;
function clearFlyMachineEnvForTest(): () => void {
const previousEnv = new Map<(typeof flyMachineEnvKeys)[number], string | undefined>();
for (const key of flyMachineEnvKeys) {
previousEnv.set(key, process.env[key]);
delete process.env[key];
}
return () => {
for (const key of flyMachineEnvKeys) {
const value = previousEnv.get(key);
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
};
}
function useClearedFlyMachineEnv() {
let restoreFlyMachineEnv: (() => void) | undefined;
beforeEach(() => {
restoreFlyMachineEnv = clearFlyMachineEnvForTest();
});
afterEach(() => {
restoreFlyMachineEnv?.();
restoreFlyMachineEnv = undefined;
});
}
describe("resolveHostName", () => {
it.each([
{ input: "localhost:18789", expected: "localhost" },
@@ -482,6 +516,8 @@ describe("isPrivateOrLoopbackHost", () => {
});
describe("isContainerEnvironment", () => {
useClearedFlyMachineEnv();
afterEach(() => {
__resetContainerCacheForTest();
vi.restoreAllMocks();
@@ -516,30 +552,15 @@ describe("isContainerEnvironment", () => {
});
it("returns true on Fly Machines without Docker sentinel files", () => {
const previousFlyMachineId = process.env.FLY_MACHINE_ID;
const previousFlyAppName = process.env.FLY_APP_NAME;
const fs = require("node:fs");
vi.spyOn(fs, "accessSync").mockImplementation(() => {
throw new Error("ENOENT");
});
vi.spyOn(fs, "readFileSync").mockReturnValue("10:cpuset:/\n9:perf_event:/\n8:memory:/\n0::/\n");
try {
process.env.FLY_MACHINE_ID = "3d8d5459a03038";
process.env.FLY_APP_NAME = "openclaw-clawcks-test";
expect(isContainerEnvironment()).toBe(true);
} finally {
if (previousFlyMachineId === undefined) {
delete process.env.FLY_MACHINE_ID;
} else {
process.env.FLY_MACHINE_ID = previousFlyMachineId;
}
if (previousFlyAppName === undefined) {
delete process.env.FLY_APP_NAME;
} else {
process.env.FLY_APP_NAME = previousFlyAppName;
}
}
process.env.FLY_MACHINE_ID = "3d8d5459a03038";
process.env.FLY_APP_NAME = "openclaw-test";
expect(isContainerEnvironment()).toBe(true);
});
it("returns true when /proc/1/cgroup contains docker marker", () => {
@@ -613,6 +634,8 @@ describe("isContainerEnvironment", () => {
});
describe("resolveGatewayBindHost", () => {
useClearedFlyMachineEnv();
afterEach(() => {
__resetContainerCacheForTest();
vi.restoreAllMocks();
@@ -652,6 +675,8 @@ describe("resolveGatewayBindHost", () => {
});
describe("defaultGatewayBindMode", () => {
useClearedFlyMachineEnv();
afterEach(() => {
__resetContainerCacheForTest();
vi.restoreAllMocks();