【mock】手把手带你用mock写自定义接口+mock常用语法

在Python中，我们可以使用mock库来模拟接口。mock库提供了多种创建模拟接口的方法，包括模拟函数、模拟类和模拟方法。

1. 使用mock.Mock()创建模拟函数

from unittest import mock
 
# 创建模拟函数
def mock_function():
    return mock.Mock()
 
# 设置返回值
mock_function.return_value = 'Hello, World!'
 
# 调用模拟函数
result = mock_function()
print(result)  # 输出: Hello, World!

2. 使用mock.patch装饰器模拟类中的方法

from unittest import mock
 
class MyClass:
    def my_method(self):
        return 'Original Method'
 
# 使用mock.patch装饰器模拟my_method方法
@mock.patch.object(MyClass, 'my_method')
def test_my_method(mock_method):
    mock_method.return_value = 'Mocked Method'
    my_class_instance = MyClass()
    result = my_class_instance.my_method()
    print(result)  # 输出: Mocked Method
 
test_my_method()

3. 使用mock.MagicMock()创建一个通用的模拟对象

from unittest import mock
 
# 创建通用模拟对象
my_mock = mock.MagicMock()
 
# 设置返回值
my_mock.my_method.return_value = 'Mocked Method'
 
# 调用模拟方法
result = my_mock.my_method()
print(result)  # 输出: Mocked Method

4. 使用mock.create_autospec创建模拟函数或方法，并自动指定签名

from unittest import mock
 
def original_function(param1, param2):
    return param1, param2
 
# 使用create_autospec创建模拟函数
mock_function = mock.create_autospec(original_function)
 
# 设置返回值
mock_function.return_value = 'Mocked Function'
 
# 调用模拟函数
result = mock_function('Hello', 'World')
print(result)  # 输出: Mocked Function

以上示例展示了如何使用mock库创建模拟函数、模拟类中的方法、模拟通用对象及自动指定签名的模拟函数。这些技巧对于测试过程中模拟外部服务或不容易访问的资源非常有用。