def setUp(self): from django.db import models self.moderator = GenericModerator(UserProfile) class CustomManager(models.Manager): pass class ModelClass(models.Model): pass self.custom_manager = CustomManager self.model_class = ModelClass
def test_add_fields_to_model_class_django_1_10(self): class CustomManager(Manager): pass moderator = GenericModerator(UserProfile) self.moderation._add_fields_to_model_class(moderator) self.assertEqual(UserProfile.objects.__class__.__name__, 'ModeratedManager') self.assertEqual(hasattr(UserProfile, 'moderated_object'), True) # clean up self.moderation._remove_fields(moderator)
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)
def test_save_new_instance_after_add_and_remove_fields_from_class(self): """Test if after removing moderation from model class new instance of model can be created""" class CustomManager(Manager): pass moderator = GenericModerator(UserProfile) self.moderation._and_fields_to_model_class(moderator) self.moderation._remove_fields(moderator) profile = UserProfile(description='Profile for new user', url='http://www.yahoo.com', user=User.objects.get(username='******')) profile.save() up = UserProfile._default_manager.filter(url='http://www.yahoo.com') self.assertEqual(up.count(), 1)
def setUp(self): self.user = User.objects.get(username='******') obj = ModeratedObject(content_object=self.user) obj.save() self.user.moderated_object = obj self.moderator = GenericModerator(UserProfile)
def setUp(self): self.user = User.objects.get(username='******') self.moderator = GenericModerator(UserProfile) self.obj = object
def test_inform_user(self): self.moderator = GenericModerator(UserProfile) self.moderator.inform_user(self.user, self.user) self.assertEqual(len(mail.outbox), 1)
def setUp(self): self.moderator = GenericModerator(UserProfile) self.custom_manager = self.CustomManager self.model_class = self.ModelClass