bgneal@1207: """
bgneal@1207: Unit tests for the messages application tasks.
bgneal@1207: 
bgneal@1207: """
bgneal@1207: import datetime
bgneal@1207: 
bgneal@1207: from django.test import TestCase
bgneal@1207: from django.contrib.auth.models import User
bgneal@1207: 
bgneal@1207: from messages.models import Message, Options, Flag
bgneal@1207: from messages.tasks import purge_messages
bgneal@1207: 
bgneal@1207: 
bgneal@1207: class PurgeMessagesTaskTestCase(TestCase):
bgneal@1207:     """Testing the purge messages celery task."""
bgneal@1207:     fixtures = ['messages_test_users.json']
bgneal@1207: 
bgneal@1207:     def setUp(self):
bgneal@1207:         now = datetime.datetime.now()
bgneal@1207:         pj = User.objects.get(username='pj')
bgneal@1207:         eddie = User.objects.get(username='eddie')
bgneal@1207:         pm = Message(sender=pj,
bgneal@1207:                      receiver=eddie,
bgneal@1207:                      send_date=now,
bgneal@1207:                      read_date=now,
bgneal@1207:                      subject='Test',
bgneal@1207:                      message='Message',
bgneal@1207:                      sender_delete_date=now,
bgneal@1207:                      receiver_delete_date=now)
bgneal@1207:         pm.save()
bgneal@1207: 
bgneal@1207:     def test_purge_command(self):
bgneal@1207:         self.assertEqual(Message.objects.all().count(), 1)
bgneal@1207:         task = purge_messages.s().apply()
bgneal@1207:         self.assertEqual(Message.objects.all().count(), 0)