Django，如何模擬註釋？

Question

我想模擬註釋，看看它是否被調用，並傳遞正確的參數。

我的查詢：

User.object.filter(pk__in=[1,2,3]).annotate(is_banned=F('profile__is_banned')) 

測試其工作：

@mock.patch('models.User.objects') 
def test_method(mock): 
    mock.return_value.filter.return_value = QuerySet() 
    method_call() 
    self.assertEqual(mock.call_count, 1) 
    mock.assert_called_with((pk__in=[1,2,3]) 

我不能嘲笑的註釋。

Answer 1

該註釋位於mock.filter.return_value.annotate。