liuyu/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-06-06 05:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: scope preauth budget env

Browse Source
This commit is contained in:
Shakker
2026-06-05 02:11:44 +01:00
parent 38e142657b
commit 85e16da2b4
1 changed files with 15 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/gateway/server/preauth-connection-budget.test.ts
View File

@@ -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);
});
});
});