93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
# coding: utf-8
|
||
# -------------------------------------------------------------------
|
||
# yakpanel
|
||
# -------------------------------------------------------------------
|
||
# Copyright (c) 2015-2099 yakpanel(http://www.yakpanel.com) All rights reserved.
|
||
# -------------------------------------------------------------------
|
||
# Author: wzz <wzz@yakpanel.com>
|
||
# -------------------------------------------------------------------
|
||
# ------------------------------
|
||
# Docker模型测试模块 - 容器模型
|
||
# ------------------------------
|
||
import sys
|
||
import unittest
|
||
|
||
if "/www/server/panel" not in sys.path:
|
||
sys.path.insert(0, '/www/server/panel')
|
||
|
||
from mod.project.docker.routetestMod import main as routetest_main
|
||
|
||
routetest = routetest_main()
|
||
|
||
|
||
#
|
||
class TestContainerModel(unittest.TestCase):
|
||
"""
|
||
创建测试用例
|
||
"""
|
||
|
||
def test_returnResult(self):
|
||
"""
|
||
测试模型测试方法，检测返回结果是否为json格式数据
|
||
@return:
|
||
"""
|
||
result = routetest.returnResult({'data': {}})
|
||
self.assertIsInstance(result, dict)
|
||
self.assertIn('status', result)
|
||
self.assertIn('msg', result)
|
||
self.assertIn('data', result)
|
||
self.assertIn('code', result)
|
||
self.assertIn('timestamp', result)
|
||
|
||
def test_wsRequest(self):
|
||
"""
|
||
使用ws长链请求ws://127.0.0.1:8888/ws_mod
|
||
并发送{"mod_name":"docker","sub_mod_name":"routetest","def_name":"wsRequest","ws_callback":"111"}，检测返回结果是否为True
|
||
|
||
备注：请将__init__.py中ws模型路由的comReturn和csrf检查注释掉再测试
|
||
@param get:
|
||
{"mod_name":"docker","sub_mod_name":"routetest","def_name":"wsRequest","ws_callback":"111"}
|
||
{"mod_name":"模型名称","sub_mod_name":"子模块名称","def_name":"函数名称","ws_callback":"ws必传参数，传111",其他参数接后面}
|
||
@return:
|
||
"""
|
||
import json
|
||
import time
|
||
from websocket import create_connection
|
||
ws = create_connection("ws://127.0.0.1:8888/ws_mod")
|
||
print("连接状态：", ws.connected)
|
||
|
||
params = {"mod_name": "docker", "sub_mod_name": "routetest", "def_name": "wsRequest", "ws_callback": "111"}
|
||
ws.send(json.dumps(params))
|
||
|
||
while True:
|
||
result = ws.recv()
|
||
print("接收到结果：", result.strip())
|
||
|
||
try:
|
||
result_data = json.loads(result)
|
||
if "result" in result_data and "callback" in result_data:
|
||
if result_data["result"] == True and result_data["callback"] == "111":
|
||
print("websocket测试成功！")
|
||
break
|
||
except Exception as e:
|
||
pass
|
||
|
||
# 等待一段时间再继续接收消息
|
||
time.sleep(0.1)
|
||
|
||
ws.close()
|
||
self.assertIn('result', result)
|
||
self.assertIn('callback', result)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# unittest.main()
|
||
# 创建测试套件
|
||
suite = unittest.TestSuite()
|
||
suite.addTest(TestContainerModel('test_returnResult'))
|
||
suite.addTest(TestContainerModel('test_wsRequest'))
|
||
|
||
# 创建测试运行器
|
||
runner = unittest.TextTestRunner()
|
||
runner.run(suite)
|