Jest如何测试WebSocket通信

要使用Jest测试WebSocket通信，你需要模拟WebSocket对象并提供一个模拟的服务器端。以下是一个简单的示例，展示了如何使用Jest测试WebSocket通信：

1. 首先，安装ws库，它是一个WebSocket服务器和客户端库：

npm install ws

2. 创建一个简单的WebSocket服务器：

// server.js
const WebSocket = require('ws');

const server = new WebSocket.Server({ port: 8080 });

server.on('connection', (socket) => {
  socket.on('message', (message) => {
    console.log(`Received message: ${message}`);
    socket.send(`Server received: ${message}`);
  });
});

3. 创建一个WebSocket客户端，用于与服务器通信：

// client.js
const WebSocket = require('ws');

const socket = new WebSocket('ws://localhost:8080');

socket.on('open', () => {
  socket.send('Hello, Server!');
});

socket.on('message', (message) => {
  console.log(`Received from server: ${message}`);
});

4. 编写一个Jest测试，模拟WebSocket通信：

// client.test.js
const WebSocket = require('ws');

jest.mock('ws');

describe('WebSocket Client', () => {
  let socket;

  beforeEach(() => {
    socket = new WebSocket('ws://localhost:8080');
  });

  afterEach(() => {
    socket.close();
  });

  test('sends and receives messages', (done) => {
    const mockMessage = 'Hello, Server!';
    const expectedResponse = `Server received: ${mockMessage}`;

    // Mock the WebSocket server
    WebSocket.Server.mockImplementation((options) => {
      expect(options.port).toBe(8080);
      return {
        on: (event, callback) => {
          if (event === 'connection') {
            callback(socket);
          }
        },
      };
    });

    // Mock the WebSocket client
    socket.send = jest.fn();
    socket.on = jest.fn((event, callback) => {
      if (event === 'open') {
        callback();
      } else if (event === 'message') {
        callback(expectedResponse);
      }
    });

    // Initialize the client
    require('./client');

    // Expectations
    expect(socket.send).toHaveBeenCalledWith(mockMessage);
    expect(socket.on).toHaveBeenCalledTimes(2);

    done();
  });
});

5. 运行测试：

npx jest

这个示例展示了如何使用Jest测试WebSocket通信。请注意，这里我们使用了jest.mock()来模拟WebSocket库，以便在测试中控制WebSocket的行为。