inital commit

Overlord-Server/src/wsValidation.test.ts

@@ -0,0 +1,35 @@
+import { describe, expect, test } from "bun:test";
+import {
+  ALLOWED_CLIENT_MESSAGE_TYPES,
+  getMaxPayloadLimit,
+  getMessageByteLength,
+  isAllowedClientMessageType,
+} from "./wsValidation";
+
+describe("wsValidation", () => {
+  test("allowed client message types include core types", () => {
+    expect(ALLOWED_CLIENT_MESSAGE_TYPES.has("hello")).toBe(true);
+    expect(ALLOWED_CLIENT_MESSAGE_TYPES.has("ping")).toBe(true);
+    expect(ALLOWED_CLIENT_MESSAGE_TYPES.has("frame")).toBe(true);
+  });
+
+  test("unknown client message types are rejected", () => {
+    expect(isAllowedClientMessageType("not_a_real_type")).toBe(false);
+  });
+
+  test("getMessageByteLength handles string and binary", () => {
+    const textLen = getMessageByteLength("hello");
+    const buffer = new Uint8Array([1, 2, 3, 4]);
+    const arrayBuf = buffer.buffer;
+
+    expect(textLen).toBe(5);
+    expect(getMessageByteLength(buffer)).toBe(4);
+    expect(getMessageByteLength(arrayBuf)).toBe(4);
+  });
+
+  test("getMaxPayloadLimit uses client limit when role is client", () => {
+    expect(getMaxPayloadLimit("client", 10, 1)).toBe(10);
+    expect(getMaxPayloadLimit("viewer", 10, 1)).toBe(1);
+    expect(getMaxPayloadLimit(undefined, 10, 1)).toBe(1);
+  });
+});