[英]How to mock chained methods on Django models
我正在尝试执行以下操作以使用Python模拟库在Django模型上模拟方法:
# file: tasks.py
def delete_ads(user):
# works fine and return a list of 4 MagicMock objects
ads = Classifieds.objects.filter(
user=user
)
# file: tests.py
def test_delete_ads():
return_list = [MagicMock(name='1'), MagicMock(name='2'), MagicMock(name='3'), MagicMock(name='4')]
with patch('user.tasks.Classifieds') as classified_mock:
classified_mock.objects.filter.return_value = return_value
上面的代码工作正常,但是在将代码更改为以下代码后,它开始返回单个MagicMock对象:
# file: tasks.py
def delete_ads(user):
# works fine and return a list of 4 MagicMock objects
ads = Classifieds.objects.filter(
user=user
).order_by('-added')
# file: tests.py
def test_delete_ads():
return_list = [MagicMock(name='1'), MagicMock(name='2'), MagicMock(name='3'), MagicMock(name='4')]
with patch('user.tasks.Classifieds') as classified_mock:
classified_mock.objects.filter.order_by.return_value = return_value
无论如何,即使我在Django模型上链接方法调用,我都可以正确地做到这一点?
当模拟一个模拟函数的返回值时,您必须模拟该函数,因为它将在代码中被调用。 parent.child
在parent
模拟上创建一个名为child
的属性。 而parent().child
在模拟的parent()
返回值上创建一个名为child
的属性。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.