一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create

这篇具有很好参考价值的文章主要介绍了一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新软件测试全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。
一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection,2024年程序员学习,python,websocket,开发语言

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以添加V获取：vip1024b （备注软件测试）
一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection,2024年程序员学习,python,websocket,开发语言

正文

这个接口很简单，就是完成一个链接，获取到链接后，直接返回对应的消息。

一个很简单的接口，那么python如何来实现webscoket协议接口测试呢，

需要安装两个依赖包

pip3 install websocket-client
pip3 install websocket

安装后，就可以去完成接口的请求。

先创建一个简单的脚本，去测试下接口是否可以调用

from websocket import create_connection
import websocket
url = ‘ws://localhost:8081/ws’
wss = create_connection(url)#创建链接
wss.settimeout(10) #设置超时时间
print(wss.getstatus())#状态码是100认为成功
wss.send(‘hello world’)#发送消息
print(wss.recv())#接受小时
wss.shutdown()

测试下

一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection,2024年程序员学习,python,websocket,开发语言

这样可以正常请求，那么这样一个简单的接口，可以设计一个简单的用例去测试

其实上面的请求测试，可以改造成一个简单的接口测试用例。导入

import unittest

然后定一个一个类，继承

class TestWebsocket(unittest.TestCase):

在setup中

def setUp(self) -> None:
self.url = ‘ws://localhost:8081/ws’
self.wss = create_connection(self.url) # 创建链接

其实就是创建链接

在tearDown中，断开

def tearDown(self) -> None:
self.wss.shutdown()

在去新增一个用例

def testone(self):

用例中有两个断言，一个是链接是否正确，另一个断言是发送的消息后是否返回对应的值

def testone(self):
status=self.wss.getstatus()
self.assertEqual(status,101,‘websocket连接错误’)
self.wss.send(‘hello world’)#发送消息
recvdata=self.wss.recv()
self.assertTrue((“hello world” in recvdata),‘接口返回错误’)

这样一个简单的用例就实现完毕了，我们测试下

一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection,2024年程序员学习,python,websocket,开发语言

执行成功，测试用例可以正常测试，这样一个简单的webscoket的接口的测试就完成了。

其实webscoket接口测试流程就是去创建链接，然后传送数据，接受数据。然后根据链接创建和返回数据进行对应的断言就可以了。

简单的接口测试可以这么来做，那么针对复杂的接口如何做呢，我们去拿之前的写的接口。

@app.websocket(“/ws”)
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
while True:
data = await websocket.receive_text()
await websocket.send_text(f"接受到的消息是: {data}")
manager = ConnectionManager()

给人们消息

@app.websocket(“/ws/{user}/”)
async def websocket_many_point(
websocket: WebSocket,
user: str,
):
await manager.connect(user, websocket)
try:
while True:
data = await websocket.receive_json()
senduser = data.get(‘username’)
if senduser:
await manager.send_other_message_json(data, senduser)
else:
await manager.broadcast_json(data)
except WebSocketDisconnect as e:
manager.disconnect(user, websocket)
@app.websocket(“/ws”)
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
while True:
data = await websocket.receive_text()
await websocket.send_text(f"接受到的消息是: {data}")
manager = ConnectionManager()

给人们消息

很简单，就是一个给多个人或者单个人发消息，这里的用例可以简单的有四个。

测试发给自己看自己能不能接受到，
一个是群发，看两个能否接受到，
一个是给某个人发，看那个人是否接受到。
A发给B，B可以接受到，C接受不到

接着去实现对应的测试用例的代码。

这里面需要注意：数据传输是根据字符串的，我们发送的是json，需要进行json处理。直接使用json模块进行处理即可。

import json
from websocket import create_connection, WebSocketTimeoutException
import unittest
class TestWebSocket(unittest.TestCase):
def setUp(self) -> None:
#初始化链接地址
self.url = ‘ws://localhost:8081/ws/1/’
#链接
self.wss = create_connection(self.url)
def tearDown(self) -> None:
#关闭链接
self.wss.shutdown()
def testnouusernmae(self):
#获取链接状态
status = self.wss.getstatus()
#断言状态码
self.assertEqual(status, 101, ‘websocket连接错误’)
#拼装参数
senddata = {“dict”: 122}
#发送
self.wss.send(json.dumps(senddata))
#看群发是否自己能收到
recvdata = self.wss.recv()
#转化下消息
reslut = json.loads(recvdata)
#断言
self.assertEqual(reslut, senddata, ‘消息发送接受到不一致’)
def testorther(self):
#链接第二个
self.urlorther = ‘ws://localhost:8081/ws/212/’
#创建链接
self.ortherwss = create_connection(self.urlorther)
#获取状态
status = self.wss.getstatus()

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注软件测试）
一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection,2024年程序员学习,python,websocket,开发语言

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！
()

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注软件测试）
[外链图片转存中…(img-S9zhLpTp-1713184121700)]

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！文章来源地址https://www.toymoban.com/news/detail-856204.html

到了这里，关于一文揭秘利用python+unittest实现Webscoket协议接口测试_from websocket import create_connection的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！