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(),
+        ])