File size: 854 Bytes
fb4d8fe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import { describe, expect, test, vi } from "vitest";

describe("GatewayClient", () => {
  test("uses a large maxPayload for node snapshots", async () => {
    vi.resetModules();

    class MockWebSocket {
      static last: { url: unknown; opts: unknown } | null = null;

      on = vi.fn();
      close = vi.fn();
      send = vi.fn();

      constructor(url: unknown, opts: unknown) {
        MockWebSocket.last = { url, opts };
      }
    }

    vi.doMock("ws", () => ({
      WebSocket: MockWebSocket,
    }));

    const { GatewayClient } = await import("./client.js");
    const client = new GatewayClient({ url: "ws://127.0.0.1:1" });
    client.start();

    expect(MockWebSocket.last?.url).toBe("ws://127.0.0.1:1");
    expect(MockWebSocket.last?.opts).toEqual(
      expect.objectContaining({ maxPayload: 25 * 1024 * 1024 }),
    );
  });
});