我有一堂这样的课。
class Upgrade:
def __init__(self, ssh_client):
self.ssh_client = ssh_client
self.channel = self.ssh_client.invoke_shell(width=1000, height=1000)
self.stdin = self.channel.makefile('wb')
self.stdout = self.channel.makefile('r')
def do_upgrade(self):
# execute some commands on paramiko channel
for line in self.stdout:
if str(line).startswith("PLAY RECAP"):
# do something
当我试图像这样模拟名为'stdout'的自我属性时(使用pytest),
def return_stdout(*args, **kwargs):
stdout = "\n\rSome return value\n\r"
return stdout
monkeypatch.setattr(Upgrade, 'stdout', return_stdout)
我收到以下错误。
> monkeypatch.setattr(Upgrade, 'stdout', return_stdout)
E AttributeError: <class 'Upgrade'> has no attribute 'stdout'
那么,如何使用 pytest 或 pytest-mock 模拟“stdout”?