使用开玩笑的 websocket 服务器"ws"库的单元测试用例

Question

我正试图弄清楚如何为使用ws库的Websocket服务器编写单元测试用例。我确实浏览了jest-websocket-mock，但我认为这是针对基于浏览器的API的，我想使用jest测试服务器。

基本代码：Server.js

import { createServer } from 'https';
import { WebSocketServer } from 'ws';
import { readFileSync } from 'fs';
const server = createServer({
cert: readFileSync(config.certs.sslCertPath),
key: readFileSync(config.certs.sslKeyPath),
});
const wss = new WebSocketServer({ noServer: true });
server.on('upgrade', (request, socket, head) => {
const origin = request && request.headers && request.headers.origin;
const corsRegex = <regex>;
if (origin.match(corsRegex) != null) {
wss.handleUpgrade(request, socket, head, (ws) => {
wss.emit('connection', ws, request);
});
} else {
socket.destroy();
}
});
wss.on('connection', (ws, req) => {
ws.on('message', (messageg) => {
try {
console.log(message);
} catch (error) {
console.log(error);
}
});
ws.on('close', () => {
console.log('close');
});
ws.on('error', (error) => {
console.log(error);
}); 
});

有人能帮我如何测试原始服务器吗？

Answer 1

您需要在这里创建某种依赖注入机制例如，让我们将所有套接字初始化逻辑移动到一个单独的函数中

function initSocketEvents(wss) {
wss.on('connection', (ws, req) => {
ws.on('message', (messageg) => {
try {
console.log(message);
} catch (error) {
console.log(error);
}
});
ws.on('close', () => {
console.log('close');
});
ws.on('error', (error) => {
console.log(error);
});
});
return wss;
}

现在，在服务器初始化时，只需从不同的文件调用函数

const {initSocketEvents}  = require("./socket-handler") 
const wss = new WebSocketServer({ noServer: true });
initSocketEvents(was);

一切都保持不变，只是现在测试起来容易多了在测试文件

const {initSocketEvents}  = require("./socket-handler") 
const { assert } = require('console');
const { EventEmitter } = require('events');
class MyTestWebSocket extends EventEmitter { }
const mockWSS = new MyTestWebSocket()
initSocketEvents(mockWSS)
mockWSS.emit('connection', mockWSS)
assert(mockWSS.listenerCount('connection')===1)
assert(mockWSS.listenerCount('message')===1)
assert(mockWSS.listenerCount('close')===1)
assert(mockWSS.listenerCount('error')===1)

现在应该可以直接地分离每个侦听器的逻辑，并将其注入函数之外，然后断言所需的逻辑。