时间戳

首页 > 解决方案 > 如何模拟在 python 中为特定路径打开的文件？

python - 如何模拟在 python 中为特定路径打开的文件？

问题描述

所以我知道在我的单元测试中我可以模拟一个上下文管理器 open()，即：

with open('file_path', 'r') as stats:

嘲笑

with mock.patch('builtins.open', mock.mock_open(read_data=mock_json)):

但是有没有办法让我只为特定的文件路径模拟它？或者可能是其他方式来确保在单元测试中使用正确的路径调用上下文管理器？

标签： pythonunit-testingmockingpython-unittest

解决方案

要仅为特定路径模拟打开，您必须提供自己的模拟对象，根据路径以不同方式处理打开。假设我们有一些功能：

def do_open(path):
    with open(path, "r") as f:
        return f.read()

open如果是“bar”，则应模拟where以返回内容为“bar”的文件path，否则照常工作，您可以执行以下操作：

from unittest import mock
from my_module.do_open import do_open

builtin_open = open  # save the unpatched version

def mock_open(*args, **kwargs):
    if args[0] == "foo":
        # mocked open for path "foo"
        return mock.mock_open(read_data="bar")(*args, **kwargs)
    # unpatched version for every other path
    return builtin_open(*args, **kwargs)

@mock.patch("builtins.open", mock_open)
def test_open():
    assert do_open("foo") == "bar"
    assert do_open(__file__) != "bar"

如果您不想将原始文件保存open在全局变量中，也可以将其包装到一个类中：

class MockOpen:
    builtin_open = open

    def open(self, *args, **kwargs):
        if args[0] == "foo":
            return mock.mock_open(read_data="bar")(*args, **kwargs)
        return self.builtin_open(*args, **kwargs)

@mock.patch("builtins.open", MockOpen().open)
def test_open():
    ...

推荐阅读