Python 修补setUp中的decorator或TestCases中的setUpClass不起作用

我试图在

unittest.TestCase

子类的

setUp

或

setUpClass

方法期间修补一些函数

给定一个模块

补丁\u me\u 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

中工作，则我不希望输出中出现两行“我没有在…”

---

如何在这些方法中应用模拟补丁？

我认为您需要这样做：

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')

类补丁测试（unittest.TestCase）：
@类方法
@补丁（'patch\u me\u not.patch\u me'，lambda x:None）
def设置等级（cls）：
打印（“我是setUpClass”。）
修补程序\u me\u not（'setUpClass'）
@补丁（'patch\u me\u not.patch\u me'，lambda x:None）
def设置（自）：
打印（'我是安装程序'）
修补程序\u me\u not（“安装程序”）
def测试_打印（自）：
打印（“我是测试对象”）
修补程序不可用（“测试打印”）

修补测试用例不起作用，因为当

patch

应用于

TestCase

时，它只修补测试方法，或者更具体地说：以可配置前缀

patch开头的方法。test\u prefix

哪个默认值是

“test”

。这就是你的解决方案不起作用的原因

以下是unittest文档中的相关引用

补丁可以用作TestCase类装饰器。它的工作原理是 修饰类中的每个测试方法。这减少了样板文件 当您的测试方法共享公共补丁集时编写代码<代码>补丁（） 通过查找以开头的方法名查找测试

补丁。测试前缀

。默认情况下，这是

'test'

，与方法匹配 unittest查找测试。您可以通过设置

补丁。测试前缀

---

你不能从你的设置方法中删除打印吗？我不明白你的问题。是否要抑制所有标准输出？这与函数本身无关，这只是一个示例。在我的实际用例中，我没有修补stdout.write，但同样的问题也适用。我更新了我的问题，以获得一个更完整的示例来解释这个问题。非常感谢，这确实有效！你能解释一下为什么类级别的decorator不适用于这两个方法吗？@yellowcap我已经用解释更新了我的答案