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

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

您正在寻找一个

MagicMock

。这支持迭代。

在模拟0.80beta4中，

patch

返回

MagicMock

。因此，这个简单的示例有效：

import mockdef foo():    for line in open('myfile'):        print [email protected]('__builtin__.open')def test_foo(open_mock):    foo()    assert open_mock.called

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

import mockdef foo():    for line in open('myfile'):        print linedef test_foo():    open_mock = mock.MagicMock()    with mock.patch('__builtin__.open', open_mock):        foo()        assert open_mock.called

---

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