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)
