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
