获取模拟文件的实际返回值。read()

问题描述

我正在使用python-mock模拟文件打开调用。我希望能够以这种方式传入假数据，这样我就可以验证read()正在被调用，并且可以使用测试数据，而不会在测试中命中文件系统。

这是我目前掌握的信息：

file_mock = MagicMock(spec=file)
file_mock.read.return_value = 'test'

with patch('__builtin__.open', create=True) as mock_open:
    mock_open.return_value = file_mock

    with open('x') as f:
        print f.read()

正如我假设的那样，它的输出是<mock.Mock object at 0x8f4aaec>的整数。我在构建此模拟时做错了什么？

编辑：

如下所示：

with open('x') as f:
     f.read()

和这个：

f = open('x')
f.read()

是不同的对象。使用模拟作为上下文管理器使其返回一个新的Mock，而直接调用它则返回我在mock_open.return_value中定义的任何内容。有什么想法吗？

---

解决方案

对于已经实现文件接口的StringIO对象来说，这听起来是一个很好的用例。也许你可以做一个file_mock = MagicMock(spec=file, wraps=StringIO('test'))。或者，您可以只让您的函数接受一个类似文件的对象，并将StringIO而不是真正的文件传递给它，从而避免了难看的修补程序。

您看过模拟文档了吗？

http://www.voidspace.org.uk/python/mock/compare.html#mocking-the-builtin-open-used-as-a-context-manager