如何用unittest.mock.patch创建无参的sys.stdout吞入测试装饰器？

你遇到的问题根源在于，你把unittest.mock.patch直接装饰在了monkey_patch_stdout方法上，这会让patch自动向这个方法注入mock_stdout参数，但当你把它作为装饰器用在测试方法时，测试方法并没有接收这个参数的逻辑，所以就抛出了参数缺失的错误。

我们需要调整实现方式，让装饰器内部处理patch的逻辑，同时不让测试方法需要额外参数。下面是修正后的Mixin实现：

import io
import unittest
from unittest.mock import patch

class StdoutCapturingMixin(unittest.TestCase):
    def capture_stdout(self, func):
        def wrapper(*args, **kwargs):
            # 使用patch上下文管理器临时替换sys.stdout，自动处理还原
            with patch('sys.stdout', new_callable=io.StringIO) as mock_stdout:
                # 执行原函数，此时所有stdout输出都会被捕获
                result = func(*args, **kwargs)
                # 可选：把捕获的输出绑定到测试实例，方便后续验证输出内容
                self.captured_stdout = mock_stdout.getvalue()
                return result
        return wrapper

接下来是使用这个Mixin的测试类示例，完全符合你想要的简洁装饰器用法：

def foo(some_str):
    print(some_str)
    return some_str.isnumeric()

class TestStdoutCapture(StdoutCapturingMixin):
    @StdoutCapturingMixin.capture_stdout
    def test_foo_returns_true_for_numeric(self):
        self.assertTrue(foo("123"))
        # 如果你需要验证输出内容，直接调用实例属性即可
        self.assertEqual(self.captured_stdout.strip(), "123")
    
    @StdoutCapturingMixin.capture_stdout
    def test_foo_returns_false_for_non_numeric(self):
        self.assertFalse(foo("abc"))
        self.assertEqual(self.captured_stdout.strip(), "abc")

方案优势说明 #

1. 自动管理stdout生命周期：用with语句配合patch，无需手动重置sys.stdout，避免了手动操作可能留下的环境污染问题。
2. 无额外参数要求：测试方法不需要接收任何来自装饰器的参数，完全保持简洁的写法。
3. 可选输出验证：如果需要验证函数的stdout输出，直接通过self.captured_stdout就能获取捕获的内容，兼顾了"吞掉输出"和"验证输出"的需求。

如果你的需求只是单纯吞掉输出不需要验证，也可以简化成更精简的版本：

class StdoutCapturingMixin(unittest.TestCase):
    def capture_stdout(self, func):
        def wrapper(*args, **kwargs):
            with patch('sys.stdout', new_callable=io.StringIO):
                return func(*args, **kwargs)
        return wrapper

内容的提问来源于stack exchange，提问作者Dominik Remetei