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)