Mercurial > public > sg101
changeset 1211:b492d640c285 modernize
Added some tests for core.tasks.
Could not get one test to work. :(
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Thu, 06 Feb 2025 20:58:33 -0600 |
parents | c4f7f77d55b2 |
children | d18db8bfe17a |
files | core/tests/test_tasks.py |
diffstat | 1 files changed, 48 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/core/tests/test_tasks.py Thu Feb 06 20:58:33 2025 -0600 @@ -0,0 +1,48 @@ +""" +Tests for the core tasks. + +""" +from django.test import TestCase +from mock import call, patch + +from core.tasks import cleanup +from core.tasks import max_users +from core.tasks import send_mail + + +class CoreTasksTestCase(TestCase): + @patch('core.tasks.django.core.mail.EmailMessage') + def test_send_mail(self, email_mock): + + kwargs = {'a': 1, 'b': 2, 'c':3} + send_mail.s(**kwargs).apply() + + self.assertEquals(email_mock.mock_calls, [ + call(**kwargs), + call().send(), + ]) + + """ + Can't get this to work... :( + @patch('django.contrib.sessions.management.commands.clearsessions') + @patch('forums.management.commands.forum_cleanup') + def test_cleanup(self, forum_cleanup_mock, clearsessions_mock): + + cleanup.s().apply() + + self.assertEquals(clearsessions.mock_calls, [ + # TODO + ]) + self.assertEquals(forum_cleanup_mock.mock_calls, [ + # TODO + ]) + """ + + @patch('core.tasks.core.whos_online') + def test_max_users(self, whos_online_mock): + + max_users.s().apply() + + self.assertEquals(whos_online_mock.mock_calls, [ + call.max_users(), + ])