您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

在python中模拟文件对象或可迭代对象

在python中模拟文件对象或可迭代对象

您正在寻找一个MagicMock。这支持迭代。

在模拟0.80beta4中,patch返回MagicMock。因此,这个简单的示例有效:

import mock

def foo():
    for line in open('myfile'):
        print line

@mock.patch('__builtin__.open')
def test_foo(open_mock):
    foo()
    assert open_mock.called

如果您正在运行模拟0.7.x(看起来像您一样),我认为您不能仅使用补丁来完成此任务。您需要单独创建模拟,然后将其传递到补丁中:

import mock

def foo():
    for line in open('myfile'):
        print line

def test_foo():
    open_mock = mock.MagicMock()
    with mock.patch('__builtin__.open', open_mock):
        foo()
        assert open_mock.called

注意-我已经使用py.test运行了这些,但是,这些相同的方法也可以用于unittest。

python 2022/1/1 18:47:06 有325人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶