Jest如何測(cè)試WebSocket通信

要使用Jest測(cè)試WebSocket通信，你需要模擬WebSocket對(duì)象并提供一個(gè)模擬的服務(wù)器端。以下是一個(gè)簡(jiǎn)單的示例，展示了如何使用Jest測(cè)試WebSocket通信：

1. 首先，安裝ws庫(kù)，它是一個(gè)WebSocket服務(wù)器和客戶端庫(kù)：

npm install ws

2. 創(chuàng)建一個(gè)簡(jiǎn)單的WebSocket服務(wù)器：

// 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. 創(chuàng)建一個(gè)WebSocket客戶端，用于與服務(wù)器通信：

// 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. 編寫(xiě)一個(gè)Jest測(cè)試，模擬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. 運(yùn)行測(cè)試：

npx jest

這個(gè)示例展示了如何使用Jest測(cè)試WebSocket通信。請(qǐng)注意，這里我們使用了jest.mock()來(lái)模擬WebSocket庫(kù)，以便在測(cè)試中控制WebSocket的行為。