Mock测试
Mock测试是什么鬼? 我们常常遇到这样一种场景, 我们测试一些函数, 而这些函数内部调用另外带有副作用的操作, 这可能导致我们在测试过程中对数据造成未知的副作用, 而这并不是我们希望在测试中看到的.
Mock测试可以替换到指定的Python对象或者方法, 并自定义指定对象或者方法的返回值, 从来模拟对象或者方法, 消除副作用.
Mock在Python3.3时加入到标准库中, 2.X版本可以通过pip安装
$ pip install mock
首先任意写一个函数
# -*- coding: utf-8 -*-
#!/usr/bin/env python
import os, sys, time
def foo():
lst = [1]
lst = give_me_five(lst)
return lst
def give_me_five(lst):
return lst * 5
我们希望通过单元测试来测试这个函数的逻辑正确性.
# -*- coding: utf-8 -*-
#!/usr/bin/env python
import os, sys, time
# sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
import unittest
from unittest import TestCase
import mock
import module
class Foo(object):
pass
class TestMock(TestCase):
# 1
def test_method(self):
obj = Foo()
obj.method = mock.MagicMock(return_value=3)
print obj.method
self.assertEqual(obj.method(4), 3)
# 2
@mock.patch('module.foo')
def test_decorator(self, foo):
# res = module.foo()
foo.return_value = [1, 2, 3]
self.assertEqual(foo(), [1, 2, 3])
# 3
def test_with(self):
with mock.patch('module.give_me_five') as give_me_five:
give_me_five.return_value = "I'm Mock"
self.assertEqual(module.foo(), "I'm Mock")
# 4
def test_module(self):
module.give_me_five = mock.Mock(return_value=[1] * 5)
module.give_me_five([1]) # 此时已经变成了一个Mock对象, 并尝试调用
module.give_me_five.assert_called_with([1]) # 对mock的参数进行断言
self.assertEqual(module.foo(), [1] * 5)
if __name__ == '__main__':
unittest.main()
- 我们首先集成TestCase创建了一个单元测试
- # 1位置, 我们通过mock提供的函数给obj的method方法设置返回值(可以看到类中并不包含method方法). 最后通过断言来判断返回值等于我们通过MagicMock设置的返回值
- # 2位置, 我们通过mock提供的装饰器, patch()可以作为函数做装饰, 类装饰器, 上下文管理器 将module中的foo函数给mock掉, 并且并mock的函数生成的Mock对象作为类成员函数参数传入, 指定了foo函数的返回值, 并通过了断言测试
- # 3位置, 将patch()作为一个上下文管理, 关于上下文管理器可以看我另一篇文章Python奇技淫巧, 用法和作为装饰器基本类似
- # 4位置, 我们调用module.foo函数, 而我们并不关系foo()调用了那些函数, 我只关心在成功调用module.give_me_five后, foo函数的逻辑正确性. 所以此次我们通过Mock函数给module.give_me_five指定我们希望的返回值. 这样就能独立的测试module.foo的逻辑
mock的主要思想: 通过mock对象对某些函数进行替换, 对在测试上下文中, 这些被mock的函数被重定向到指定的mock对象
mock还有一些更高级的应用
- MagicMock是Mock的子类, 并且包含一些如__str__一样的黑魔法函数, 使用MagicMock甚至可以mock掉黑魔法函数
- 通过patch.object可以mock掉类中指定的成员函数
- 通过patch.dict可以将对象mock为字典
- 通过patch中的start和stop方法可以控制mock的生效范围, 更加灵活的运行mock测试
最后: 为了回馈铁杆粉丝们,我给大家整理了完整的软件测试视频学习教程,朋友们如果需要可以自行免费领取 【保证100%免费】
全套资料获取方式:点击下方小卡片自行领取即可