Question

我正在尝试使用python模拟库来模拟文件。虽然很简单，但我仍然不了解如何在必须接收大小参数时模拟读取函数。我试图使用side_effect创建一个替代函数，它将读取足够的数据作为值传递。

这是个主意：

def mock_read(value):
    test_string = "abcdefghijklmnopqrs"

    '''
     Now it should read enough values from the test string, but 
     I haven't figured out a way to store the position where the
     "read" method has stopped.
    '''

mock_file = MagicMock(spec=file)
mock_file.read.side_effect = mock_read

但是，我还没想出如何在side_effect函数中存储阅读器的当前位置，以便在此之后阅读。我认为可能有更好的方法，但我还是想出来了。

Answer 1

不幸的是mock_open不支持部分读取，而且你使用python 2.7（我假设它是因为你写了MagicMock(spec=file)）而且mock_open非常有限。

我们可以概括您的问题，例如我们可以编写可以保存状态的side_effect 。有一些方法可以在python中做到这一点，但恕我直言，最简单的是使用一个实现__call__的类（这里不能使用生成器，因为mock解释生成器，如副作用列表）：

from mock import MagicMock

class my_read_side_effect():
    def __init__(self,data=""):
        self._data = data
    def __call__(self, l=0): #That make my_read_side_effect a callable
        if not self._data:
            return ""
        if not l:
            l = len(self._data)
        r, self._data = self._data[:l], self._data[l:]
        return r

mock_file = MagicMock(spec=file)
mock_file.read.side_effect = my_read_side_effect("abcdefghijklmnopqrs")
assert "abcdef" == mock_file.read(6)
assert "ghijklm" == mock_file.read(7)
assert "nopqrs" == mock_file.read()

此外，我们可以在mock_open处理程序中将该实现注入补丁mock_open.read()方法。

from mock patch, mock_open

with patch("__builtin__.open", new_callable=mock_open) as mo:
    mock_file = mo.return_value
    mock_file.read.side_effect = my_read_side_effect("abcdefghijklmnopqrs")
    assert "abcdef" == mock_file.read(6)
    assert "ghijklm" == mock_file.read(7)
    assert "nopqrs" == mock_file.read()

这给了你一个简单的方法，在你的测试中使用它，其中文件在函数中打开而不作为参数传递。

使用size变量模拟文件读取方法

1 个答案: