Как вызвать метод count у контекста?

Question

[Rus496]

Как вызвать метод count у context для подсчета передаваемых объектов модели при тестировании паджинатора? В текущем виде код такой, но хотелось бы уйти от подсчета длины списка:

response = self.client.get(reverse('index')) self.assertEqual(len(response.context['object_list']), 10)

Answer 1

[Владислава]

Я когда тудушник делала, сделала вот так:

class TaskList(LoginRequiredMixin, ListView):
model = Task
context_object_name = 'tasks'

def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context['tasks'] = context['tasks'].filter(user=self.request.user)
context['count'] = context['tasks'].filter(complete=False).count()
context['k'] = context['tasks'].filter(complete=True).count()

search_input = self.request.GET.get('search-area') or ''
if search_input:
context['tasks'] = context['tasks'].filter(title__icontains=search_input)
context['search_input'] = search_input

return context