如何对打印输出到文件的方法进行单元测试？

你遇到的PermissionError是因为线上测试环境的运行账号通常对代码拉取的工作目录只有读权限，你硬编码在当前目录创建temp_file.txt的操作没有写入权限，同时硬编码文件名还会存在多测试进程冲突、测试后遗留临时文件的问题。

方案1：使用内存类文件对象测试（最优） #

你的函数只需要接收TextIO类型的文件对象即可，不需要真实写入磁盘，直接用Python标准库的io.StringIO内存类文件对象测试即可，完全不会触发磁盘权限问题，执行速度更快：

import io
from unittest import TestCase
from typing import TextIO

# 你的原有业务函数
def print_to_file(file: TextIO, value: str) -> None:
  print(value, file=file)

class TestPrintToFile(TestCase):
    def test_print_to_file(self):
        value = '123'
        expected_content = '123\n'
        # 用内存中的类文件对象代替真实磁盘文件
        mock_file = io.StringIO()
        
        print_to_file(mock_file, value)
        
        # 直接读取内存中写入的内容对比
        self.assertEqual(mock_file.getvalue(), expected_content)

方案2：使用系统临时文件（需要真实文件场景时用） #

如果你的测试逻辑必须依赖真实磁盘文件，可以用Python标准库的tempfile模块创建系统临时目录下的文件，系统临时目录默认对所有运行用户开放读写权限，且文件使用后会自动销毁，不会遗留垃圾：

import tempfile
from unittest import TestCase
from typing import TextIO

# 你的原有业务函数
def print_to_file(file: TextIO, value: str) -> None:
  print(value, file=file)

class TestPrintToFile(TestCase):
    def test_print_to_file(self):
        value = '123'
        expected_content = '123\n'
        
        # 创建临时文件，使用完成后自动删除，无需手动清理
        with tempfile.NamedTemporaryFile(mode='w+', encoding='utf-8', delete=True) as temp_file:
            print_to_file(temp_file, value)
            # 游标回到文件开头读取写入的内容
            temp_file.seek(0)
            self.assertEqual(temp_file.read(), expected_content)

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