Harden websocket client error handling

packages/loro-websocket/src/client/index.test.ts

@@ -1,7 +1,142 @@
-import { describe, it, expect } from "vitest";
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import {
+  CrdtType,
+  JoinErrorCode,
+  MessageType,
+  type JoinError,
+} from "loro-protocol";
+import * as protocol from "loro-protocol";
+import { LoroWebsocketClient } from "./index";
+
+class FakeWebSocket {
+  static CONNECTING = 0;
+  static OPEN = 1;
+  static CLOSING = 2;
+  static CLOSED = 3;
+
+  readyState = FakeWebSocket.CLOSED;
+  bufferedAmount = 0;
+  binaryType: any = "arraybuffer";
+  url: string;
+  lastSent: unknown;
+  private listeners = new Map<string, Set<(ev: any) => void>>();
+
+  constructor(url: string) {
+    this.url = url;
+  }
+
+  addEventListener(type: string, listener: (ev: any) => void) {
+    const set = this.listeners.get(type) ?? new Set();
+    set.add(listener);
+    this.listeners.set(type, set);
+  }
+
+  removeEventListener(type: string, listener: (ev: any) => void) {
+    const set = this.listeners.get(type);
+    set?.delete(listener);
+  }
+
+  dispatch(type: string, ev: any) {
+    const set = this.listeners.get(type);
+    if (!set) return;
+    for (const l of Array.from(set)) l(ev);
+  }
+
+  send(data: any) {
+    if (this.readyState !== FakeWebSocket.OPEN) {
+      throw new Error("WebSocket is not open");
+    }
+    this.lastSent = data;
+  }
+
+  close() {
+    this.readyState = FakeWebSocket.CLOSED;
+  }
+}
 
 describe("LoroWebsocketClient", () => {
-  it("is placeholder", () => {
-    expect(true).toBe(true);
+  let originalWebSocket: any;
+
+  beforeEach(() => {
+    originalWebSocket = (globalThis as any).WebSocket;
+    (globalThis as any).WebSocket = FakeWebSocket as any;
+  });
+
+  afterEach(() => {
+    (globalThis as any).WebSocket = originalWebSocket;
+    vi.restoreAllMocks();
+  });
+
+  it("does not throw when retrying join after closed socket and reports via onError", async () => {
+    const onError = vi.fn();
+    const client = new LoroWebsocketClient({
+      url: "ws://test",
+      disablePing: true,
+      reconnect: { enabled: false },
+      onError,
+    });
+
+    const adaptor = {
+      crdtType: CrdtType.Loro,
+      setCtx: () => { },
+      getVersion: () => new Uint8Array([0]),
+      getAlternativeVersion: () => new Uint8Array([1]),
+      handleJoinOk: async () => { },
+      waitForReachingServerVersion: async () => { },
+      destroy: () => { },
+    } satisfies any;
+
+    const joinError: JoinError = {
+      type: MessageType.JoinError,
+      code: JoinErrorCode.VersionUnknown,
+      message: "",
+      crdt: adaptor.crdtType,
+      roomId: "room",
+    };
+
+    const pending = {
+      room: Promise.resolve({} as any),
+      resolve: () => { },
+      reject: () => { },
+      adaptor,
+      roomId: "room",
+    } satisfies any;
+
+    // Avoid unhandled rejection when the client is destroyed without ever opening.
+    (client as any).connectedPromise?.catch(() => { });
+
+    // Force the current socket to a closed state so send will fail.
+    (client as any).ws.readyState = FakeWebSocket.CLOSED;
+
+    await expect(
+      (client as any).handleJoinError(joinError, pending, adaptor.crdtType + "room")
+    ).resolves.not.toThrow();
+
+    expect(onError).toHaveBeenCalledTimes(1);
+    expect(((client as any).queuedJoins ?? []).length).toBeGreaterThan(0);
+
+  });
+
+  it("forwards decode or handler errors to onError instead of crashing", async () => {
+    const onError = vi.fn();
+    const client = new LoroWebsocketClient({
+      url: "ws://test",
+      disablePing: true,
+      reconnect: { enabled: false },
+      onError,
+    });
+
+    (client as any).connectedPromise?.catch(() => { });
+
+    vi.spyOn(protocol, "tryDecode").mockImplementation(() => {
+      throw new Error("decode failed");
+    });
+
+    await (client as any).onSocketMessage((client as any).ws, {
+      data: new ArrayBuffer(0),
+    } as MessageEvent<ArrayBuffer>);
+
+    expect(onError).toHaveBeenCalledTimes(1);
+
   });
 });