在python单元测试中模拟选择性文件写入

Question

我在SO上四处寻找，但没有找到我想要的东西，我很确定其他地方已经找到了答案。所以我在函数中有两个文件写入，如下所示：

def write_files():
with open("a.txt", 'w') as f_h:
f_h.write("data1")
with open("b.txt", 'w') as f_h:
f_h.write("data2")

如何选择性地模拟f_h.write((，使其中一个返回异常，而另一个不返回异常？我试图设置side_effect，但不清楚它适合哪里。我试验过的测试代码是这样的：

from unittest.mock import patch, call, mock_open
import unittest
class Tester(unittest.TestCase):
def test_analyze(self):
with patch("builtins.open", mock_open(read_data="data")) as mf:
# mf.side_effect = [None, Exception()] ?
write_files()
if __name__ == '__main__':
unittest.main()

Answer 1

两件事：你必须模拟上下文管理器，例如__enter__的结果，你必须把副作用放在模拟文件句柄的write方法上(例如__enter__调用的结果(：

class Tester(unittest.TestCase):
def test_analyze(self):
with patch("builtins.open", mock_open(read_data="data")) as mf:
fh_mock = mf.return_value.__enter__.return_value
fh_mock.write.side_effect = [None, Exception]
with self.assertRaises(Exception):
write_files()