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)
