diff accounts/tests/test_stats.py @ 1213:5e898f91fe36 modernize

Add more unit tests for accounts app.
author Brian Neal <bgneal@gmail.com>
date Sun, 09 Feb 2025 14:31:35 -0600
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/accounts/tests/test_stats.py	Sun Feb 09 14:31:35 2025 -0600
@@ -0,0 +1,89 @@
+"""
+Tests for accounts' stats module.
+
+"""
+from django.contrib.auth.models import User
+from django.test import TestCase
+from mock import call, patch, Mock
+
+import accounts.stats
+
+
+class AccountStatsTestCase(TestCase):
+    fixtures = ['accounts.json']
+
+    @patch('accounts.tasks.user_stats_task')
+    def test_on_user_save_created(self, task_mock):
+        user = User.objects.get(pk=1)
+        kwargs = {
+            'created': True,
+            'instance': user,
+        }
+        mock_sender = Mock()
+        accounts.stats.on_user_save(mock_sender, **kwargs)
+        self.assertEqual(task_mock.mock_calls, [
+            call.delay(1),
+        ])
+
+    @patch('accounts.tasks.user_stats_task')
+    def test_on_user_save_updated(self, task_mock):
+        user = User.objects.get(pk=1)
+        kwargs = {
+            'created': False,
+            'instance': user,
+        }
+        mock_sender = Mock()
+        accounts.stats.on_user_save(mock_sender, **kwargs)
+        self.assertEqual(task_mock.mock_calls, [])
+
+    @patch('accounts.stats.get_redis_connection')
+    def test_update_user_status(self, connection_mock):
+        redis_mock = Mock()
+        connection_mock.return_value = redis_mock
+        redis_mock.incr.return_value = 1
+        accounts.stats.update_user_stats(1)
+        self.assertEqual(redis_mock.mock_calls, [
+            call.incr('accounts:user_count'),
+            call.set('accounts:user_count', 1),
+            call.pipeline(),
+            call.pipeline().lpush('accounts:new_users', u'Gremmie'),
+            call.pipeline().ltrim('accounts:new_users', 0, 9),
+            call.pipeline().execute(),
+        ])
+
+    @patch('accounts.stats.get_redis_connection')
+    def test_get_user_count(self, connection_mock):
+        redis_mock = Mock()
+        connection_mock.return_value = redis_mock
+        redis_mock.get.return_value = 42
+        result = accounts.stats.get_user_count()
+        self.assertEqual(result, 42)
+        self.assertEqual(redis_mock.mock_calls, [
+            call.get('accounts:user_count'),
+        ])
+
+    @patch('accounts.stats.get_redis_connection')
+    def test_get_new_users(self, connection_mock):
+        redis_mock = Mock()
+        connection_mock.return_value = redis_mock
+        redis_mock.lrange.return_value = ['Gremmie', 'pjmoto']
+        result = accounts.stats.get_new_users()
+        self.assertEqual(result, ['Gremmie', 'pjmoto'])
+        self.assertEqual(redis_mock.mock_calls, [
+            call.lrange('accounts:new_users', 0, -1),
+        ])
+
+    @patch('accounts.stats.get_redis_connection')
+    def test_get_new_users(self, connection_mock):
+        redis_mock = Mock()
+        connection_mock.return_value = redis_mock
+        redis_mock.get.return_value = 42
+        redis_mock.lrange.return_value = ['Gremmie', 'pjmoto']
+
+        result = accounts.stats.get_user_stats()
+
+        self.assertEqual(result, (42, ['Gremmie', 'pjmoto']))
+        self.assertEqual(redis_mock.mock_calls, [
+            call.get('accounts:user_count'),
+            call.lrange('accounts:new_users', 0, -1),
+        ])