使用Python unittest无法捕获os.system执行命令的stdout输出

你遇到的核心问题是：os.system()执行的shell命令是作为独立子进程运行的，它们的输出不会流向当前Python进程的sys.stdout。你用patch('sys.stdout')只能捕获Python代码中print()这类直接操作当前进程stdout的输出，而子进程的输出会直接绕过这个捕获，直接输出到终端或者测试环境的原始标准输出，所以你的fakeOutput始终是空的。

下面提供两种实用的解决思路，你可以根据是否能修改业务代码来选择：

方案1：替换os.system为subprocess模块（推荐，更可控） #

把业务代码里的os.system换成subprocess.run，这样可以直接捕获子进程的输出，再把它输出到当前Python进程的sys.stdout，这样测试时就能通过patch('sys.stdout')捕获到内容。

修改你的业务代码（比如application.py里的任务执行部分）：

import subprocess
import sys

def execute_task(task):
    # 捕获子进程的stdout和stderr，text=True让输出是字符串格式
    result = subprocess.run(
        task,
        shell=True,
        capture_output=True,
        text=True
    )
    # 把子进程的stdout打印到当前进程的stdout
    if result.stdout:
        print(result.stdout.strip())
    # 处理stderr（如果需要的话）
    if result.stderr:
        print(f"Error: {result.stderr.strip()}", file=sys.stderr)

之后你的原有测试代码就能正常工作了——因为现在子进程的输出被转成了Python的print()调用，会被sys.stdout的patch捕获到。

方案2：Mock os.system来模拟输出（无需修改业务代码） #

如果不能修改业务代码，你可以直接mockos.system函数，让它在被调用时主动向sys.stdout写入对应的输出，这样测试就能捕获到预期内容，同时还能验证命令是否被正确执行。

修改你的测试代码：

from application import application
from io import StringIO
import unittest
from unittest.mock import patch, call

class TestApplication(unittest.TestCase):
    def test_application_tasks(self):
        expected_output = '1\n2\n3\n4\n5\n'
        
        with patch('application.os.system') as mock_system, \
             patch('sys.stdout', new=StringIO()) as fakeOutput:
            
            # 定义mock的副作用：根据传入的echo命令，输出对应内容
            def mock_system_side_effect(cmd):
                # 提取echo命令里的内容（简单处理，实际可以更健壮）
                if "echo '1'" in cmd:
                    print('1')
                elif "echo '2'" in cmd:
                    print('2')
                elif "echo '3'" in cmd:
                    print('3')
                elif "echo '4'" in cmd:
                    print('4')
                elif "echo '5'" in cmd:
                    print('5')
                return 0  # 返回os.system的正常退出码（0表示成功）
            
            mock_system.side_effect = mock_system_side_effect
            application.parse_event('some event')
            
            # 先验证os.system是否被正确调用了所有命令
            mock_system.assert_has_calls([
                call("echo '1'"),
                call("echo '2'"),
                call("echo '3'"),
                call("echo '4'"),
                call("echo '5'")
            ], any_order=False)
            
            # 再验证输出是否符合预期
            self.assertEqual(fakeOutput.getvalue(), expected_output)

这个方案通过mock完全模拟了os.system的行为，既验证了命令执行的正确性，又捕获到了预期的输出。

内容的提问来源于stack exchange，提问作者Kevin Kopf