【进阶】使用unittest.mock模块进行Mock

# 2.1 Mock对象创建和使用

### 2.1.1 Mock对象创建

在unittest.mock模块中，可以使用`mock.Mock()`函数创建Mock对象。Mock对象是一个模拟真实对象的伪对象，它可以用来模拟真实对象的属性和方法，从而方便测试。

# 创建一个Mock对象
mock_object = mock.Mock()

### 2.1.2 Mock对象方法和属性

Mock对象可以模拟真实对象的属性和方法。可以通过以下方式访问Mock对象的方法和属性：

* **方法：**使用点号(.)运算符访问，例如`mock_object.method_name()`
* **属性：**使用点号(.)运算符或中括号([])运算符访问，例如`mock_object.attribute_name`或`mock_object['attribute_name']`

Mock对象的方法和属性可以被配置为返回特定值或执行特定操作。例如，可以配置`mock_object.method_name()`方法返回`"Hello world"`：

# 配置Mock对象方法返回特定值
mock_object.method_name.return_value = "Hello world"

# 2. unittest.mock模块的基本用法

### 2.1 Mock对象创建和使用

#### 2.1.1 Mock对象创建

在unittest.mock模块中，可以使用`mock.Mock()`函数创建Mock对象。Mock对象是一个模拟真实对象的伪对象，它可以用来模拟真实对象的的行为，并且可以验证对它的调用。

# 创建一个Mock对象
mock_object = mock.Mock()

#### 2.1.2 Mock对象方法和属性

Mock对象具有与真实对象类似的方法和属性。这些方法和属性可以用来模拟真实对象的调用和行为。

**方法：**

Mock对象的方法可以被调用，并且可以返回预先定义的值。

# 设置Mock对象方法的返回值
mock_object.method.return_value = 10

# 调用Mock对象的方法
result = mock_object.method()

# 断言Mock对象的方法被调用
mock_object.method.assert_called_once()

**属性：**

Mock对象还可以具有属性，这些属性可以被读取和设置。

# 设置Mock对象属性的值
mock_object.attribute = 10

# 获取Mock对象属性的值
value = mock_object.attribute

# 断言Mock对象属性被访问
mock_object.attribute.assert_called_once()

### 2.2 Mock对象验证

#### 2.2.1 Mock对象方法调用验证

可以使用`assert_called()`方法来验证Mock对象的方法是否被调用。

# 断言Mock对象方法被调用一次
mock_object.method.assert_called_once()

# 断言Mock对象方法被调用两次
mock_object.method.assert_called_twice()

# 断言Mock对象方法被调用特定次数
mock_object.method.assert_called_with(10)

#### 2.2.2 Mock对象属性验证

可以使用`assert_called()`方法来验证Mock对象属性是否被访问。

# 断言Mock对象属性被访问一次
mock_object.attribute.assert_called_once()

# 断言Mock对象属性被访问两次
mock_object.attribute.assert_called_twice()

# 断言Mock对象属性被访问特定次数
mock_object.attribute.assert_called_with(10)

# 3. unittest.mock模块的高级用法

### 3.1 Mock对象定制

#### 3.1.1 Mock对象方法定制

**方法定制**：允许我们对Mock对象的方法进行自定义，指定其行为和返回值。

**语法**：

mock_object.configure_mock(return_value=None, side_effect=None, **kwargs)

**参数**：

- `return_value`：指定方法的返回值。
- `side_effect`：指定方法的副作用，可以是异常、值或可调用对象。
- `**kwargs`：其他自定义参数，例如`spec`、`spec_set`等。

**示例**：

import unittest
from unittest.mock import Mock

class TestClass(unittest.TestCase):

    def test_mock_method_customization(self):
        mock_object = Mock()

        # 设置方法返回值
        mock_object.configure_mock(return_value=10)
        self.assertEqual(mock_object(), 10)

        # 设置方法副作用
        mock_object.configure_mock(side_effect=ValueError("Error occurred"))
        with self.assertRaises(ValueError):
            mock_object()

#### 3.1.2 Mock对象属性定制

**属性定制**：允许我们对Mock对象属性进行自定义，指定其值或行为。

**语法**：

mock_object.attribute = value

**示例**：

import unittest
from unit