如何在 python3 中模拟文件对象

Question

在python2中，我的测试方法中有这个：

mock_file = MagicMock(spec=file)

我正在转向python3，我不知道如何进行类似的模拟。我试过：

from io import IOBase
mock_file = MagicMock(spec=IOBase)
mock_file = create_autospec(IOBase)

我错过了什么？

Answer 1

IOBase没有实现关键的文件方法，如read和write，因此通常不适合作为创建模拟文件对象的规范。根据您是要模拟原始流、二进制文件还是文本文件，您可以使用 RawIOBase 、 BufferedIOBase 或 TextIOBase 作为规范：

from io import BufferedIOBase
mock_file = MagicMock(spec=BufferedIOBase)