python - 模拟调用 write()

Question

我有一个需要将字符串写入文件的记录器类。所以，我有这样的方法:

def write_to_file(self, string):
  self.__file_handle.write(string)

请注意，错误处理已被删除。显然，我想在不写入文件的情况下进行测试。因此通过 Mock 进行模拟.我看过this这解释了如何模拟打开，但它在这里对我没有帮助——我在 __init__ 中打开 file_handle。现在，我可以在 setUp() 中执行此操作，但 mock_open 似乎在 setUp 之后超出了范围，因此在测试用例中没有用。

您将如何使用 Mock 编写测试方法来测试 write_to_file 方法？

Answer 1

我从谷歌到达这里，过了一会儿才意识到这是自 1.0 以来使用 mock_open 内置在 mock 中的。