如何使用 pytest 设备优雅地模拟一个类？

Question

假设我有一个fixture 来模拟一个使用monkeypatch 的类。

# conftest.py
@pytest.fixture
def mock_dummyclass(monkeypatch):
    def mock_function():
        return None
 
    monkeypatch.setattr(dummypackage, "dummyclass", mock_function)

现在，我必须在我的测试中使用这个夹具，以便模拟这个类。

#test_dummy.py
@pytest.mark.usefixtures("mock_dummyclass")
class TestManyDummyMethods:
    def test_dummy_one():
        import dummypackage  # Flag A
    def test_dummy_two():
        import dummypackage  # Flag B
    ...
    def test_dummy_n():
        import dummypackage  # Flag N

正如您在标记行中看到的，我必须在每个函数中导​​入此模块dummypackage，以确保在导入模块之前应用夹具（否则夹具将无效）。

在类中导入dummypackage 会起作用，但在函数中调用self.dummypackage 似乎也不是很优雅。

有没有更优雅的方法来实现这一点？

Answer 1

评论：monkeypatch 库似乎不再维护。 unittest.mock 应该能满足你的所有需求。

我会尽量避免依赖于将模块作为测试的一部分导入的解决方案，因为如果出于其他原因（例如导入其他函数）导入它会中断。

我以os 为例，因为它存在并且可以重现。

如何最好地修补似乎取决于您如何从另一个模块导入。

示例1：target_module1.py（从os.path导入join方法）：

from os.path import join

def some_method_using_join():
    return join('parent', 'child')

这需要我们修补target_module1.py的join方法：

target_module1_test.py:

from unittest.mock import patch, MagicMock

import pytest

from target_module1 import some_method_using_join

@patch('target_module1.join')
def some_test(join_mock: MagicMock):
    some_method_using_join()

示例2：target_module2.py（导入os模块）：

import os

def some_method_using_join():
    return os.path.join('parent', 'child')

这允许我们在os.path 上修补join 方法：

target_module2_test.py:

from unittest.mock import patch, MagicMock

import pytest

from target_module2 import some_method_using_join

@patch('os.path.join')
def some_test(join_mock: MagicMock):
    some_method_using_join()

这假设您不需要修补在模块级别（即导入时）使用的类或方法。