如何Mock Pathlib的write_text方法捕获写入内容完成单元测试？

你可以直接通过mock实例记录的调用参数获取写入内容，具体实现如下：

方法1：直接读取mock的调用参数（最常用） #

patch生成的mock对象会自动记录所有调用的入参，你只需要把patch的返回值绑定到变量，调用完成后读取即可：

from unittest import mock

def test_path_write():
    # 注意：patch的路径需要和你被测代码中导入Path的路径匹配
    with mock.patch("pathlib.Path.write_text", return_value=None) as mock_write:
        # 执行被测函数
        function_that_uses_path_write_text()
        
        # 校验write_text确实被调用过（可选，提前拦截未调用的异常场景）
        mock_write.assert_called_once()
        
        # 提取写入内容：call_args[0]是所有位置参数元组，第一个参数就是写入的文本
        written_text = mock_write.call_args[0][0]
        assert "Content I would like to be in the file" in written_text

方法2：通过side_effect自定义捕获逻辑（适合多调用、自定义处理场景） #

如果你需要处理多次写入、或者需要额外加工捕获的内容，可以通过side_effect自定义捕获函数：

from unittest import mock

def test_path_write():
    captured_texts = []
    
    def capture_write_call(text, *args, **kwargs):
        captured_texts.append(text)
        return None # 模拟原方法的返回值
    
    with mock.patch("pathlib.Path.write_text", side_effect=capture_write_call):
        function_that_uses_path_write_text()
        
        # 直接从捕获列表中读取内容
        assert "Content I would like to be in the file" in captured_texts[0]

常见补充说明 #

• 如果被测函数会多次调用write_text，可以用mock_write.call_args_list获取所有调用的参数，第N次调用的写入内容为mock_write.call_args_list[N-1][0][0]
• 如果需要校验其他入参，比如编码、换行符等，可以通过mock_write.call_args.kwargs获取关键字参数，比如assert mock_write.call_args.kwargs.get("encoding") == "utf-8"
• 注意patch路径的正确性：如果被测代码中是from pathlib import Path，则需要patch[被测模块名].Path.write_text，而非直接patchpathlib.Path.write_text，避免mock不生效。

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