Mercurial > public > sg101

annotate wiki/tests/test_tasks.py @ 1210:c4f7f77d55b2 modernize

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add unit test for wiki task.
author Brian Neal <bgneal@gmail.com>
date Mon, 03 Feb 2025 16:06:31 -0600
parents
children
rev   line source
bgneal@1210 1 """
bgneal@1210 2 Tests for the wiki application tasks.
bgneal@1210 3
bgneal@1210 4 """
bgneal@1210 5 from django.test import TestCase
bgneal@1210 6 from mock import call, patch, ANY, Mock
bgneal@1210 7
bgneal@1210 8 from wiki.tasks import expire_cookies
bgneal@1210 9
bgneal@1210 10
bgneal@1210 11 class ExpireCookiesTaskTestCase(TestCase):
bgneal@1210 12 @patch('wiki.tasks.get_redis_connection')
bgneal@1210 13 def test_expire_cookies_task(self, connection_mock):
bgneal@1210 14 redis = Mock()
bgneal@1210 15 redis.zcard.return_value = 42
bgneal@1210 16 redis.zremrangebyscore.return_value = 4
bgneal@1210 17 connection_mock.return_value = redis
bgneal@1210 18
bgneal@1210 19 task = expire_cookies.s().apply()
bgneal@1210 20
bgneal@1210 21 self.assertEquals(redis.mock_calls, [
bgneal@1210 22 call.zcard('wiki_cookie_keys'),
bgneal@1210 23 call.zremrangebyscore('wiki_cookie_keys', 0.0, ANY),
bgneal@1210 24 ])