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

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

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

解决方案

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

class classa:
    def __init__(self):
        self.class_b = classb()
        self.class_c = classc()
        self.count = 0

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

立即学习“Python免费学习笔记（深入）”；

在我们的例子中，让我们决定 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

希望这有帮助！ :)