def test_get_context_data(self): filter = CategoryFilter() filter.value = "django" query = Article.objects.filter(is_public=True, category__slug='django') context = filter.get_context_data([item.id for item in query]) self.assertEqual(len(context['categories']), 3) self.assertEqual(context['categories'][0].selected, True) self.assertEqual(context['categories'][0].count, 2) self.assertEqual(context['categories'][1].count, 0) self.assertEqual(context['categories'][2].count, 0)
def test_filter(self): filter = CategoryFilter() filter.value = "django" query = Article.get_published() query = filter.filter(query) self.assertEqual(len(query), 2)