Mercurial > public > sg101
annotate wiki/tests/test_tasks.py @ 1213:5e898f91fe36 modernize tip
Add more unit tests for accounts app.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Sun, 09 Feb 2025 14:31:35 -0600 |
| parents | c4f7f77d55b2 |
| 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 ]) |