python - 在 Python 中模拟标准输出

Question

我是 Python 单元测试的新手，尤其是 Mock。我将如何模拟一个可以执行以下操作的对象？我只需要一个不会使循环崩溃的对象，以完成测试。

for ln in theMock.stdout.readlines()

我试着通过做来创建一个模拟

Mock(stdout=Mock(readlines= Lambda: []))

和

Mock(stdout=Mock(spec=file, wraps=StringIO())

但是它说列表对象没有属性 stdout。

Answer 1

这个怎么样？

from mock import Mock

readlines = Mock(return_value=[])
stdout = Mock(readlines=readlines)
theMock = Mock(stdout=stdout)
print(theMock.stdout.readlines())

输出:

[]

您的 for 循环将被跳过，因为 readlines() 将返回一个空列表。