def test_message_queue(self):
message_user(self.user, "First Message")
message_user(self.user, "Second Message")
response = self.client.get('/')
msgs = list(response.context['messages'])
self.assertEqual(2, len((msgs)))
self.assertEqual('Second Message', str((msgs)[1]))
def liberation_finish(result, options):
""" Create email to send to user """
user = get_user_model().objects.get(id=options['user'])
lib_status = user.liberation
lib_status.running = False
lib_status.last_finished = timezone.now()
lib_status.content_type = int(options.get('compression_type', '0'))
lib_status.save()
message = _("Your request for your personal data has been completed. Click "
"<a class=\"alert-link\" href=\"%s\">here</a> to download your archive.")
message_user(user, safestring.mark_safe(message % urls.reverse("user-liberate-get")))
log.info("Finished liberation for %s", options['user'])
# run a garbage collection on all workers - liberation is leaky
tasks.force_garbage_collection.delay()
def test_message_appears_for_user(self):
message_user(self.user, "Hello")
response = self.client.get('/')
msgs = list(response.context['messages'])
self.assertEqual(1, len((msgs)))
self.assertEqual('Hello', str((msgs)[0]))