python - 仅针对一个模块使用 Mock 修补功能？

Question

我需要修补 os.listdir 和其他 os 函数来测试我的 Python 函数。但是当它们被修补时，import 语句会失败。是否可以只在那个单独的模块(真正的模块)中修补这个函数，并让 tests.py 正常工作？

这是一个破坏 import 的例子:

import os
from mock import patch

# when both isdir and isfile are patched
# the function crashes
@patch('os.path.isdir', return_value=False)
@patch('os.path.isfile', return_value=False)
def test(*args):
    import ipdb; ipdb.set_trace()
    real_function(some_arguments)
    pass

test()

我希望 real_function 看到打过补丁的 os.path，并进行测试以查看正常功能。

here's the traceback

Answer 1

您可以使用 patch 作为上下文管理器，因此它只会应用于 with 语句中的代码:

import os
from mock import patch

def test(*args):
    import ipdb; ipdb.set_trace()
    with patch('os.path.isdir', return_value=False):
        with patch('os.path.isfile', return_value=False):
            real_function(some_arguments)
    pass

test()