Jest如何测试WebSocket连接

要使用Jest测试WebSocket连接，您需要创建一个模拟的WebSocket服务器和客户端。这里是一个简单的示例，说明如何使用ws库（WebSocket库）和Jest来测试WebSocket连接。

首先，确保已安装以下依赖项：

npm install ws jest

然后，创建一个名为websocket.test.js的文件，其中包含以下内容：

const WebSocket = require('ws');

// 创建一个模拟的WebSocket服务器
const server = new WebSocket.Server({ port: 8080 });

// 当客户端连接时，向其发送一条消息
server.on('connection', (socket) => {
  socket.send('Hello from server!');
});

// 测试WebSocket连接
test('WebSocket connection', (done) => {
  // 创建一个WebSocket客户端
  const client = new WebSocket('ws://localhost:8080');

  // 当客户端收到服务器发来的消息时
  client.onmessage = (event) => {
    // 验证消息内容
    expect(event.data).toBe('Hello from server!');
    done();
  };

  // 关闭服务器和客户端
  afterAll(() => {
    client.close();
    server.close();
  });
});

最后，运行测试：

npx jest websocket.test.js

这将启动一个模拟的WebSocket服务器，并创建一个客户端连接到该服务器。当客户端收到服务器发来的消息时，测试将通过验证消息内容。在测试完成后，服务器和客户端将关闭。