Python单测各种模拟的使用
相关包
from unittest import mock
from unittest.mock import patch
from unittest.mock import MagicMock
模拟类方法
同《模拟函数》
模拟requests
requests.get = MagicMock(return_value=MagicMock(json=MagicMock(return_value={"data": "日志"})))
模拟变量
根据变量所在的包路径引入
with patch("path1.path2.变量名", "变量值"):
resp_data = self.handler.get_data("other", "scenelog 2023-05-06")
\# 断言返回值
self.assertEqual(resp_data, "")
模拟函数
可以通过装饰器模拟，多个模拟多行，参数顺序为装饰器从下到上
class TestTegcloudOwnScenePullHandle(object):
@mock.patch.object(HttpRequestHandle, "only_request")
@mock.patch.object(TegcloudOwnScenePullHandler, "get_dimensions")
def test_get_data_err(self, ins_get_dimensions, ins_only_request):
ins_get_dimensions.return_value = (0, [])
ins_only_request.return_value = (
404, "接口异常"
)
模拟aiohttp请求
目前GET请求是可以的，但是如果想模拟POST并携带body目前还没发现可行方案（make_mocked_request 没有body等参数）
from aiohttp.test_utils import make_mocked_request
class TestViewDataHandler(unittest.TestCase):
def test_get_health_res(self, mock_verify_url):
# 构造请求参数
request = make_mocked_request('GET', '/health',
headers={'msg_signature': 'signature'})
Pytest mock相同函数
class TestXXXclass(unittest.TestCase):
def test_func(self, scene_map_db_ins):
scene_map_db_ins.side_effect = [[{"xx": "123", "id": 21}],
[{"xx": "124", "id": 20}]]
Mock 异常
# 调用方法并捕获异常，在with内调用方法
with self.assertRaises(KeyError):
ins = test1(a=["123"],
b="2023-10-12 12:00:00",
c="2023-10-22 13:00:00",
d=s,
e=False)