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 。