python - py.test fixture 上的补丁

Question

我使用以下代码模拟常量值以使用 py.test 进行测试:

@patch('ConstantsModule.ConstantsClass.DELAY_TIME', 10)
def test_PowerUp():
    ...
    thing = Thing.Thing()
    assert thing.a == 1

这模拟了在测试和 Thing 中使用的 DELAY_TIME，这是我所期望的。

我想对这个文件中的所有测试都这样做，所以我尝试了

@patch('ConstantsModule.ConstantsClass.DELAY_TIME', 10)
@pytest.fixture(autouse=True)
def NoDelay():
    pass

但这似乎没有同样的效果。

这是一个类似的问题:pytest-mock mocker in pytest fixture ，但模拟似乎是以非装饰器的方式完成的。

Answer 1

我想说通过装饰器打补丁并不是这里的最佳方法。我会使用上下文管理器:

import pytest
from unittest.mock import patch


@pytest.fixture(autouse=True)
def no_delay():
    with patch('ConstantsModule.ConstantsClass.DELAY_TIME', 10):
        yield

通过这种方式，补丁在测试拆卸时完全恢复。