【Mock】mock基础、简单的单元测试代码练习。
说到接口测试,必问 mock,mock 通俗一点来说就是模拟接口返回。解决接口的依赖关系,主要是为了解耦,单元测试用的多。
什么是Mock
unittest.mock 是一个用于在 Python 中进行单元测试的库,Mock 翻译过来就是模拟的意思,顾名思义这个库的主要功能是模拟一些东西。它的主要功能是使用 mock 对象替代掉指定的 Python 对象,以达到模拟对象的行为。
学过 python 自动化的对 unittest 并不陌生,unittest 其实是单元测试框架, 但对于单元测试,估计很多小伙伴都不懂,单元测试才是自动化测试的至高境界,其中 mock 是单元测试的脊髓所在。
mock 能做什么
1.前后端联调,如果你是一个前端页面开发,现在需要开发一个功能:下一个订单,支付页面的接口,根据支付结果,支付成功,展示支付成功页,支付失败,展示支付失败页。要完成此功能,你需要调用后端的接口,根据返回给你的结果,来展示不同的页面。此时后端接口还没开发好,作为一个前端开发总不能等别人开发好了,你再开发,那你只有加班的命了。为了同步开发完成任务,此时,你可以根据接口文档的规定,把接口的地址和入参传过去,然后自己 mock 接口的不同返回界面,来完成前端的开发任务
2.单元测试,单元测试的目的是测试某个小小单元的功能,但现实中开发的函数或方法都是有依赖关系的,比如 b 函数的参数,需要调用 a 函数返回结果,但是我前面已经测试 a 函数了。这种情况下,就不需要再测一次 a 函数了,此时就可以用 mock 模块来模拟调用这部分内容,并给出返回结果
3.第三方接口依赖,在做接口自动化的时候,有时候需要调用第三方的接口,但是别人公司的接口服务不受你的控制,有可能别人提供的测试环境今天服务给你开着,别人就关掉了,给自动化接口测试带来很多的麻烦,此时就可以自己写一个 mock-server来模拟接口的返回数据。
mock 环境准备
1.python2.x 的版本,mock 是一个独立的模块,需要用 pip 安装
pip install -U mock
3.从 Python 3.3 以后的版本 mock 已经合并到 unittest 模块中了,是 unittest单元测试的一部分,直接导入过来就行
from unittest import mock
PS:我用的是pycharm运行的下面的代码,在pycharm中,没有安装上unittest(但是后来发现运行代码并不影响),直接安装了mock,在安装mock之前需要先升级pip到19以上的版本才行。
单元测试场景(依赖关系)
1.如下场景:支付是一个独立的接口,由其它开发提供,根据支付的接口返回状态去显示失败,还是成功,这个是你需要实现的功能
也就是说你写一个 b 功能,你的同事写一个 a 功能,你的 b 功能需要根据 a 功能的结果去判断,然后实现对应的功能。这就是存在依赖关系,你同事开发的进度你是无法控制的你要是等他开发完了,你再开发,那你就坐等加班吧.
2.以下是自己写的 zhifu_statues()函数功能,大概设计如下,保存为temple.py 文件。
#保存为temple.py #coding:utf-8 def zhifu(): '''假设这里是一个支付的功能,未开发完 支付成功返回:{"result": "success", "reason":"null"} 支付失败返回:{"result": "fail", "reason":"余额不足"} reason 返回失败原因 ''' pass def zhifu_statues(): '''根据支付的结果 or 判断跳转到对应页面''' result = zhifu() print(result) try: if result["result"] == "success": return "支付成功" elif result["result"] == "fail": print("失败原因:%s" % result["reason"]) return "支付失败" else: return "未知错误异常" except: return "Error,服务端返回异常"
以下为单元测试用例设计:
#保存为unittest_cases.py
# coding:utf-8
from unittest import mock import unittest import temple class Test_zhifu_statues(unittest.TestCase): '''单元测试用例''' def test_01(self): '''测试支付成功场景''' #mock一个支付成功数据 temple.zhifu = mock.Mock(return_value={"result": "success", "reason": "null"}) #根据支付结果测试页面跳转 statues = temple.zhifu_statues() print(statues) self.assertEqual(statues, "支付成功") def test_02(self): '''测试支付失败场景''' #mock一个支付失败的数据 temple.zhifu = mock.Mock(return_value={"result": "fail", "reason": "余额不足"}) #根据支付结果测试页面跳转 statues = temple.zhifu_statues() print(statues) self.assertEqual(statues,'支付失败') def test_03(self): temple.zhifu = mock.Mock(return_value={"result": "1", "reason": "未知错误"}) statues = temple.zhifu_statues() print(statues) self.assertEqual(statues,'未知错误异常') if __name__=="__main__": unittest.main()
运行效果截图: