mock 简介
mock 原是 python 的第三方库
python3 以后 mock 模块已经整合到了 unittest 测试框架中, 不用再单独安装
Mock 这个词在英语中有模拟的意思, 因此我们可以猜测出这个库的主要功能是模拟一些东西
准确的说, Mock 是 Python 中一个用于支持单元测试的库, 它的主要功能是使用 mock 对象替代掉指定的 Python 对象, 以达到模拟对象的行为
既然 mock 已经被整合到了 unittest 单元测试框架中, 可想而知 mock 的目的就是为了让我们更好的进行测试
mock 作用
1. 解决依赖问题: 当我们测试一个接口或者功能模块的时候, 如果这个接口或者功能模块依赖其他接口或其他模块, 那么如果所依赖的接口或功能模块未开发完毕, 那么我们就可以
使用 mock 模拟被依赖接口, 完成目标接口的测试
2. 单元测试: 如果某个功能未开发完成, 我们又要进行测试用例的代码编写, 我们也可以先模拟这个功能进行测试
3. 模拟复杂业务的接口: 实际工作中如果我们在测试一个接口功能时, 如果这个接口依赖一个非常复杂的接口业务, 那么我们完全可以使用 mock 来模拟这个复杂的业务接口, 其实
这个和解决接口依赖是一样的原理
4. 前后端联调: 如果你是一个前端页面开发, 现在需要开发一个功能: 根据后台返回的状态展示不同的页面, 那么你就需要调用后台的接口, 但是后台接口还未开发完成, 是不是你
就停止这部分工作呢? 答案是否定的, 你完全可以借助 mock 来模拟后台这个接口返回你想要的数据
mock 安装
python 3 的 mock 模块已经被整合到了 unittest 框架中, 所以你使用的时候只需要在文件开头 from unittest import mock 导入即可
如果你使用的是 python2 那么你需要执行 pip install mock 安装后再 import mock 即可
mock 实例
一个未开发完成的功能如何测试?
假如们现在有一个实现两个数相加的功能需要编写测试用例, 但是由于开发进度缓慢, 只搭两个简单的框架, 并没有内部实现
- """
- ------------------------------------
- @Time : 2019/6/26 14:09
- @Auth : linux 超
- @File : ClassFunc.py
- @IDE : PyCharm
- @Motto: Real warriors,dare to face the bleak warning,dare to face the incisive error!
- @QQ : 28174043@qq.com
- @GROUP: 878565760
- ------------------------------------
- """
- import unittest
- from unittest import mock
- class SubClass(object):
- def add(self, a, b):
- """两个数相加"""
- pass
- class TestSub(unittest.TestCase):
- """测试两个数相加用例"""
- def test_sub(self):
- sub = SubClass() # 初始化被测函数类实例
- sub.add = mock.Mock(return_value=10) # mock add 方法 返回 10
- result = sub.add(5, 5) # 调用被测函数
- self.assertEqual(result, 10) # 断言实际结果和预期结果
- if __name__ == '__main__':
- unittest.main()
测试结果
- .
- ----------------------------------------------------------------------
- Ran 1 test in 0.000s
- OK
- Process finished with exit code
测试结果显示, 测试用例执行已经通过
实际上 mock 模拟 add 方法的原理是 使用相同的对象方法接收 mock 的对象 (使用 sub.add 接收), 那么当 mock 对象被调用时(sub.add()) 就会返回 return_value 参数对应的数据
这样一来, 表面看起来就是模拟了 add 方法(这里只是我个人理解, 不对请忽略)
你可以做一个实验, 把用例中的 add 改成别的名字也一样可以测试通过
ok, 继续
我们用例编写完了, 而且开发既然也把功能开发完了(要骂街吗?), 既然真实的功能已经可以测试了, 那么我们怎么在上面用例的基础上直接测试真实功能呢?
完整的功能如何测试?
我们把用例的代码稍做修改
- """
- ------------------------------------
- @Time : 2019/6/26 14:09
- @Auth : linux 超
- @File : ClassFunc.py
- @IDE : PyCharm
- @Motto: Real warriors,dare to face the bleak warning,dare to face the incisive error!
- @QQ : 28174043@qq.com
- @GROUP: 878565760
- ------------------------------------
- """
- import unittest
- from unittest import mock
- class SubClass(object):
- def add(self, a, b):
- """两个数相加"""
- return a + b
- class TestSub(unittest.TestCase):
- """测试两个数相加"""
- def test_sub(self):
- sub = SubClass() # 初始化被测函数类实例
- sub.add = mock.Mock(return_value=10, side_effect=sub.add) # 传递 side_effect 关键字参数, 会覆盖 return_value 参数值, 使用真实的 add 方法测试
- result = sub.add(5, 11) # 真正的调用被测函数
- self.assertEqual(result, 16) # 断言实际结果和预期结果
- if __name__ == '__main__':
- unittest.main()
side_effect 参数
代码中我们给 Mock 方法添加了另一个关键字参数 side_effect = sub.add, 这个参数和 return_value 正好相反, 当传递这个参数的时候 return_value 参数就会失效
而 side_effect 生效, 这里我给的参数值是 sub.add 相当于 add 方法的地址, 那么当调用 add 方法时就会真实的使用 add 方法, 也就达到了我们测试实际的 add 方法.
你也可以理解为当传递了 side_effect 参数且值为被测方法地址时, mock 就不会起作用
side_effect 接收的是一个可迭代序列, 当传递多个值时, 那么每次调用 mock 时会返回不同的值
- mock_obj = mock.Mock(side_effect= [1,2,3])
- print(mock_obj())
- print(mock_obj())
- print(mock_obj())
- print(mock_obj())
输出
- Traceback (most recent call last):
- 1
- File "D:/MyThreading/mymock.py", line 37, in <module>
- 2
- print(mock_obj())
- 3
- File "C:\Python36\lib\unittest\mock.py", line 939, in __call__
- return _mock_self._mock_call(*args, **kwargs)
- File "C:\Python36\lib\unittest\mock.py", line 998, in _mock_call
- result = next(effect)
- StopIteration
- Process finished with exit code 1
当所有值被取完后就会报错(这个地方有点类似生成器的原理)
存在依赖关系的功能如何测试?
假设有这样一个场景: 我们要测试一个支付接口但是这个支付接口又依赖一个第三方支付接口, 那么第三方支付接口我们暂时没有权限使用, 那么我们该如何测试我们自己这个接口呢?
看下面的实例
假设第三方接口和我们自己的支付接口如下
- """
- ------------------------------------
- @Time : 2019/6/26 15:09
- @Auth : linux 超
- @File : PayMent.py
- @IDE : PyCharm
- @Motto: Real warriors,dare to face the bleak warning,dare to face the incisive error!
- @QQ : 28174043@qq.com
- @GROUP: 878565760
- ------------------------------------
- """
- import requests
- class PayApi(object):
- @staticmethod
- def auth(card, amount):
- """
- 第三方支付接口
- :param card: 卡号
- :param amount: 支付金额
- :return:
- """ pay_url ="http://www.zhifubao.com" # 第三方支付接口地址
- data = {"card": card, "amount": amount}
- response = requests.post(pay_url, data=data) # 请求第三方支付接口
- return response # 返回状态码
- def pay(self, user_id, card, amount):
- """
- 我们自己的支付接口
- :param user_id: 用户 id
- :param card: 卡号
- :param amount: 支付金额
- :return:
- """
- # 调用第三方支付接口
- response = self.auth(card, amount)
- try:
- if response['status_code'] == '200':
- print('用户 {} 支付金额 {} 成功'.format(user_id, amount))
- return '支付成功'
- elif response['status_code'] == '500':
- print('用户 {} 支付失败, 金额不变'.format(user_id))
- return '支付失败'
- else:
- return '未知错误'
- except Exception:
- return "Error, 服务器异常!"
- if __name__ == '__main__':
- pass
很明显第三方支付接口是无法访问的, 因为接口的地址是我 DIY 的, 为了模拟实际中我们无法使用的第三方支付接口
编写测试用例
- """
- ------------------------------------
- @Time : 2019/6/26 15:22
- @Auth : linux 超
- @File : testpay.py
- @IDE : PyCharm
- @Motto: Real warriors,dare to face the bleak warning,dare to face the incisive error!
- @QQ : 28174043@qq.com
- @GROUP: 878565760
- ------------------------------------
- """
- import unittest
- from unittest import mock
- from payment.PayMent import PayApi
- class TestPayApi(unittest.TestCase):
- def test_success(self):
- pay = PayApi()
- pay.auth = mock.Mock(return_value={'status_code':'200'})
- status = pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '支付成功')
- def test_fail(self):
- pay = PayApi()
- pay.auth = mock.Mock(return_value={'status_code':'500'})
- status = pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '支付失败')
- def test_error(self):
- pay = PayApi()
- pay.auth = mock.Mock(return_value={'status_code':'300'})
- status = pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '未知错误')
- def test_exception(self):
- pay = PayApi()
- pay.auth = mock.Mock(return_value='200')
- status = pay.pay('1000', '12345', '10000')
- self.assertEqual(status, 'Error, 服务器异常!')
- if __name__ == '__main__':
- unittest.main()
测试输出结果
.... 用户 1000 支付失败, 金额不变
用户 1000 支付金额 10000 成功
- ----------------------------------------------------------------------
- Ran 4 tests in 0.001s
- OK
- Process finished with exit code 0
从执行结果可以看出, 即使第三方支付接口无法使用, 但是我们自己的支付接口仍然测试通过了
也许有人会问, 第三方支付都不能用, 我们的测试结果是否是有效的呢?
通常我们在测试一个模块的时候, 我们是可以认为其他模块的功能是正常的, 只针对目标模块进行测试是没有任何问题的, 所以说测试结果也是正确的
其实上述代码还可以使用另一种方式来写
mock 对象的方法
- """
- ------------------------------------
- @Time : 2019/6/26 15:22
- @Auth : linux 超
- @File : testpay.py
- @IDE : PyCharm
- @Motto: Real warriors,dare to face the bleak warning,dare to face the incisive error!
- @QQ : 28174043@qq.com
- @GROUP: 878565760
- ------------------------------------
- """
- import unittest
- from unittest import mock
- from unittest.mock import patch
- from payment.PayMent import PayApi
- class TestPayApi(unittest.TestCase):
- def setUp(self):
- self.pay = PayApi()
- @patch.object(PayApi, 'auth')
- def test_success(self, mock_auth):
- mock_auth.return_value = {'status_code':'200'}
- status = self.pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '支付成功')
- @patch.object(PayApi, 'auth')
- def test_fail(self, mock_auth):
- mock_auth.return_value={'status_code':'500'}
- status = self.pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '支付失败')
- @patch.object(PayApi, 'auth')
- def test_error(self, mock_auth):
- mock_auth.return_value={'status_code':'300'}
- status = self.pay.pay('1000', '12345', '10000')
- self.assertEqual(status, '未知错误')
- @patch.object(PayApi, 'auth')
- def test_exception(self, mock_auth):
- mock_auth.return_value='200'
- status = self.pay.pay('1000', '12345', '10000')
- self.assertEqual(status, 'Error, 服务器异常!')
- if __name__ == '__main__':
- unittest.main()
还有 mock 一个普通函数, mock 多个方法等, 这里先不赘述, 写法和上面实例差不多
最后
mock 还有很多自带的功能方法
且 mock 功能很强大, 也不是一句两句话就能说完了, 本篇文章主要介绍了 mock 的基本使用方法, 甚是简单, 对于实际中如何应用, 如何掌握更强大的方法还需自己慢慢摸索
来源: https://www.cnblogs.com/linuxchao/p/linuxchao-mock.html