Python单元测试之mock使用

深藏阁楼爱情的钟 2022-10-19 05:43 214阅读 0赞

![9c5838fd5f1b74eb6357e9843bb78dac.png][]

> 什么是mock？比如我们需要用一个接口时，这个接口还没有实现或者依赖第三方服务，为了保证当前功能的开发和测试，就要使用mock模拟接口的功能。

Python中使用mock对象替代指定的Python对象，实现控制模拟对象的行为。mock模块在Python 3.3以后合并到unittest模块中了，可以直接通过导入使用。

## Mock基本使用 ##

Mock对象就是mock模块中的一个类的实例，创建后，可以指定返回值并设置所需的属性，也可以断言调用了哪些方法/属性及其参数。

`class Mock(spec=None, side_effect=None, return_value=DEFAULT, wraps=None, name=None, spec_set=None, **kwargs)`

**Mock类主要的几个参数：**

*  **name**:命名一个mock对象，只是起到标识作用，可以通过print查看。
 *  **return\_value**: 定义mock方法的返回值，可以指定一个值（或者对象），当mock对象被调用时，返回**return\_value**指定的值。
 *  **side\_effect**: 这个参数指向一个可调用对象，接收一个可迭代序列。可以抛出异常或者动态改变值。当传递这个参数的时候return\_value 参数就会失效。

from unittest import mock
    result1 = mock.Mock(name='mock名称')
    print(result1)
    mock_value1 = mock.Mock(return_value="返回值1")
    print(mock_value1())
    mock_value2 = mock.Mock(return_value="返回值2",side_effect= [1,2,3])
    print(mock_value2())
    print(mock_value2())
    print(mock_value2())

![c20a24310d26bdee6ba49776d2fcd52c.png][]

![c4f103e3032f656ada7d3fb7a6ce8546.png][]

Mock 步骤如下：

*  导入 unittest 框架中的 mock
 *  找到要替换的对象A，可以是一个类、函数或者类实例
 *  实例化mock对象，设置mock对象的行为，比如调用的时候返回的值，被访问成员的时候返回什么值等。
 *  使用mock对象替换对象A
 *  调用并断言

mock一个未开发的接口

![2fd299ae5bc6f03e17a81524289b45fc.png][]

![72db1dae5370c8eee246711bbe81970a.png][]

mock一个依赖关系的功能

实际工作中，我们也会遇到这样的场景，测试A模块，然后A模块依赖于B模块的调用，这时就可以借助mock在单元测试中分别测试正常返回和异常返回的情况。

**下面是一个访问baidu的功能，visit\_baidu()方法依赖send\_request的返回结果。**

import requests
    
    
    def send_request(url):
        r = requests.get(url)
        return r.status_code
    
    
    def visit_baidu():
        url = 'http://www.baidu.com'
        return send_request(url)

**通过mock，在单元测试中分别测试正常返回和异常返回的情况**

from unittest import mock
    import unittest
    import demo
    
    
    class TestReq(unittest.TestCase):
        def test_request_01(self):
            # 实例化mock对象，指定返回值，替换原有对象
            demo.send_request = mock.Mock(return_value='200')
            print(demo.send_request())
            self.assertEqual(demo.visit_baidu(), '200')
    
        def test_request_02(self):
            # 实例化mock对象，指定返回值，替换原有对象
            demo.send_request = mock.Mock(return_value='404')
            print(demo.send_request())
            self.assertEqual(demo.visit_baidu(), '404')
    
    
    if __name__ == '__main__':
        unittest.main(verbosity=2)

![d8e38630a742782d6ae726ea7ca520a3.png][]

## Mock的高级用法 ##

mock库提供了patch函数来简化mock对象对原对象的替换，该函数会返回一个mock内部的类实例，它可以控制mock的范围，可以作为装饰器或者上下文管理器使用。

`mock.patch（target，new = DEFAULT，spec = None，create = False，spec_set = None，autospec = None，new_callable = None，** kwargs ）`

**mock装饰器使用格式**

*  @patch("module名字.方法名")
 *  @patch.object(类名, "方法名")

patch作为装饰器，需要把你想模拟的函数写在里面，然后在后面的单元测试案例中为它赋一个具体实例，再用return\_value 来指定模拟函数返回的结果。

改造上面的单元测试：

from unittest import mock
    import unittest
    import demo
    
    
    class TestReq(unittest.TestCase):
    #在测试的参数里对该Mock对象设置一个参数
        @mock.patch("demo.send_request")
        def test_request_01(self,mock_request):
      # 指定一个返回值
            mock_request.return_value='200'
            self.assertEqual(demo.visit_baidu(), '200')
    
        @mock.patch("demo.send_request")
        def test_request_02(self,mock_request):
      # 指定一个返回值
            mock_request.return_value='404'
            self.assertEqual(demo.visit_baidu(), '404')
    
    
    if __name__ == '__main__':
        unittest.main(verbosity=2)

更多mock方法可以参考官方文档进行学习https://docs.python.org/zh-cn/dev/library/unittest.mock.html

参考：搜狗测试《控制你的数据——Python mock的基本使用》

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1hpbmdMb25nU0tZ_size_16_color_FFFFFF_t_70][]

[9c5838fd5f1b74eb6357e9843bb78dac.png]: /images/20221005/8efb53c943a9411597269887765e14ac.png
[c20a24310d26bdee6ba49776d2fcd52c.png]: /images/20221005/fa1663de33d440b98247a7fec2780afb.png
[c4f103e3032f656ada7d3fb7a6ce8546.png]: /images/20221005/3ed27f0bc1694d29955df00d03f6ca7e.png
[2fd299ae5bc6f03e17a81524289b45fc.png]: /images/20221005/7326d8692fd04578a75d1a7008b249b1.png
[72db1dae5370c8eee246711bbe81970a.png]: /images/20221005/3c0e9e07a5df48e1999f5d56ecb293c9.png
[d8e38630a742782d6ae726ea7ca520a3.png]: /images/20221005/c71b0a8d503a473bac36fa62bc9cdeb5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L1hpbmdMb25nU0tZ_size_16_color_FFFFFF_t_70]: /images/20221005/ca37ed66541642cf8c90d533b4281511.png