test: reuse discovery env snapshot

This commit is contained in:
Shakker
2026-06-20 21:09:10 +01:00
parent 71a75b9b28
commit 1a5d84d3fe
+3 -10
View File
@@ -2,6 +2,7 @@
// wide-area DNS records, Bonjour naming, and shutdown cleanup.
import { afterEach, describe, expect, it, vi } from "vitest";
import type { PluginGatewayDiscoveryServiceRegistration } from "../plugins/registry-types.js";
import { captureFullEnv } from "../test-utils/env.js";
type WriteWideAreaGatewayZone = typeof import("../infra/widearea-dns.js").writeWideAreaGatewayZone;
type ResolveWideAreaDiscoveryDomain =
@@ -118,19 +119,11 @@ function startStuckDiscovery(timeoutMs: string) {
}
describe("startGatewayDiscovery", () => {
const prevEnv = { ...process.env };
const envSnapshot = captureFullEnv();
afterEach(() => {
vi.useRealTimers();
for (const key of Object.keys(process.env)) {
if (!(key in prevEnv)) {
delete process.env[key];
}
}
for (const [key, value] of Object.entries(prevEnv)) {
process.env[key] = value;
}
envSnapshot.restore();
vi.clearAllMocks();
});