bgneal@1211: """
bgneal@1211: Tests for the core tasks.
bgneal@1211: 
bgneal@1211: """
bgneal@1211: from django.test import TestCase
bgneal@1211: from mock import call, patch
bgneal@1211: 
bgneal@1211: from core.tasks import cleanup
bgneal@1211: from core.tasks import max_users
bgneal@1211: from core.tasks import send_mail
bgneal@1211: 
bgneal@1211: 
bgneal@1211: class CoreTasksTestCase(TestCase):
bgneal@1211:     @patch('core.tasks.django.core.mail.EmailMessage')
bgneal@1211:     def test_send_mail(self, email_mock):
bgneal@1211: 
bgneal@1211:         kwargs = {'a': 1, 'b': 2, 'c':3}
bgneal@1211:         send_mail.s(**kwargs).apply()
bgneal@1211: 
bgneal@1211:         self.assertEquals(email_mock.mock_calls, [
bgneal@1211:             call(**kwargs),
bgneal@1211:             call().send(),
bgneal@1211:         ])
bgneal@1211: 
bgneal@1211:     """
bgneal@1211:     Can't get this to work... :(
bgneal@1211:     @patch('django.contrib.sessions.management.commands.clearsessions')
bgneal@1211:     @patch('forums.management.commands.forum_cleanup')
bgneal@1211:     def test_cleanup(self, forum_cleanup_mock, clearsessions_mock):
bgneal@1211: 
bgneal@1211:         cleanup.s().apply()
bgneal@1211: 
bgneal@1211:         self.assertEquals(clearsessions.mock_calls, [
bgneal@1211:             # TODO
bgneal@1211:         ])
bgneal@1211:         self.assertEquals(forum_cleanup_mock.mock_calls, [
bgneal@1211:             # TODO
bgneal@1211:         ])
bgneal@1211:     """
bgneal@1211: 
bgneal@1211:     @patch('core.tasks.core.whos_online')
bgneal@1211:     def test_max_users(self, whos_online_mock):
bgneal@1211: 
bgneal@1211:         max_users.s().apply()
bgneal@1211: 
bgneal@1211:         self.assertEquals(whos_online_mock.mock_calls, [
bgneal@1211:             call.max_users(),
bgneal@1211:         ])