python技术栈之单元测试中mock的使用
什么是mock？
mock测试就是在测试过程中，对于某些不容易构造或者不容易获取的对象，用一个虚拟的对象来创建以便测试的测试方法。
mock的作用
特别是开发过程中上下游未完成的工序导致当前无法测试，需要虚拟某些特定对象以便测试。
unittest是python内置的单元测试库，在做接口测试时，如果开发的接口未开发出来，我们如果想要测试接口联调，又不能干等着，这时可以使用unittest.mock模拟接口返回，进行接口测试。
举个栗子
示例：1、新增登录和获取个人信息两个接口，分别由A和B开发。2、A登录接口尚未开发完成，B获取个人信息接口已经开发完成。3、已知登录接口返回3种状态：登录成功，登录失败，登录异常。
场景源码：case.py
-
# -*-coding:utf-8 -*- -
def login(): -
# 登录接口，尚未开发完成 -
# 登录成功返回：{"result": "success", "message": "登录成功"} -
# 登录失败返回：{"result": "fail", "message":"账号或密码错误"} -
# 登录异常返回：{"code": "404", "message": "找不到页面"} -
# message返回失败原因 -
pass -
def get_user_info(): -
# 根据登录的结果success or fail，判断跳转到对应页面 -
result = login() -
print(result) -
try: -
if result["result"] == "success": -
return "登录成功" -
elif result["result"] == "fail": -
return "登录失败" -
else: -
return "未知失败" -
except: -
return "服务端异常"
单元测试用例设计：
-
# -*-coding:utf-8 -*- -
from unittest import mock -
import unittest -
import case -
class TestLogin(unittest.TestCase): -
# 单元测试用例 -
def test_login_success(self): -
# 测试登录成功场景 -
# mock一个支付成功的数据 -
case.login = mock.Mock(return_value={"result": "success", "message": "登录成功"}) -
# 根据支付结果测试页面跳转 -
statues = case.get_user_info() -
self.assertEqual(statues, "登录成功") -
def test_login_fail(self): -
# 测试登录失败场景 -
# mock一个登录失败的数据 -
case.login = mock.Mock(return_value={"result": "fail", "message": "账号或密码错误"}) -
# 根据登录结果测试页面跳转 -
statues = case.get_user_info() -
self.assertEqual(statues, "登录失败") -
def test_login_fail2(self): -
# 测试登录失败场景 -
# mock一个登录失败的数据 -
case.login = mock.Mock(return_value={"result": "", "message": "账号或者密码为空"}) -
# 根据登录结果测试页面跳转 -
statues = case.get_user_info() -
self.assertEqual(statues, "未知失败") -
def test_login_error(self): -
# 测试登录异常场景 -
# mock一个登录异常的数据 -
case.login = mock.Mock(return_value={"code": "404", "message": "找不到页面"}) -
# 根据登录结果测试页面跳转 -
statues = case.get_user_info() -
self.assertEqual(statues, "服务端异常") -
if __name__ == "__main__": -
unittest.main()
执行unittest单元测试用例：
学习安排上
作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。【保证100%免费】
视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取