最近，我必须使用 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 模块。它在测试中用于临时用模拟或其他值替换给定对象的属性。
争论
- target=classa：我们要修补其属性的对象（通常是一个类）。
- attribute="__init__": 我们要修补的属性的名称。
- 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
希望这有帮助！ :)
