在 TestCases 中的 setUp 或 setUpClass 中修补装饰器不起作用

Question

我正在尝试在 setUp 或 setUpClass 子类的 unittest.TestCase 方法期间修补一些函数。

给定一个模块patch_me_not.py

# patch_me_not.py
def patch_me(at):
    print('I am not patched at {}.'.format(at))

def patch_me_not(at):
    patch_me(at)

以下脚本产生的输出超出我的预期。

# main.py
import unittest
from unittest.mock import patch
from patch_me_not import patch_me_not


@patch('patch_me_not.patch_me', lambda x: None)
class PatchMeNotTests(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        print('I am the setUpClass.')
        patch_me_not('setUpClass')

    def setUp(self):
        print('I am the setUp.')
        patch_me_not('setUp')

    def test_print(self):
        print('I am the test')
        patch_me_not('test_print')


if __name__ == '__main__':
    unittest.main()

测试脚本输出为

I am the setUpClass.
I am not patched at setUpClass.
I am the setUp.
I am not patched at setUp.
I am the test
.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

如果补丁在setUp 和setUpClass 中有效，我不会期望输出中出现两行“我没有修补...”。

我怎样才能在这些方法中应用模拟补丁？

Answer 1

我认为你需要这样做：

class PatchMeNotTests(unittest.TestCase):

    @classmethod
    @patch('patch_me_not.patch_me', lambda x: None)
    def setUpClass(cls):
        print('I am the setUpClass.')
        patch_me_not('setUpClass')

    @patch('patch_me_not.patch_me', lambda x: None)
    def setUp(self):
        print('I am the setUp.')
        patch_me_not('setUp')

    def test_print(self):
        print('I am the test')
        patch_me_not('test_print')

修补您的测试用例不起作用，因为当patch 应用于TestCase 时，它仅 修补测试方法或更具体：以可配置前缀patch.TEST_PREFIX 开头的方法默认值为"test"。这就是您的解决方案不起作用的原因。

这是来自 unittest 文档的相关引用

Patch 可以用作 TestCase 类的装饰器。它的工作原理 装饰类中的每个测试方法。这减少了样板 当您的测试方法共享一个公共补丁集时的代码。 patch() 通过查找以开头的方法名称来查找测试 patch.TEST_PREFIX。默认情况下，这是'test'，匹配方式 unittest 查找测试。您可以通过设置指定替代前缀 patch.TEST_PREFIX.