def test_moderation_objects_manager(self):
ManagerClass = ModerationObjectsManager()(Manager)
self.assertEqual(
unicode(ManagerClass.__bases__),
u"(<class 'moderation.managers.ModerationObjectsManager'>"\
u", <class 'django.db.models.manager.Manager'>)")
def test_filter_moderated_objects_returns_empty_queryset(self):
"""Test filter_moderated_objects returns empty queryset
for object that has moderated object"""
ManagerClass = ModerationObjectsManager()(Manager)
manager = ManagerClass()
manager.model = UserProfile
query_set = UserProfile._default_manager.all()
moderated_object = ModeratedObject(content_object=self.profile)
moderated_object.save()
self.assertEqual(unicode(manager.filter_moderated_objects(query_set)),
u"[]")
def test_exclude_objs_by_visibility_col(self):
ManagerClass = ModerationObjectsManager()(Manager)
manager = ManagerClass()
manager.model = ModelWithVisibilityField
ModelWithVisibilityField(test='test 1').save()
ModelWithVisibilityField(test='test 2', is_public=True).save()
query_set = ModelWithVisibilityField.objects.all()
query_set = manager.exclude_objs_by_visibility_col(query_set)
self.assertEqual(
unicode(query_set),
u"[<ModelWithVisibilityField: test 2 - is public True>]")
def test_and_fields_to_model_class(self):
class CustomManager(Manager):
pass
moderator = GenericModerator(UserProfile)
self.moderation._and_fields_to_model_class(moderator)
manager = ModerationObjectsManager()(CustomManager)()
self.assertEqual(repr(UserProfile.objects.__class__),
repr(manager.__class__))
self.assertEqual(hasattr(UserProfile, 'moderated_object'), True)
# clean up
self.moderation._remove_fields(moderator)