github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-02-19 18:39:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

test: reuse shared env helper in runtime version tests

Browse Source
This commit is contained in:
Gustavo Madeira Santana
2026-02-17 20:56:20 -05:00
parent cd3a97cce1
commit 10d9df5263
2 changed files with 16 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/gateway/server.auth.e2e.test.ts
View File

@@ -1,5 +1,6 @@
import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, test, vi } from "vitest";
import { WebSocket } from "ws";
import { withEnvAsync } from "../test-utils/env.js";
import { GATEWAY_CLIENT_MODES, GATEWAY_CLIENT_NAMES } from "../utils/message-channel.js";
import { buildDeviceAuthPayload } from "./device-auth.js";
import { PROTOCOL_VERSION } from "./protocol/index.js";
@@ -63,39 +64,11 @@ function restoreGatewayToken(prevToken: string | undefined) {
}
}
const RUNTIME_VERSION_ENV_KEYS = [
"OPENCLAW_VERSION",
"OPENCLAW_SERVICE_VERSION",
"npm_package_version",
] as const;
type RuntimeVersionEnvKey = (typeof RUNTIME_VERSION_ENV_KEYS)[number];
type RuntimeVersionEnv = Partial<Record<RuntimeVersionEnvKey, string | undefined>>;
async function withRuntimeVersionEnv<T>(env: RuntimeVersionEnv, run: () => Promise<T>): Promise<T> {
const previous = Object.fromEntries(
RUNTIME_VERSION_ENV_KEYS.map((key) => [key, process.env[key]]),
) as Record<RuntimeVersionEnvKey, string | undefined>;
for (const key of RUNTIME_VERSION_ENV_KEYS) {
const value = env[key];
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
try {
return await run();
} finally {
for (const key of RUNTIME_VERSION_ENV_KEYS) {
const value = previous[key];
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
}
async function withRuntimeVersionEnv<T>(
env: Record<string, string | undefined>,
run: () => Promise<T>,
): Promise<T> {
return withEnvAsync(env, run);
}
const TEST_OPERATOR_CLIENT = {

47
src/infra/system-presence.version.test.ts
View File

@@ -1,46 +1,19 @@
import { describe, expect, it, vi } from "vitest";
const VERSION_ENV_KEYS = [
"OPENCLAW_VERSION",
"OPENCLAW_SERVICE_VERSION",
"npm_package_version",
] as const;
type VersionEnvKey = (typeof VERSION_ENV_KEYS)[number];
type VersionEnv = Partial<Record<VersionEnvKey, string | undefined>>;
import { withEnvAsync } from "../test-utils/env.js";
async function withPresenceModule<T>(
env: VersionEnv,
env: Record<string, string | undefined>,
run: (module: typeof import("./system-presence.js")) => Promise<T> | T,
): Promise<T> {
const previous = Object.fromEntries(
VERSION_ENV_KEYS.map((key) => [key, process.env[key]]),
) as Record<VersionEnvKey, string | undefined>;
for (const key of VERSION_ENV_KEYS) {
const value = env[key];
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
vi.resetModules();
try {
const module = await import("./system-presence.js");
return await run(module);
} finally {
for (const key of VERSION_ENV_KEYS) {
const value = previous[key];
if (value === undefined) {
delete process.env[key];
} else {
process.env[key] = value;
}
}
return withEnvAsync(env, async () => {
vi.resetModules();
}
try {
const module = await import("./system-presence.js");
return await run(module);
} finally {
vi.resetModules();
}
});
}
describe("system-presence version fallback", () => {