@myclaw163/clawclaw-cli 0.6.70 → 0.6.74

package/src/runtime/ws-client.test.ts

@@ -1,104 +1,125 @@
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-
-const wsMock = vi.hoisted(() => ({
-  constructed: [] as Array<{ url: string; options: any; instance: any }>,
-}));
-
-vi.mock('ws', () => {
-  class MockWebSocket {
-    onopen: ((event: any) => void) | null = null;
-    onerror: ((event: any) => void) | null = null;
-    onclose: ((event: any) => void) | null = null;
-    onmessage: ((event: any) => void) | null = null;
-    send = vi.fn();
-    close = vi.fn();
-
-    constructor(url: string, options: any) {
-      wsMock.constructed.push({ url, options, instance: this });
-    }
-  }
-
-  return { default: MockWebSocket };
-});
-
-describe('WsClient', () => {
-  beforeEach(() => {
-    wsMock.constructed.length = 0;
-  });
-
-  it('connects with Authorization header instead of api_key query', async () => {
-    const { WsClient } = await import('./ws-client.js');
-    const client = new WsClient({
-      url: 'wss://host.example/api/v1/game/stream',
-      apiKey: 'claw_secret',
-    });
-
-    const connected = client.connect();
-    expect(wsMock.constructed).toHaveLength(1);
-    expect(wsMock.constructed[0].url).toBe('wss://host.example/api/v1/game/stream');
-    expect(wsMock.constructed[0].url).not.toContain('api_key');
-    expect(wsMock.constructed[0].options).toMatchObject({
-      headers: { Authorization: 'Bearer claw_secret' },
-    });
-
-    wsMock.constructed[0].instance.onopen?.({});
-    await connected;
-  });
-
-  it('dispatches state.new_events before the _state snapshot', async () => {
-    const { WsClient } = await import('./ws-client.js');
-    const client = new WsClient({
-      url: 'wss://host.example/api/v1/game/stream',
-      apiKey: 'claw_secret',
-    });
-    const seen: Record<string, any>[] = [];
-    client.on('*', (event) => seen.push(event));
-
-    (client as any).handleMessage({
-      type: 'state',
-      data: {
-        phase: 'wandering',
-        tick: 20,
-        new_events: [{ type: 'game_over', tick: 20, winner: 'crab' }],
-      },
-    });
-
-    expect(seen).toEqual([
-      { type: 'game_over', tick: 20, winner: 'crab' },
-      {
-        type: '_state',
-        phase: 'wandering',
-        tick: 20,
-        new_events: [{ type: 'game_over', tick: 20, winner: 'crab' }],
-      },
-    ]);
-  });
-
-  it('dispatches a _state snapshot when phase is game_over without new_events', async () => {
-    const { WsClient } = await import('./ws-client.js');
-    const client = new WsClient({
-      url: 'wss://host.example/api/v1/game/stream',
-      apiKey: 'claw_secret',
-    });
-    const seen: Record<string, any>[] = [];
-    client.on('*', (event) => seen.push(event));
-
-    (client as any).handleMessage({
-      type: 'state',
-      data: {
-        phase: 'game_over',
-        tick: 21,
-        new_events: [],
-      },
-    });
-
-    expect(seen).toEqual([
-      {
-        type: '_state',
-        phase: 'game_over',
-        tick: 21,
-        new_events: [],
-      },
-    ]);
-  });
-});
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+const wsMock = vi.hoisted(() => ({
+  constructed: [] as Array<{ url: string; options: any; instance: any }>,
+}));
+
+vi.mock('ws', () => {
+  class MockWebSocket {
+    onopen: ((event: any) => void) | null = null;
+    onerror: ((event: any) => void) | null = null;
+    onclose: ((event: any) => void) | null = null;
+    onmessage: ((event: any) => void) | null = null;
+    send = vi.fn();
+    close = vi.fn();
+
+    constructor(url: string, options: any) {
+      wsMock.constructed.push({ url, options, instance: this });
+    }
+  }
+
+  return { default: MockWebSocket };
+});
+
+describe('WsClient', () => {
+  beforeEach(() => {
+    wsMock.constructed.length = 0;
+  });
+
+  it('connects with Authorization header instead of api_key query', async () => {
+    const { WsClient } = await import('./ws-client.js');
+    const client = new WsClient({
+      url: 'wss://host.example/api/v1/game/stream',
+      apiKey: 'claw_secret',
+    });
+
+    const connected = client.connect();
+    expect(wsMock.constructed).toHaveLength(1);
+    expect(wsMock.constructed[0].url).toBe('wss://host.example/api/v1/game/stream');
+    expect(wsMock.constructed[0].url).not.toContain('api_key');
+    expect(wsMock.constructed[0].options).toMatchObject({
+      headers: { Authorization: 'Bearer claw_secret' },
+    });
+
+    wsMock.constructed[0].instance.onopen?.({});
+    await connected;
+  });
+
+  it('passes WebSocket close code and reason to disconnect handlers', async () => {
+    const { WsClient } = await import('./ws-client.js');
+    const client = new WsClient({
+      url: 'wss://host.example/api/v1/game/stream',
+      apiKey: 'claw_secret',
+    });
+    const closes: Record<string, any>[] = [];
+    client.onDisconnect((info) => closes.push(info));
+
+    const connected = client.connect();
+    wsMock.constructed[0].instance.onopen?.({});
+    await connected;
+
+    wsMock.constructed[0].instance.onclose?.({ code: 1000, reason: 'Left game', wasClean: true });
+    client.disconnect();
+
+    expect(closes).toEqual([
+      { code: 1000, reason: 'Left game', wasClean: true },
+    ]);
+  });
+
+  it('dispatches state.new_events before the _state snapshot', async () => {
+    const { WsClient } = await import('./ws-client.js');
+    const client = new WsClient({
+      url: 'wss://host.example/api/v1/game/stream',
+      apiKey: 'claw_secret',
+    });
+    const seen: Record<string, any>[] = [];
+    client.on('*', (event) => seen.push(event));
+
+    (client as any).handleMessage({
+      type: 'state',
+      data: {
+        phase: 'wandering',
+        tick: 20,
+        new_events: [{ type: 'game_over', tick: 20, winner: 'crab' }],
+      },
+    });
+
+    expect(seen).toEqual([
+      { type: 'game_over', tick: 20, winner: 'crab' },
+      {
+        type: '_state',
+        phase: 'wandering',
+        tick: 20,
+        new_events: [{ type: 'game_over', tick: 20, winner: 'crab' }],
+      },
+    ]);
+  });
+
+  it('dispatches a _state snapshot when phase is game_over without new_events', async () => {
+    const { WsClient } = await import('./ws-client.js');
+    const client = new WsClient({
+      url: 'wss://host.example/api/v1/game/stream',
+      apiKey: 'claw_secret',
+    });
+    const seen: Record<string, any>[] = [];
+    client.on('*', (event) => seen.push(event));
+
+    (client as any).handleMessage({
+      type: 'state',
+      data: {
+        phase: 'game_over',
+        tick: 21,
+        new_events: [],
+      },
+    });
+
+    expect(seen).toEqual([
+      {
+        type: '_state',
+        phase: 'game_over',
+        tick: 21,
+        new_events: [],
+      },
+    ]);
+  });
+});