python Mock模块

我的理解：

python的mock模块主要通过Mock类实例化出一个对象test_mock，通过test_mock对象设置返回值(return_value)或者返回迭代器(side_effect)，在测试用例运行到执行的Mock函数的时候，返回值使用通过Mock对象设置的返回值。以达到模拟函数返回的作用。

主要使用方法：

1、setUp()初始化测试类的时候，通过Mock()初始化一个类对象，测试函数调用到api.list_names时候会自动返回Mock对象初始化的返回值。达到mock的目的。

import api

class TestCode(TestCase):
    def setUp(self):
        api.list_names = Mock(side_effect=patch_t1)
        api.req_by_api = Mock(side_effect=patch_t2)
        api.get_resource= Mock(side_effect=patch_t3)
        api.recycle_resource = Mock(return_value="")

2、为测试函数添加@mock_path，通过这个装饰器模拟某个函数

    def multiple(a, b):
        return a*b

    class TestProducer(unittest.TestCase):
        def setUp(self):
            self.calculator = Calculator()

        @mock.patch('multiple')
        def test_multiple(self, mock_multiple):
            mock_multiple.return_value = 3
            self.assertEqual(multiple(8, 14), 3)

3、通过@mock.patch.object模拟一个类里面的某个方法

from unittest import mock
import unittest

class Calculator(object):
    def add(self, a, b):
        return a+b


class TestProducer(unittest.TestCase):
    def setUp(self):
        self.calculator = Calculator()

    @mock.patch.object(Calculator, 'add')
    def test_add(self, mock_add):
        mock_add.return_value = 3
        self.assertEqual(Calculator().add(8, 14), 3)

 

https://blog.csdn.net/younger_china/article/details/72904448
https://www.jianshu.com/p/da9a6ffecf8b