Mercurial > public > sg101
changeset 1207:80f206a12027 modernize tip
Add unit tests for messages tasks.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Sun, 26 Jan 2025 11:41:28 -0600 |
parents | 02181fa5ac9d |
children | |
files | messages/tests/test_tasks.py |
diffstat | 1 files changed, 35 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/messages/tests/test_tasks.py Sun Jan 26 11:41:28 2025 -0600 @@ -0,0 +1,35 @@ +""" +Unit tests for the messages application tasks. + +""" +import datetime + +from django.test import TestCase +from django.contrib.auth.models import User + +from messages.models import Message, Options, Flag +from messages.tasks import purge_messages + + +class PurgeMessagesTaskTestCase(TestCase): + """Testing the purge messages celery task.""" + fixtures = ['messages_test_users.json'] + + def setUp(self): + now = datetime.datetime.now() + pj = User.objects.get(username='pj') + eddie = User.objects.get(username='eddie') + pm = Message(sender=pj, + receiver=eddie, + send_date=now, + read_date=now, + subject='Test', + message='Message', + sender_delete_date=now, + receiver_delete_date=now) + pm.save() + + def test_purge_command(self): + self.assertEqual(Message.objects.all().count(), 1) + task = purge_messages.s().apply() + self.assertEqual(Message.objects.all().count(), 0)