如何Mock subprocess.Popen并模拟stdout写入文件？

问题核心是原函数中文件对象被提前关闭，导致mock的Popen尝试写入时触发IO错误。以下是两种便捷的解决思路：

方案一：用mock_open保持文件对象“活跃” #

unittest.mock.mock_open可以创建模拟的文件对象，默认不会执行真实的关闭操作，完全适配原函数的文件操作逻辑。

假设你的业务函数是这样的：

import subprocess

def capture_process_output():
    # 打开文件并传给Popen
    with open("stdout.txt", "w") as stdout_file:
        subprocess.Popen(["target_cmd"], stdout=stdout_file)

对应的测试代码：

from unittest.mock import patch, mock_open

def test_capture_process_output():
    # 创建模拟文件对象
    mock_file = mock_open()
    
    with patch("builtins.open", mock_file), patch("subprocess.Popen") as mock_popen:
        # 执行待测试函数
        capture_process_output()
        
        # 验证文件打开逻辑正确
        mock_file.assert_called_once_with("stdout.txt", "w")
        
        # 获取传给Popen的stdout参数（即模拟文件句柄）
        passed_stdout = mock_popen.call_args.kwargs["stdout"]
        
        # 模拟子进程写入操作，不会触发关闭错误
        passed_stdout.write("mock process output")
        passed_stdout.flush()
        
        # 验证内容被正确写入模拟文件
        mock_file().write.assert_called_once_with("mock process output")

方案二：重定向Popen的写入到内存缓冲区 #

如果不需要验证文件操作细节，直接把Popen的stdout替换为内存缓冲区（比如io.StringIO），绕开真实文件的打开/关闭逻辑：

from unittest.mock import patch
import io

def test_capture_process_output():
    mock_buffer = io.StringIO()
    
    with patch("subprocess.Popen") as mock_popen:
        # 自定义Popen的行为，将写入操作定向到内存缓冲区
        def mock_popen_impl(args, stdout=None, **kwargs):
            if stdout:
                # 模拟子进程输出内容
                stdout.write("mock output")
                stdout.flush()
            return mock_popen.return_value
        
        mock_popen.side_effect = mock_popen_impl
        
        # 执行测试函数
        capture_process_output()
        
        # 验证Popen调用参数正确
        mock_popen.assert_called_once_with(["target_cmd"], stdout=...)
        
        # 可选：检查内存缓冲区的内容
        mock_buffer.seek(0)
        assert mock_buffer.read() == "mock output"

额外处理：手动管理文件关闭的场景 #

如果你的业务函数是手动打开/关闭文件（而非with语句），可以给模拟文件对象的close方法做空实现，避免写入时触发关闭错误：

mock_file = mock_open()
# 让close操作不实际关闭文件
mock_file().close = lambda: None

内容的提问来源于stack exchange，提问作者Maxim Ivanov