如何在Django模型上模擬鏈式方法

Question

我正在嘗試執行以下操作以使用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模型上鏈接方法調用，我都可以正確地做到這一點？

Answer 1

當模擬一個模擬函數的返回值時，您必須模擬該函數，因為它將在代碼中被調用。 parent.child在parent模擬上創建一個名為child的屬性。 而parent().child在模擬的parent()返回值上創建一個名為child的屬性。