如何模拟在with语句中使用的open（在Python中使用模拟框架）？

Question

如何使用mock测试以下代码（使用mock、patch decorator和Michael Foord's Mock framework提供的sentinel）：

def testme(filepath):
    with open(filepath, 'r') as f:
        return f.read()

Answer 1

实现这一点的方法在mock 0.7.0中已经改变，mock最终支持模拟python协议方法（magic方法），特别是使用MagicMock：

http://www.voidspace.org.uk/python/mock/magicmock.html

模拟作为上下文管理器打开的示例（来自模拟文档中的示例页）：

>>> open_name = '%s.open' % __name__
>>> with patch(open_name, create=True) as mock_open:
...     mock_open.return_value = MagicMock(spec=file)
...
...     with open('/some/path', 'w') as f:
...         f.write('something')
...
<mock.Mock object at 0x...>
>>> file_handle = mock_open.return_value.__enter__.return_value
>>> file_handle.write.assert_called_with('something')