From 85e16da2b48a9ec150c519946cbeab5d190cdd99 Mon Sep 17 00:00:00 2001 From: Shakker Date: Fri, 5 Jun 2026 02:11:44 +0100 Subject: [PATCH] test: scope preauth budget env --- .../server/preauth-connection-budget.test.ts | 31 +++++++++---------- 1 file changed, 15 insertions(+), 16 deletions(-) diff --git a/src/gateway/server/preauth-connection-budget.test.ts b/src/gateway/server/preauth-connection-budget.test.ts index 58ea2454e651..e195f7ac5f5d 100644 --- a/src/gateway/server/preauth-connection-budget.test.ts +++ b/src/gateway/server/preauth-connection-budget.test.ts @@ -1,14 +1,11 @@ /** * Pre-auth WebSocket connection-budget regression tests. */ -import { afterEach, describe, expect, it, vi } from "vitest"; +import { describe, expect, it } from "vitest"; +import { withEnv } from "../../test-utils/env.js"; import { createPreauthConnectionBudget } from "./preauth-connection-budget.js"; describe("createPreauthConnectionBudget", () => { - afterEach(() => { - vi.unstubAllEnvs(); - }); - it("caps connections with a finite configured limit", () => { const budget = createPreauthConnectionBudget(2); @@ -39,21 +36,23 @@ describe("createPreauthConnectionBudget", () => { }); it("accepts strict plus-signed env limits", () => { - vi.stubEnv("OPENCLAW_MAX_PREAUTH_CONNECTIONS_PER_IP", "+02"); - const budget = createPreauthConnectionBudget(); + withEnv({ OPENCLAW_MAX_PREAUTH_CONNECTIONS_PER_IP: "+02" }, () => { + const budget = createPreauthConnectionBudget(); - expect(budget.acquire("127.0.0.1")).toBe(true); - expect(budget.acquire("127.0.0.1")).toBe(true); - expect(budget.acquire("127.0.0.1")).toBe(false); + expect(budget.acquire("127.0.0.1")).toBe(true); + expect(budget.acquire("127.0.0.1")).toBe(true); + expect(budget.acquire("127.0.0.1")).toBe(false); + }); }); it("ignores non-decimal env limits", () => { - vi.stubEnv("OPENCLAW_MAX_PREAUTH_CONNECTIONS_PER_IP", "0x2"); - const budget = createPreauthConnectionBudget(); + withEnv({ OPENCLAW_MAX_PREAUTH_CONNECTIONS_PER_IP: "0x2" }, () => { + const budget = createPreauthConnectionBudget(); - for (let i = 0; i < 32; i += 1) { - expect(budget.acquire("127.0.0.1")).toBe(true); - } - expect(budget.acquire("127.0.0.1")).toBe(false); + for (let i = 0; i < 32; i += 1) { + expect(budget.acquire("127.0.0.1")).toBe(true); + } + expect(budget.acquire("127.0.0.1")).toBe(false); + }); }); });