python - 模拟文件句柄

Question

There's a couple of stack overflow posts out there talking about mocking the open call in Python.这很好，但如果函数接受文件句柄或流对象而不是文件路径，它并没有真正帮助我。

到目前为止我一直在使用的一个解决方案是cStringIO 对象。但是我遇到了一个问题。

如果我想测试在某种失败时是否正确记录文件名(例如文件/流是否为空并且您期望某种数据)

cStringIO
fd = cStringIO("")
fd.name = "testing/path" # Throws an AttributeError

我无法设置 name 属性，因为 cStringIO 和 StringIO 是开槽类。

如果切换到使用open_mock

with mock.patch('__main__.open', mock.mock_open(read_data=''), create=True) as m:

我遇到了

AttributeError: Mock object has no attribute 'tell'

此时感觉我必须使用临时文件，但如果可能的话我想避免实际调用文件系统。

如何测试接受文件句柄的函数，而无需在文件系统上创建实际文件？

Answer 1

您可以使用 Mock.return_value 为模拟对象显式设置 tell 属性:

import mock

def function_under_test(f):
    f.tell()  # => 0
    f.read()
    f.tell()  # => 0
    return f.name

with mock.patch('__main__.open', mock.mock_open(read_data=''), create=True) as m:
    with open('/tmp/1') as f:
        f.name = '/tmp/1'
        f.tell.return_value = 0
        assert function_under_test(f) == '/tmp/1'