def test_generic_queryset_for_category(self):
featured = Featured.objects.create(content_object=self.user, category=self.category)
featured = Featured.objects.create(content_object=self.user2, category=self.category2)
manager = get_featured_queryset_for(User, category=self.category)
self.assertEqual(list(manager.all()), [self.user])
def get_context_data(self, slug, model):
app_label, model_name = model.split('.')
model_class = self.get_model(model)
self.category = self.get_category(slug)
object_name = self.get_object_name(model)
manager = get_featured_queryset_for(model_class, category=self.category)
return {'object_list': manager, object_name: manager}
def test_generic_queryset_for_category(self):
featured = Featured.objects.create(content_object=self.user,
category=self.category)
featured = Featured.objects.create(content_object=self.user2,
category=self.category2)
manager = get_featured_queryset_for(User, category=self.category)
self.assertEqual(list(manager.all()), [self.user])
def get_context_data(self, slug, model):
app_label, model_name = model.split('.')
model_class = self.get_model(model)
self.category = self.get_category(slug)
object_name = self.get_object_name(model)
manager = get_featured_queryset_for(model_class,
category=self.category)
return {'object_list': manager, object_name: manager}
def test_get_context_data(self):
view = FeaturedListView()
queryset = get_featured_queryset_for(User, category=self.category)
context = view.get_context_data(self.category.slug, 'auth.user')
self.assertEqual(2, len(context))
self.assertTrue('object_list' in context)
self.assertEqual(list(queryset), list(context['object_list']))
self.assertTrue('user_list' in context)
self.assertEqual(list(queryset), list(context['user_list']))
def test_get_context_data(self):
view = FeaturedListView()
queryset = get_featured_queryset_for(User, category=self.category)
context = view.get_context_data(self.category.slug, 'auth.user')
self.assertEqual(2, len(context))
self.assertTrue('object_list' in context)
self.assertEqual(list(queryset), list(context['object_list']))
self.assertTrue('user_list' in context)
self.assertEqual(list(queryset), list(context['user_list']))
