本文共 2400 字，大约阅读时间需要 8 分钟。

什么是mock?

mock在翻译过来有模拟的意思。这里要介绍的mock是辅助单元测试的一个模块。它允许您用模拟对象替换您的系统的部分，并对它们已使用的方式进行断言。

在Python2.x 中 mock是一个单独模块，需要单独安装。

> pip install -U mock

在Python3.x中，mock已经被集成到了unittest单元测试框架中，所以，可以直接使用。

什么情况下使用mock

在项目的单元测试过程中，会遇到：

1、接口的依赖 2、外部接口调用 3、测试环境非常复杂

单元测试应该只针对当前单元进行测试, 所有的内部或外部的依赖应该是稳定的, 已经在别处进行测试过的.使用mock 就可以对外部依赖组件实现进行模拟并且替换掉, 从而使得单元测试将焦点只放在当前的单元功能。

解决测试依赖

例如，我们要测试A模块，然后A模块依赖于B模块的调用。但是，由于B模块的改变，导致了A模块返回结果的改变，从而使A模块的测试用例失败。其实，对于A模块，以及A模块的用例来说，并没有变化，不应该失败才对。

这个时候就是mock发挥作用的时候了。通过mock模拟掉影响A模块的部分（B模块）。至于mock掉的部分（B模块）应该由其它用例来测试。

# function.pydef add_and_multiply(x, y):    addition = x + y    multiple = multiply(x, y)    return (addition, multiple)def multiply(x, y):    return x * y

然后，针对 add_and_multiply()函数编写测试用例。func_test.py

import unittestimport functionclass 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.000sOK

目前运行一切正确常，然而，add_and_multiply()函数依赖了multiply()函数的返回值。如果这个时候修改multiply()函数的代码。

def multiply(x, y):    return x * y + 3

这个时候，multiply()函数返回的结果变成了x*y加3。再次运行测试，则会报错。然而，add_and_multiply()函数以及它的测试用例并没有做任何修改，罪魁祸首是multiply()函数引起的，我们应该把 multiply()函数mock掉。

import unittestfrom unittest.mock import patchimport functionclass 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()函数（对象）重命名为 mock_multiply对象。

mock_multiply.return_value = 15

设定mock_multiply对象的返回值为固定的15。

ock_multiply.assert_called_once_with(3, 5)

检查ock_multiply方法的参数是否正确。

再次，运行测试用例，通过！

作者：Root_123 链接：https://www.jianshu.com/p/7c4c9c68373f 来源：简书 简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。