class AutomoderationRuntimeErrorRegresionTestCase(SettingsTestCase):
fixtures = ['test_users.json', 'test_moderation.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.moderation.register(UserProfile)
def tearDown(self):
self.moderation.unregister(UserProfile)
import moderation
setattr(moderation, 'moderation', self.old_moderation)
def test_RuntimeError(self):
from moderation.helpers import automoderate
profile = UserProfile.objects.get(user__username='******')
profile.description = 'Change description'
profile.save()
profile.moderated_object.changed_by = self.user
profile.moderated_object.save()
automoderate(profile, self.user)
profile.moderated_object.save()
class IntegrityErrorTestCase(SettingsTestCase):
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
self.moderation = ModerationManager()
self.moderation.register(ModelWithSlugField)
def tearDown(self):
self.moderation.unregister(ModelWithSlugField)
def test_raise_integrity_error_model_registered_with_moderation(self):
m1 = ModelWithSlugField(slug='test')
m1.save()
self.assertRaises(ObjectDoesNotExist, ModelWithSlugField.objects.get,
slug='test')
m2 = ModelWithSlugField(slug='test')
self.assertRaises(IntegrityError, m2.save)
self.assertEqual(ModeratedObject.objects.all().count(), 1)
def test_raise_integrity_error_model_not_registered_with_moderation(self):
m1 = ModelWithSlugField2(slug='test')
m1.save()
m1 = ModelWithSlugField2.objects.get(slug='test')
m2 = ModelWithSlugField2(slug='test')
self.assertRaises(IntegrityError, m2.save)
self.assertEqual(ModeratedObject.objects.all().count(), 0)
class RegisterMultipleManagersTestCase(SettingsTestCase):
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
self.moderation = ModerationManager()
class ModelWithMultipleManagersModerator(GenericModerator):
manager_names = ['objects', 'men', 'women']
self.moderation.register(ModelWithMultipleManagers,
ModelWithMultipleManagersModerator)
def tearDown(self):
self.moderation.unregister(ModelWithMultipleManagers)
def test_multiple_managers(self):
obj = ModelWithMultipleManagers(gender=0)
obj.save()
obj2 = ModelWithMultipleManagers(gender=1)
obj2.save()
men = ModelWithMultipleManagers.men.all()
women = ModelWithMultipleManagers.women.all()
self.assertEqual(men.count(), 0)
self.assertEqual(women.count(), 0)
class AutoModerateTestCase(SettingsTestCase):
fixtures = ['test_users.json', 'test_moderation.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
class UserProfileModerator(GenericModerator):
auto_approve_for_superusers = True
auto_approve_for_staff = True
auto_reject_for_groups = ['banned']
self.moderation.register(UserProfile, UserProfileModerator)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.profile = UserProfile.objects.get(user__username='******')
def tearDown(self):
import moderation
self.moderation.unregister(UserProfile)
setattr(moderation, 'moderation', self.old_moderation)
def test_auto_approve_helper_status_approved(self):
self.profile.description = 'New description'
self.profile.save()
status = automoderate(self.profile, self.user)
self.assertEqual(status, MODERATION_STATUS_APPROVED)
profile = UserProfile.objects.get(user__username='******')
self.assertEqual(profile.description, 'New description')
def test_auto_approve_helper_status_rejected(self):
group = Group(name='banned')
group.save()
self.user.groups.add(group)
self.user.save()
self.profile.description = 'New description'
self.profile.save()
status = automoderate(self.profile, self.user)
profile = UserProfile.objects.get(user__username='******')
self.assertEqual(status,
MODERATION_STATUS_REJECTED)
self.assertEqual(profile.description, 'Profile description')
def test_model_not_registered_with_moderation(self):
obj = ModelWithSlugField2(slug='test')
obj.save()
self.assertRaises(RegistrationError, automoderate, obj, self.user)
class ModerationObjectsManagerTestCase(SettingsTestCase):
fixtures = ["test_users.json", "test_moderation.json"]
urls = "moderation.tests.test_urls"
test_settings = "moderation.tests.settings.generic"
def setUp(self):
from django.db.models import signals
self.user = User.objects.get(username="******")
self.profile = UserProfile.objects.get(user__username="******")
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
def tearDown(self):
self.moderation.unregister(UserProfile)
def test_moderation_objects_manager(self):
ManagerClass = ModerationObjectsManager()(Manager)
self.assertEqual(
unicode(ManagerClass.__bases__),
u"(<class 'moderation.managers.ModerationObjectsManager'>" ", <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()
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_filter_moderated_objects_returns_object(self):
"""Test if filter_moderated_objects return object when object
doesnt have moderated object or deserialised object is <> object"""
moderated_object = ModeratedObject(content_object=self.profile)
moderated_object.save()
self.profile.description = "New"
self.profile.save()
self.assertEqual(unicode(UserProfile.objects.all()), u"[<UserProfile: moderator - http://www.google.com>]")
def setUp(self):
self.moderation = ModerationManager()
self.moderation.register(ModelWithSlugField)
self.filter_moderated_objects\
= ModelWithSlugField.objects.filter_moderated_objects
def filter_moderated_objects(query_set):
from moderation.models import MODERATION_STATUS_PENDING,\
MODERATION_STATUS_REJECTED
exclude_pks = []
for obj in query_set:
try:
if obj.moderated_object.moderation_status \
in [MODERATION_STATUS_PENDING,
MODERATION_STATUS_REJECTED] \
and obj.__dict__\
== obj.moderated_object.changed_object.__dict__:
exclude_pks.append(object.pk)
except ObjectDoesNotExist:
pass
return query_set.exclude(pk__in=exclude_pks)
setattr(ModelWithSlugField.objects,
'filter_moderated_objects',
filter_moderated_objects)
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
from django.db.models import signals
self.user = User.objects.get(username='******')
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.moderation.register(UserProfile)
def setUp(self):
self.moderation = ModerationManager()
class ModelWithMultipleManagersModerator(GenericModerator):
manager_names = ['objects', 'men', 'women']
self.moderation.register(ModelWithMultipleManagers,
ModelWithMultipleManagersModerator)
def setUp(self):
from django.db.models import signals
self.user = User.objects.get(username="******")
self.profile = UserProfile.objects.get(user__username="******")
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
class CSRFMiddlewareTestCase(SettingsTestCase):
fixtures = ['test_users.json']
urls = 'moderation.tests.test_urls'
test_settings = 'moderation.tests.settings.csrf_middleware'
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
def tearDown(self):
import moderation
self.moderation.unregister(UserProfile)
setattr(moderation, 'moderation', self.old_moderation)
def test_csrf_token(self):
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
self.client.login(username='******', password='******')
url = profile.moderated_object.get_admin_moderate_url()
if django_version == '1.1':
from django.contrib.csrf.middleware import _make_token
csrf_token = _make_token(self.client.session.session_key)
post_data = {'approve': 'Approve',
'csrfmiddlewaretoken': csrf_token}
else:
post_data = {'approve': 'Approve'}
response = self.client.post(url, post_data)
self.assertEqual(response.status_code, 302)
profile = UserProfile.objects.get(pk=profile.pk)
self.assertEqual(profile.moderated_object.moderation_status,
MODERATION_STATUS_APPROVED)
class LoadingFixturesTestCase(SettingsTestCase):
fixtures = ['test_users.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
from django.db.models import signals
self.user = User.objects.get(username='******')
self.moderation.register(UserProfile)
def tearDown(self):
self.moderation.unregister(UserProfile)
import moderation
setattr(moderation, 'moderation', self.old_moderation)
def test_loading_fixture_for_moderated_model(self):
management.call_command('loaddata', 'test_moderation.json',
verbosity=0)
self.assertEqual(UserProfile.objects.all().count(), 1)
def test_loading_objs_from_fixture_should_not_create_moderated_obj(self):
management.call_command('loaddata', 'test_moderation.json',
verbosity=0)
profile = UserProfile.objects.get(user__username='******')
self.assertRaises(ObjectDoesNotExist,
ModeratedObject.objects.get, object_pk=profile.pk)
def test_moderated_object_is_created_when_not_loaded_from_fixture(self):
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
moderated_objs = ModeratedObject.objects.filter(object_pk=profile.pk)
self.assertEqual(moderated_objs.count(), 1)
class IntegrityErrorRegresionTestCase(SettingsTestCase):
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
self.moderation = ModerationManager()
self.moderation.register(ModelWithSlugField)
self.filter_moderated_objects\
= ModelWithSlugField.objects.filter_moderated_objects
def filter_moderated_objects(query_set):
from moderation.models import MODERATION_STATUS_PENDING,\
MODERATION_STATUS_REJECTED
exclude_pks = []
for obj in query_set:
try:
if obj.moderated_object.moderation_status \
in [MODERATION_STATUS_PENDING,
MODERATION_STATUS_REJECTED] \
and obj.__dict__\
== obj.moderated_object.changed_object.__dict__:
exclude_pks.append(object.pk)
except ObjectDoesNotExist:
pass
return query_set.exclude(pk__in=exclude_pks)
setattr(ModelWithSlugField.objects,
'filter_moderated_objects',
filter_moderated_objects)
def tearDown(self):
self.moderation.unregister(ModelWithSlugField)
def test_old_version_of_filter_moderated_objects_method(self):
m1 = ModelWithSlugField(slug='test')
m1.save()
m2 = ModelWithSlugField(slug='test')
self.assertRaises(IntegrityError, m2.save)
self.assertEqual(ModeratedObject.objects.all().count(), 1)
def setUp(self):
self.user = User.objects.get(username='******')
class ModeratedObjectForm(BaseModeratedObjectForm):
class Meta:
model = UserProfile
self.ModeratedObjectForm = ModeratedObjectForm
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, "moderation", self.moderation)
from django.db.models import signals
self.user = User.objects.get(username="******")
self.moderation.register(UserProfile)
self.moderation.register(ModelWithSlugField2)
def setUp(self):
import moderation
self.moderation = ModerationManager()
class UserProfileModerator(GenericModerator):
notify_moderator = False
self.moderation.register(UserProfile, UserProfileModerator)
self.moderation._disconnect_signals(UserProfile)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.profile = UserProfile.objects.get(user__username='******')
def setUp(self):
import moderation
self.moderation = ModerationManager()
class UserProfileModerator(GenericModerator):
auto_approve_for_superusers = True
auto_approve_for_staff = True
auto_reject_for_groups = ['banned']
self.moderation.register(UserProfile, UserProfileModerator)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.profile = UserProfile.objects.get(user__username='******')
class ModeratedObjectManagerTestCase(SettingsTestCase):
fixtures = ["test_users.json"]
test_settings = "moderation.tests.settings.generic"
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, "moderation", self.moderation)
from django.db.models import signals
self.user = User.objects.get(username="******")
self.moderation.register(UserProfile)
self.moderation.register(ModelWithSlugField2)
def tearDown(self):
self.moderation.unregister(UserProfile)
self.moderation.unregister(ModelWithSlugField2)
import moderation
setattr(moderation, "moderation", self.old_moderation)
def test_objects_with_same_object_id(self):
model1 = ModelWithSlugField2(slug="test")
model1.save()
model2 = UserProfile(
description="Profile for new user", url="http://www.yahoo.com", user=User.objects.get(username="******")
)
model2.save()
self.assertRaises(MultipleObjectsReturned, ModeratedObject.objects.get, object_pk=model2.pk)
moderated_obj1 = ModeratedObject.objects.get_for_instance(model1)
moderated_obj2 = ModeratedObject.objects.get_for_instance(model2)
self.assertEqual(repr(moderated_obj1), u"<ModeratedObject: ModelWithSlugField2 object>")
self.assertEqual(repr(moderated_obj2), u"<ModeratedObject: user1 - http://www.yahoo.com>")
def setUp(self):
self.moderation = ModerationManager()
self.moderation.register(ModelWithSlugField)
class RegistrationTestCase(SettingsTestCase):
fixtures = ['test_users.json', 'test_moderation.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
#moderation = self.moderation
from django.db.models import signals
self.user = User.objects.get(username='******')
self.moderation.register(UserProfile)
def tearDown(self):
import moderation
setattr(moderation, 'moderation', self.old_moderation)
self.moderation.unregister(UserProfile)
del self.moderation
def test_get_moderator(self):
moderator = self.moderation.get_moderator(UserProfile)
self.assertTrue(isinstance(moderator, GenericModerator))
def test_get_of_new_object_should_raise_exception(self):
"""Tests if after register of model class with moderation,
when new object is created and getting of object
raise ObjectDoesNotExist"""
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
self.assertRaises(ObjectDoesNotExist, UserProfile.objects.get,
pk=profile.pk)
def test_creation_of_moderated_object(self):
"""
Test if after create of new object moderated object should be created
"""
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
modearated_object = ModeratedObject.objects.get_for_instance(profile)
self.assertEqual(unicode(modearated_object),
u"user1 - http://www.yahoo.com")
def test_get_of_existing_object_should_return_old_version_of_object(self):
"""Tests if after register of model class with moderation,
when existing object is saved, when getting of object returns
old version of object"""
profile = UserProfile.objects.get(user__username='******')
profile.description = "New description"
profile.save()
old_profile = UserProfile.objects.get(pk=profile.pk)
self.assertEqual(old_profile.description, 'Profile description')
def test_register(self):
"""Tests if after creation of new model instance new
moderation object is created"""
UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******')).save()
self.assertEqual(ModeratedObject.objects.all().count(),
1,
"New moderation object was not created"\
" after creation of new model instance "\
"from model class that is registered with moderation")
def test_exception_is_raised_when_class_is_registered(self):
self.assertRaises(RegistrationError, self.moderation.register,
UserProfile)
class ModerationManagerTestCase(SettingsTestCase):
fixtures = ['test_users.json', 'test_moderation.json']
urls = 'django-moderation.test_urls'
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
from django.db.models import signals
self.user = User.objects.get(username='******')
def tearDown(self):
import moderation
setattr(moderation, 'moderation', self.old_moderation)
del self.moderation
def test_unregister(self):
"""Tests if model class is sucessfuly unregistered from moderation"""
from django.db.models import signals
old_pre_save_receivers = [r for r in signals.pre_save.receivers]
old_post_save_receivers = [r for r in signals.post_save.receivers]
signals.pre_save.receivers = []
signals.post_save.receivers = []
self.moderation.register(UserProfile)
self.assertNotEqual(signals.pre_save.receivers, [])
self.assertNotEqual(signals.post_save.receivers, [])
UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******')).save()
self.moderation.unregister(UserProfile)
self.assertEqual(signals.pre_save.receivers, [])
self.assertEqual(signals.post_save.receivers, [])
self.assertEqual(UserProfile.objects.__class__, Manager)
self.assertEqual(hasattr(UserProfile, 'moderated_object'), False)
signals.pre_save.receivers = old_pre_save_receivers
signals.post_save.receivers = old_post_save_receivers
UserProfile.objects.get(user__username='******')
User.objects.get(username='******')
management.call_command('loaddata', 'test_moderation.json',
verbosity=0)
def test_moderation_manager(self):
moderation = ModerationManager()
self.assertEqual(moderation._registered_models, {})
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"""
from django.db.models import signals
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 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_get_or_create_moderated_object_exist(self):
profile = UserProfile.objects.get(user__username='******')
ModeratedObject(content_object=profile).save()
profile.description = "New description"
unchanged_obj = self.moderation._get_unchanged_object(profile)
object = self.moderation._get_or_create_moderated_object(profile,
unchanged_obj)
self.assertNotEqual(object.pk, None)
self.assertEqual(object.changed_object.description,
u'Profile description')
def test_get_or_create_moderated_object_does_not_exist(self):
profile = UserProfile.objects.get(user__username='******')
profile.description = "New description"
unchanged_obj = self.moderation._get_unchanged_object(profile)
object = self.moderation._get_or_create_moderated_object(profile,
unchanged_obj)
self.assertEqual(object.pk, None)
self.assertEqual(object.changed_object.description,
u'Profile description')
def test_get_unchanged_object(self):
profile = UserProfile.objects.get(user__username='******')
profile.description = "New description"
object = self.moderation._get_unchanged_object(profile)
self.assertEqual(object.description,
u'Profile description')
class FormsTestCase(SettingsTestCase):
fixtures = ['test_users.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
self.user = User.objects.get(username='******')
class ModeratedObjectForm(BaseModeratedObjectForm):
class Meta:
model = UserProfile
self.ModeratedObjectForm = ModeratedObjectForm
self.moderation = ModerationManager()
self.moderation.register(UserProfile)
def tearDown(self):
self.moderation.unregister(UserProfile)
def test_create_form_class(self):
form = self.ModeratedObjectForm()
self.assertEqual(form._meta.model.__name__, 'UserProfile')
def test_if_form_is_initialized_new_object(self):
profile = UserProfile(description="New description",
url='http://test.com',
user=self.user)
profile.save()
form = self.ModeratedObjectForm(instance=profile)
self.assertEqual(form.initial['description'], u'New description')
def test_if_form_is_initialized_existing_object(self):
profile = UserProfile(description="old description",
url='http://test.com',
user=self.user)
profile.save()
profile.moderated_object.approve(moderated_by=self.user)
profile.description = u"Changed description"
profile.save()
form = self.ModeratedObjectForm(instance=profile)
profile = UserProfile.objects.get(id=1)
self.assertEqual(profile.description, u"old description")
self.assertEqual(form.initial['description'], u'Changed description')
def test_form_when_obj_has_no_moderated_obj(self):
self.moderation.unregister(UserProfile)
profile = UserProfile(description="old description",
url='http://test.com',
user=self.user)
profile.save()
self.moderation.register(UserProfile)
form = self.ModeratedObjectForm(instance=profile)
self.assertEqual(form.initial['description'], u'old description')
class ModerationSignalsTestCase(SettingsTestCase):
fixtures = ['test_users.json', 'test_moderation.json']
test_settings = 'moderation.tests.settings.generic'
def setUp(self):
import moderation
self.moderation = ModerationManager()
class UserProfileModerator(GenericModerator):
notify_moderator = False
self.moderation.register(UserProfile, UserProfileModerator)
self.moderation._disconnect_signals(UserProfile)
self.old_moderation = moderation
setattr(moderation, 'moderation', self.moderation)
self.user = User.objects.get(username='******')
self.profile = UserProfile.objects.get(user__username='******')
def tearDown(self):
import moderation
self.moderation.unregister(UserProfile)
setattr(moderation, 'moderation', self.old_moderation)
def test_send_pre_moderation_signal(self):
"""check if custom_approve_handler function was called when """
"""moderation_approve signal was send"""
def custom_pre_moderation_handler(sender, instance, status, **kwargs):
# do some stuff with approved instance
instance.description = 'Change description'
instance.save()
pre_moderation.connect(custom_pre_moderation_handler,
sender=UserProfile)
pre_moderation.send(sender=UserProfile, instance=self.profile,
status=MODERATION_STATUS_APPROVED)
self.assertEqual(self.profile.description, 'Change description')
def test_send_post_moderation_signal(self):
"""check if custom_approve_handler function was called when """
"""moderation_approve signal was send"""
def custom_post_moderation_handler(sender, instance, status, **kwargs):
# do some stuff with approved instance
instance.description = 'Change description'
instance.save()
post_moderation.connect(custom_post_moderation_handler,
sender=UserProfile)
post_moderation.send(sender=UserProfile, instance=self.profile,
status=MODERATION_STATUS_APPROVED)
self.assertEqual(self.profile.description, 'Change description')
def test_connect_and_disconnect_signals(self):
from django.db.models import signals
old_pre_save_receivers = [r for r in signals.pre_save.receivers]
old_post_save_receivers = [r for r in signals.post_save.receivers]
signals.pre_save.receivers = []
signals.post_save.receivers = []
self.moderation._connect_signals(UserProfile)
self.assertNotEqual(signals.pre_save.receivers, [])
self.assertNotEqual(signals.post_save.receivers, [])
self.moderation._disconnect_signals(UserProfile)
self.assertEqual(signals.pre_save.receivers, [])
self.assertEqual(signals.post_save.receivers, [])
signals.pre_save.receivers = old_pre_save_receivers
signals.post_save.receivers = old_post_save_receivers
def test_after_disconnecting_signals_moderation_object(self):
self.moderation._connect_signals(UserProfile)
self.moderation._disconnect_signals(UserProfile)
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
self.assertRaises(ObjectDoesNotExist, ModeratedObject.objects.get,
object_pk=profile.pk)
def test_post_save_handler_for_exinsting_object(self):
from django.db.models import signals
signals.pre_save.connect(self.moderation.pre_save_handler,
sender=UserProfile)
signals.post_save.connect(self.moderation.post_save_handler,
sender=UserProfile)
profile = UserProfile.objects.get(user__username='******')
ModeratedObject(content_object=profile).save()
profile.description = 'New description of user profile'
profile.save()
moderated_object = ModeratedObject.objects.get_for_instance(profile)
orginal_object = moderated_object.changed_object
self.assertEqual(orginal_object.description,
'New description of user profile')
self.assertEqual(UserProfile.objects.get(pk=profile.pk).description,
"Profile description")
signals.pre_save.disconnect(self.moderation.pre_save_handler,
UserProfile)
signals.post_save.disconnect(self.moderation.post_save_handler,
UserProfile)
def test_pre_save_handler_for_exinsting_object(self):
from django.db.models import signals
signals.pre_save.connect(self.moderation.pre_save_handler,
sender=UserProfile)
profile = UserProfile.objects.get(user__username='******')
profile.description = 'New description of user profile'
profile.save()
moderated_object = ModeratedObject.objects.get_for_instance(profile)
orginal_object = moderated_object.changed_object
content_object = moderated_object.content_object
self.assertEqual(orginal_object.description,
"Profile description")
self.assertEqual(content_object.description,
'New description of user profile')
signals.pre_save.disconnect(self.moderation.pre_save_handler,
UserProfile)
def test_post_save_handler_for_new_object(self):
from django.db.models import signals
signals.pre_save.connect(self.moderation.pre_save_handler,
sender=UserProfile)
signals.post_save.connect(self.moderation.post_save_handler,
sender=UserProfile)
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
moderated_object = ModeratedObject.objects.get_for_instance(profile)
self.assertEqual(moderated_object.content_object, profile)
signals.pre_save.disconnect(self.moderation.pre_save_handler,
UserProfile)
signals.post_save.disconnect(self.moderation.post_save_handler,
UserProfile)
def test_pre_save_handler_for_new_object(self):
from django.db.models import signals
signals.pre_save.connect(self.moderation.pre_save_handler,
sender=UserProfile)
profile = UserProfile(description='Profile for new user',
url='http://www.yahoo.com',
user=User.objects.get(username='******'))
profile.save()
self.assertRaises(ObjectDoesNotExist,
ModeratedObject.objects.get_for_instance,
profile)
signals.pre_save.disconnect(self.moderation.pre_save_handler,
UserProfile)