什么是mock?
mock在翻译过来有模拟的意思。这里要介绍的mock是辅助单元测试的一个模块。它允许您用模拟对象替换您的系统的部分，并对它们已使用的方式进行断言。
> pip install -U mock
在Python3.x中，mock已经被集成到了unittest单元测试框架中，所以，可以直接使用。
可能你和我初次接触这个概念的时候会有这样的疑问：把要测的东西都模拟掉了还测试什么呢？
但在，实际生产中的项目是非常复杂的，对其进行单元测试的时候，会遇到以下问题：
- 接口的依赖
- 外部接口调用
- 测试环境非常复杂
我们先从最简单例子开始。
modular.py
#modular.py class Count(): def add(self): pass
mock_demo01.py
from unittest import mock import unittest from modular import Count # test Count class class TestCount(unittest.TestCase): def test_add(self): count = Count() count.add = mock.Mock(return_value=13) result = count.add(8,5) self.assertEqual(result,13) if __name__ == ‘__main__‘: unittest.main()
count = Count()
count.add = mock.Mock(return_value=7)
result = count.add(2,5)
接下来，相当于在正常的调用add()方法，传两个参数2和5，然后会得到相加的结果7。然后，7的结果是我们在上一步就预先设定好的。
self.assertEqual(result,7)
最后，通过assertEqual()方法断言，返回的结果是否是预期的结果7。
> python3 mock_demo01.py . ---------------------------------------------------------------------- Ran 1 test in 0.000s OK
这样一个用例就在mock的帮助下编写完成，并且测试通过了。
#module.py class Count(): def add(self, a, b): return a + b
然后，修改测试用例：
from unittest import mock import unittest from module import Count class MockDemo(unittest.TestCase): def test_add(self): count = Count() count.add = mock.Mock(return_value=13, side_effect=count.add) result = count.add(8, 8) print(result) count.add.assert_called_with(8, 8) self.assertEqual(result, 16) if __name__ == ‘__main__‘: unittest.main()
side_effect参数和return_value是相反的。它给mock分配了可替换的结果，覆盖了return_value。简单的说，一个模拟工厂调用将返回side_effect值，而不是return_value。
所以，设置side_effect参数为Count类add()方法，那么return_value的作用失效。
result = count.add(8, 8)
print(result)
这次将会真正的调用add()方法，得到的返回值为16（8+8）。通过print打印结果。
assert_called_with(8,8)
检查mock方法是否获得了正确的参数。
例如，我们要测试A模块，然后A模块依赖于B模块的调用。但是，由于B模块的改变，导致了A模块返回结果的改变，从而使A模块的测试用例失败。其实，对于A模块，以及A模块的用例来说，并没有变化，不应该失败才对。
这个时候就是mock发挥作用的时候了。通过mock模拟掉影响A模块的部分（B模块）。至于mock掉的部分（B模块）应该由其它用例来测试。
# function.py def add_and_multiply(x, y): addition = x + y multiple = multiply(x, y) return (addition, multiple) def multiply(x, y): return x * y
import unittest import function class MyTestCase(unittest.TestCase): def test_add_and_multiply(self): x = 3 y = 5 addition, multiple = function.add_and_multiply(x, y) self.assertEqual(8, addition) self.assertEqual(15, multiple) if __name__ == "__main__": unittest.main()
> python3 func_test.py . ---------------------------------------------------------------------- Ran 1 test in 0.000s OK
目前运行一切正确常，然而，add_and_multiply()函数依赖了multiply()函数的返回值。如果这个时候修改multiply()函数的代码。
…… def multiply(x, y): return x * y + 3
这个时候，multiply()函数返回的结果变成了x*y加3。
再次运行测试：
> python3 func_test.py F ====================================================================== FAIL: test_add_and_multiply (__main__.MyTestCase) ---------------------------------------------------------------------- Traceback (most recent call last): File "fun_test.py", line 19, in test_add_and_multiply self.assertEqual(15, multiple) AssertionError: 15 != 18 ---------------------------------------------------------------------- Ran 1 test in 0.000s FAILED (failures=1)
import unittest from unittest.mock import patch import function class MyTestCase(unittest.TestCase): @patch("function.multiply") def test_add_and_multiply2(self, mock_multiply): x = 3 y = 5 mock_multiply.return_value = 15 addition, multiple = function.add_and_multiply(x, y) mock_multiply.assert_called_once_with(3, 5) self.assertEqual(8, addition) self.assertEqual(15, multiple) if __name__ == "__main__": unittest.main() @patch("function.multiply")
patch()装饰/上下文管理器可以很容易地模拟类或对象在模块测试。在测试过程中，您指定的对象将被替换为一个模拟（或其他对象），并在测试结束时还原。
这里模拟function.py文件中multiply()函数。
def test_add_and_multiply2(self, mock_multiply):
mock_multiply.return_value = 15
设定mock_multiply对象的返回值为固定的15。
ock_multiply.assert_called_once_with(3, 5)
检查ock_multiply方法的参数是否正确。
再次，运行测试用例，通过！
本文摘自虫师：https://www.cnblogs.com/fnng/p/5648247.html
原文：https://www.cnblogs.com/ermm/p/9316014.html