【进阶】pytest-mock插件的使用

# 2.1 mock对象的基本用法

`mock` 对象提供了多种方法来模拟对象的行为，最基本的方法是使用 `mock.Mock()` 函数创建一个模拟对象。这个模拟对象可以被用来替换任何对象，并可以被配置为返回特定的值或引发特定的异常。

例如，以下代码创建一个模拟对象来模拟 `os.path.exists` 函数：

import os
from unittest.mock import Mock

# 创建一个模拟对象来模拟 os.path.exists 函数
mock_exists = Mock(return_value=True)

# 使用模拟对象来替换 os.path.exists 函数
os.path.exists = mock_exists

# 调用 os.path.exists 函数，它将返回模拟值 True
result = os.path.exists('/tmp/test.txt')
assert result == True

在这个例子中，`mock_exists` 模拟对象被配置为返回 `True`，因此当 `os.path.exists('/tmp/test.txt')` 被调用时，它将返回 `True`。

# 2. mock对象的使用

### 2.1 mock对象的基本用法

#### 创建mock对象

使用`mock.Mock()`函数创建mock对象，它可以模拟任何对象。

import unittest
from unittest.mock import Mock

class MyClass:
    def my_method(self):
        pass

# 创建一个MyClass的mock对象
mock_object = Mock(spec=MyClass)

#### 设置mock对象的方法和属性

使用`mock_object.configure_mock()`方法设置mock对象的方法和属性。

# 设置mock_object的my_method方法返回"Hello, world!"
mock_object.configure_mock(my_method=Mock(return_value="Hello, world!"))

# 设置mock_object的my_attribute属性为10
mock_object.configure_mock(my_attribute=10)

#### 调用mock对象的方法和属性

调用mock对象的方法和属性就像调用真实对象一样。

# 调用mock_object的my_method方法
result = mock_object.my_method()

# 获取mock_object的my_attribute属性
value = mock_object.my_attribute

### 2.2 mock对象的高级用法

#### 2.2.1 mock对象的方法和属性

mock对象的方法和属性可以进行更高级的配置。

* **side_effect：**指定方法调用的副作用，可以是异常、值或其他函数。
* **assert_called_once_with：**断言方法只被调用了一次，并且传入的参数与给定的参数匹配。
* **assert_called_with：**断言方法被调用，并且传入的参数与给定的参数匹配。

#### 2.2.2 mock对象与上下文管理器的结合

mock对象可以与上下文管理器结合使用，以在特定作用域内自动重置mock对象。

with mock.patch('my_module.my_function') as mock_function:
    # 在此作用域内，my_module.my_function将被mock_function替换
    pass

# 作用域结束后，my_module.my_function将恢复为原始函数

# 3.1 mock调用行为的断言

在测试中，断言是用来验证实际结果与预期结果是否一致的手段。pytest-mock插件提供了多种断言方法来验证mock对象的调用行为是否符合预期。

#### assert_called_once()

`assert_called_once()`断言mock对象被调用了一次，并且只调用了一次。如果mock对象被调用了多次或没有被调用，断言将失败。