def test_own_filter_drafts(self): filter = OwnerFilter() filter.value = 'drafts' filter.user = User.objects.get(username="******") query = Article.objects.filter() query = filter.filter(query) self.assertEqual(len(query), 1) # context = filter.get_context_data([item.id for item in query]) self.assertTrue(context.has_key('own_drafts')) self.assertEqual(context['own_articles_count'], 6) self.assertEqual(context['own_drafts_count'], 1)
def test_own_filter_anonymous(self): filter = OwnerFilter() filter.value = 'articles' query = Article.get_published() query = filter.filter(query) self.assertEqual(len(query), 39) context = filter.get_context_data([item.id for item in query]) self.assertFalse(context.has_key('own_articles')) self.assertFalse(context.has_key('own_articles_count')) self.assertFalse(context.has_key('own_drafts')) self.assertFalse(context.has_key('own_drafts_count'))