在前端开发过程中，我们经常需要使用测试来验证代码是否正确、性能是否良好以及是否能满足预期的需求。Mocha 是一个流行的 JavaScript 测试框架，而 supertest 则是一个方便的工具，可以帮助我们在 Mocha 测试中测试 WebSocket 服务。
WebSocket 是什么？
WebSocket 是一种基于 TCP 的协议，用于实现双向通信。与 HTTP 协议不同，WebSocket 使得客户端和服务器之间可以开放一个持久的双向连接，而无需先在客户端发起请求。因此，WebSocket 更加适合实时通信。
如何使用 supertest 测试 WebSocket 服务
- 安装依赖
首先，我们需要安装 Mocha 和 supertest。
npm install --save-dev mocha supertest
- 设置测试环境
在开始测试 WebSocket 服务之前，我们需要设置测试环境。首先，在项目根目录下，创建一个名为 test 的文件夹。然后，在该文件夹下创建一个名为 app.js 的文件，用于创建 Web 服务器。
-- -------------------- ---- ------- ----- ---- - ---------------- ----- ------ - ----------------------- ---- -- - ----------------------------- -------------- -------------- --------- --- ------------------- -- -- - ------------------- ------- -- ---- ------- --- -------------- - -------
接着，在该文件夹下创建一个名为 websocket.js 的文件，用于创建 WebSocket 服务器。
const WebSocket = require('ws');
const wss = new WebSocket.Server({ port: 8081 });
wss.on('connection', (ws) => {
ws.send('Greetings from WebSocket server!');
});最后，在 package.json 文件中，设置测试执行命令。
{
"scripts": {
"test": "mocha"
}
}- 编写测试用例
现在，我们可以开始编写测试用例了。在 test 文件夹下，创建一个名为 websocket.test.js 的文件，用于测试 WebSocket 服务器。
-- -------------------- ---- -------
----- ------- - ---------------------
----- --------- - --------------
----- --- - -----------------
----- --- - -----------------------
------------------- -------- -- -- -
---------- ---- - -------- ------- -- --------- ------ -- -
----- -- - --- ---------------------------------
---------------- ------ -- -
----- ------- - -----------------
-- ------------- --- ----------- -
-----------
-------
-
---
---
---在该测试用例中，我们创建了一个 WebSocket 客户端，连接到 WebSocket 服务器，并等待服务器发送一条问候消息。如果收到了该消息，说明测试通过。
- 运行测试用例
运行测试用例，确保一切都能正常工作。
npm test
总结
supertest 是一个非常方便的工具，可以帮助我们在 Mocha 测试中测试 WebSocket 服务。WebSocket 是一种用于实现双向通信的协议，能更好地满足实时通信需求。通过本文的学习，我们可以更好地了解如何使用 supertest 测试 WebSocket 服务。如有疑问请讨论。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/65a36c00add4f0e0ffb901f5