最近，我不得不使用 Pytest 为 Python 模块编写单元测试。该模块包含一个类，其他类在其构造函数中初始化。

像往常一样，我为此类创建了一个固定装置，以便轻松为每个类方法编写测试。此时，当我尝试模拟构造函数中启动的不同类时，我遇到了一些问题。模拟不起作用，这些类的实例仍在创建中。

经过一些研究并结合我在网上找到的一些不同的解决方案，我想分享我如何设法模拟课程。

解决方案

这是我尝试模拟的类的示例：

class ClassA:
    def __init__(self):
        self.class_b = ClassB()
        self.class_c = ClassC()
        self.count = 0

我们希望在测试期间为此类的每个字段设置一个值。该值可以是 None 或类模拟，但我们不希望启动 ClassB 和 ClassC 类。

在我们的例子中，让我们决定 self.class_b 和 self.class_c 应该是模拟：

@pytest.fixture
def mock_class_b():
    class_b = Mock(spec=ClassB)
    return class_b

@pytest.fixture
def mock_class_c():
    class_c = Mock(spec=ClassC)
    return class_c

因此，此类符合我们目标的固定装置如下所示：

@pytest.fixture
def class_a_mock(mock_class_b, mock_class_c):
    with patch.object(target=ClassA, attribute="__init__", return_value=None) as mock_init:
        class_a = ClassA()
        class_a.class_b = mock_class_b
        class_a.class_c = mock_class_c
        class_b.count = 0
        return class_a

重要的部分是如何使用patch.object函数，该函数来自Python中的unittest.mock模块。它在测试中用于临时用模拟或其他值替换给定对象的属性。

参数

1. target=ClassA：我们要修补其属性的对象（通常是类）。
2. attribute="__init__": 我们要修补的属性的名称。
3. return_value=None：用不执行任何操作的函数替换 __init__ 方法

通过这种方式，我们可以在我们的装置中创建模拟变量。
阅读有关 patch.object

测试技巧

我写这个教程是为了这种情况，无论出于何种原因，我们都无法更改A类的代码。但是，我通常建议如果可能的话修改代码，不是为了改变逻辑，而是为了使其更易于测试.

以下是如何修改 A 类以使其更易于测试的一些示例：

选项 1： 将 B 类和 C 类的实例作为参数传递。
这样，当我们编写固定装置时，我们可以传递模拟而不是实例。

class ClassA:
    def __init__(self, class_b_instance, class_c_instance):
        self.class_b = class_b_instance
        self.class_c = class_c_instance
        self.count = 0

选项 2： 创建一个指示测试模式的布尔变量。
这样我们就可以决定A类的哪些字段在启动时会或不会获取值。

class ClassA:
    def __init__(self, test_mode=False):
        if not test_mode:
            self.class_b = ClassB()
            self.class_c = ClassC()
        self.count = 0

选项 3： 在单独的方法中进行类启动。
这种方法使我们可以选择避免在测试模块中调用 set_class_variables。

class ClassA:
    def __init__(self):
        self.class_b = None
        self.class_c = None
        self.count = None

    def set_class_variables(self):
        self.class_b = ClassB()
        self.class_c = ClassC()
        self.count = 0

希望这有帮助！ :)