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'))
