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]))