unittest.mock是一个用于在Python中进行单元测试的库,Mock翻译过来就是模拟的意思,顾名思义这个库的主要功能是模拟一些东西。 它的主要功能是使用mock对象替代掉指定的Python对象,以达到模拟对象的行为。
学过python自动化的对unittest并不陌生,unittest其实是单元测试框架, 但对于单元测试,估计很多小伙伴都不懂,单元测试才是自动化测试的至高境界,其中mock是单元测试的脊髓所在
1.python2.x的版本,mock是一个独立的模块,需要用pip安装
pip install -U mock
3.从Python 3.3以后的版本mock已经合并到unittest模块中了,是unittest单元测试的一部分,直接导入过来就行
from unittest import mock
1.如下场景:支付是一个独立的接口,由其它开发提供,根据支付的接口返回状态去显示失败,还是成功,这个是你需要实现的功能
也就是说你写一个b功能,你的同事写一个a功能,你的b功能需要根据a功能的结果去判断,然后实现对应的功能。这就是存在依赖关系,你同事开发的进度你是无法控制的 你要是等他开发完了,你再开发,那你就坐等加班吧.
2.以下是自己写的 zhifu_statues()函数功能,大概设计如下,保存为zhifu.py文件
# 保存为zhifu.py
# coding:utf-8
# 作者:上海-悠悠 QQ交流群:588402570
def zhifu():
'''假设这里是一个支付的功能,未开发完
支付成功返回:{"result": "success", "reason":"null"}
支付失败返回:{"result": "fail", "reason":"余额不足"}
reason返回失败原因
'''
pass
def zhifu_statues():
'''根据支付的结果success or fail,判断跳转到对应页面'''
result = zhifu()
print(result)
try:
if result["result"] == "success":
return "支付成功"
elif result["result"] == "fail":
print("失败原因:%s" % result["reason"])
return "支付失败"
else:
return "未知错误异常"
except:
return "Error, 服务端返回异常!"
3.单元测试用例设计
# coding:utf-8
from unittest import mock
import unittest
import zhifu
# 作者:上海-悠悠 QQ交流群:588402570
class Test_zhifu_statues(unittest.TestCase):
'''单元测试用例'''
def test_01(self):
'''测试支付成功场景'''
# mock一个支付成功的数据
zhifu.zhifu = mock.Mock(return_value={"result": "success", "reason":"null"})
# 根据支付结果测试页面跳转
statues = zhifu.zhifu_statues()
print(statues)
self.assertEqual(statues, "支付成功")
def test_02(self):
'''测试支付失败场景'''
# mock一个支付成功的数据
zhifu.zhifu = mock.Mock(return_value={"result": "fail", "reason": "余额不足"})
# 根据支付结果测试页面跳转
statues = zhifu.zhifu_statues()
self.assertEqual(statues, "支付失败")
if __name__ == "__main__":
unittest.main()
4.单元测试是为了保证函数的每个分支都测到,以上zhifu_statues()函数有四个分支,也就是说要写四个案例,这里我只写了2个,还有2个案例,你们可以自己练习下