如何在 Django 模型上模拟链式方法答案

【问题标题】：How to mock chained methods on Django models如何在 Django 模型上模拟链式方法
【发布时间】：2014-07-13 10:00:48
【问题描述】：

我正在尝试使用 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 模型上链接方法调用，是否仍然可以正确执行此操作？

【问题讨论】：

您的问题中有许多错别字。您确定这些不在您的代码中吗？
@Dunes 是的，我很确定我的代码中没有错别字。我是单独输入的。

标签： python unit-testing django-models mocking python-mock

【解决方案1】：

当模拟一个模拟函数的返回值时，您必须模拟该函数，因为它会在代码中被调用。 parent.child 在 parent 模拟上创建一个名为 child 的属性。而parent().child 在parent() 的模拟返回值上创建了一个名为child 的属性。

【讨论】：