本文介绍了如何在 Python 中对象的模拟方法中调用 self?的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着跟版网的小编来一起学习吧！
问题描述
我尝试测试一些不返回任何内容但将结果保存到数据库的代码.通过模拟 save 方法，我希望检查事情是否已正确处理:
I try to test some codes which don't return anything but save the result to the DB. By mocking the save method, I wish to check whether things have been processed correctly:
def mock_save(self):
assert(self.attr, 'dest_val')
with mock.patch.object(Item, "save", create=True) as save:
save.side_effect = mock_save
func_to_call() //in func_to_call, I call item.save()
但是，这似乎是不允许的.它表示参数的数量不匹配.
However, it seems that this is not allowed. It says that the number of argument mismatch.
如果我执行 def mock_save()，它将不起作用.
If I do def mock_save(), it won't work.
我如何才能引用模拟方法所作用的对象?(我在另一个适用于 __init__ 方法的线程中看到它可以直接从类中调用).
How can I have a reference to the object which the mock method act upon too? (I saw it in another thread that is applicable to __init__ method which can be called directly from the class).
推荐答案
你需要 autospec=True
def mock_save(self):
assert self.attr == 'dest_val'
with mock.patch.object(Item, "save", autospec=True) as save:
save.side_effect = mock_save
func_to_call()
这篇关于如何在 Python 中对象的模拟方法中调用 self?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持跟版网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！