模拟不适用于pytest和flexmock

Question

我正在尝试使用 pytest 夹具来模拟对open()的调用，然后在测试拆卸时重置它，但由于某种原因，模拟未应用于测试函数。

以下是我所拥有的示例：

# tests.py
@pytest.fixture(scope='module')
def mock_open(request):
    mock = flexmock(sys.modules['__builtin__'])
    mock.should_call('open')
    m = (mock.should_receive('open')
        .with_args('/tmp/somefile')
        .and_return(read=lambda: 'file contents'))
    request.addfinalizer(lambda: m.reset())
def test_something(mock_open):
    ret = function_to_test()
    ret[1]()  # This actually uses the original open()

而且，如果它很重要，以下是function_to_test()的样子：

# some_module.py
def function_to_test():
    def inner_func():
        open('/tmp/somefile').read()   # <-- I want this call mocked
        # More code ...
    return (True, inner_func)

如果我使用 xUnit 样式的 setup_module()/teardown_module() 函数，也会发生这种情况。但是如果我把模拟代码放在测试函数本身中（我显然不想这样做），那么它就可以正常工作。

我错过了什么？谢谢！

Answer 1

使用mock怎么样？

---

tests.py：

import mock # import unittest.mock (Python 3.3+)
import pytest
from some_module import function_to_test
@pytest.fixture(scope='function')
def mock_open(request):
    m = mock.patch('__builtin__.open', mock.mock_open(read_data='file content'))
    m.start()
    request.addfinalizer(m.stop)
def test_something(mock_open):
    ret = function_to_test()
    assert ret[1]() == 'file content'

some_module.py：

def function_to_test():
    def inner_func():
        return open('/tmp/somefile').read()
    return True, inner_func