SnowLuma/packages/onebot/tests/network-utils.test.ts

import { describe, expect, it, vi } from 'vitest';
import type { WebSocket } from '@snowluma/websocket';
import { StreamTransportClosedError } from '../src/streaming';
import { safeSendAsync } from '../src/network/utils';

describe('safeSendAsync', () => {
  it('resolves after a successful WebSocket write callback', async () => {
    const socket = {
      readyState: 1,
      send: vi.fn((_payload: string, callback: (error?: Error | null) => void) => {
        callback(null);
      }),
    } as unknown as WebSocket;

    await expect(safeSendAsync(socket, 'frame')).resolves.toBeUndefined();
    expect(socket.send).toHaveBeenCalledWith('frame', expect.any(Function));
  });

  it('rejects when the socket is not open', async () => {
    const socket = {
      readyState: 3,
      send: vi.fn(),
    } as unknown as WebSocket;

    await expect(safeSendAsync(socket, 'frame'))
      .rejects.toBeInstanceOf(StreamTransportClosedError);
    expect(socket.send).not.toHaveBeenCalled();
  });

  it('rejects a synchronous WebSocket send failure', async () => {
    const socket = {
      readyState: 1,
      send: vi.fn(() => { throw new Error('socket gone'); }),
    } as unknown as WebSocket;

    await expect(safeSendAsync(socket, 'frame'))
      .rejects.toBeInstanceOf(StreamTransportClosedError);
  });

  it('rejects an asynchronous WebSocket write failure', async () => {
    const socket = {
      readyState: 1,
      send: vi.fn((_payload: string, callback: (error?: Error | null) => void) => {
        callback(new Error('write failed'));
      }),
    } as unknown as WebSocket;

    await expect(safeSendAsync(socket, 'frame'))
      .rejects.toBeInstanceOf(StreamTransportClosedError);
  });
});