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