Python中unittest.mock.patch库的基本用法

unittest.mock.patch是Python的mock模块中的一个类，用于在单元测试中临时替换掉指定对象的属性或方法，从而达到对被测试对象的行为进行模拟或控制的目的。它可以用于模拟函数的返回值、模拟类的实例、模拟类方法的行为等。

unittest.mock.patch有几种常用的用法，下面将逐个进行介绍，并附带使用例子。

1. patch装饰器的用法

patch装饰器是unittest.mock.patch提供的一种装饰器语法，它用于临时替换指定对象的属性或方法。下面是一个使用例子：

import unittest
from unittest.mock import patch

def func():
    return 1

class TestPatch(unittest.TestCase):

    @patch('__main__.func')
    def test_patch_decorator(self, mock_func):
        mock_func.return_value = 2
        result = func()
        self.assertEqual(result, 2)

if __name__ == '__main__':
    unittest.main()

在这个例子中，我们定义了一个func函数，并编写了一个使用了patch装饰器的单元测试方法test_patch_decorator。patch装饰器的参数为'__main__.func'，表示我们要替换的是当前模块中的func函数。然后我们将mock_func设为2，然后调用func函数，预期的返回结果为2。运行这个测试方法，预期的断言成功。

2. patch对象的用法

patch对象是unittest.mock.patch返回的一个上下文管理器，用于临时替换指定对象的属性或方法。下面是一个使用例子：

import unittest
from unittest.mock import patch

def func():
    return 1

class TestPatch(unittest.TestCase):

    def test_patch_object(self):
        with patch('__main__.func') as mock_func:
            mock_func.return_value = 2
            result = func()
            self.assertEqual(result, 2)

if __name__ == '__main__':
    unittest.main()

在这个例子中，我们使用with patch语句创建了一个patch对象，将其赋值给mock_func。在该with语句块中，我们将mock_func设为2，然后调用func函数，预期的返回结果为2。运行这个测试方法，预期的断言成功。

3. patch.object的用法

patch.object是unittest.mock.patch的一个类方法，用于临时替换指定对象的属性或方法。下面是一个使用例子：

import unittest
from unittest.mock import patch

class MyClass:
    def method(self):
        return 1

class TestPatch(unittest.TestCase):

    def test_patch_object(self):
        my_obj = MyClass()
        with patch.object(my_obj, 'method') as mock_method:
            mock_method.return_value = 2
            result = my_obj.method()
            self.assertEqual(result, 2)

if __name__ == '__main__':
    unittest.main()

在这个例子中，我们定义了一个类MyClass，它有一个方法method。然后我们创建了一个my_obj实例，并使用with patch.object语句创建了一个patch对象，将其赋值给mock_method。在该with语句块中，我们将mock_method设为2，然后调用my_obj的method方法，预期的返回结果为2。运行这个测试方法，预期的断言成功。

除了上述几种用法外，unittest.mock.patch还有其他一些用法，如patch.dict用于替换字典的键值对、patch.multiple用于替换多个属性或方法等。根据被测试对象的不同，可以选择合适的用法来模拟或控制被测试对象的行为。