如何在Python中使用Mock来模拟'with”语句中的'open”句柄？

在 Python 中使用 Mock 在 With 语句中模拟 Open Handle

为使用 with 语句和文件操作的代码编写单元测试时，为了隔离测试代码的行为，需要模拟打开的句柄。以下是如何使用 Python 中的 Mock 框架实现此目的：

Python 3

使用 unittest.mock 模块，您可以修补 open 等内置函数。具体方法如下：

<code class="python">from unittest.mock import patch, mock_open

with patch("builtins.open", mock_open(read_data="data")) as mock_file:
    assert open("path/to/open").read() == "data"
mock_file.assert_called_with("path/to/open")</code>

Python 2

对于 Python 2，语法略有不同。您不需要修补builtins.open，而是需要单独修补__builtin__.open并导入mock：

<code class="python">import mock

with mock.patch("__builtin__.open", mock_open(read_data="data")) as mock_file:
    assert open("path/to/open").read() == "data"
mock_file.assert_called_with("path/to/open")</code>

使用装饰器修补

您也可以使用补丁装饰器以获得更清晰的语法。然而，直接在 patch 的 new= 参数中使用mock_open 的结果可能会很棘手。相反，请按照文档中所述利用 new_callable= 参数：

<code class="python">@patch("builtins.open", new_callable=mock_open, read_data="data")
def test_patch(mock_file):
    assert open("path/to/open").read() == "data"
    mock_file.assert_called_with("path/to/open")</code>

注意： 请记住，在这种情况下，修补的对象将作为参数传递给您的测试功能。

以上是如何在Python中使用Mock来模拟'with”语句中的'open”句柄？的详细内容。更多信息请关注PHP中文网其他相关文章！