bgneal@1210: """
bgneal@1210: Tests for the wiki application tasks.
bgneal@1210: 
bgneal@1210: """
bgneal@1210: from django.test import TestCase
bgneal@1210: from mock import call, patch, ANY, Mock
bgneal@1210: 
bgneal@1210: from wiki.tasks import expire_cookies
bgneal@1210: 
bgneal@1210: 
bgneal@1210: class ExpireCookiesTaskTestCase(TestCase):
bgneal@1210:     @patch('wiki.tasks.get_redis_connection')
bgneal@1210:     def test_expire_cookies_task(self, connection_mock):
bgneal@1210:         redis = Mock()
bgneal@1210:         redis.zcard.return_value = 42
bgneal@1210:         redis.zremrangebyscore.return_value = 4
bgneal@1210:         connection_mock.return_value = redis
bgneal@1210: 
bgneal@1210:         task = expire_cookies.s().apply()
bgneal@1210: 
bgneal@1210:         self.assertEquals(redis.mock_calls, [
bgneal@1210:             call.zcard('wiki_cookie_keys'),
bgneal@1210:             call.zremrangebyscore('wiki_cookie_keys', 0.0, ANY),
bgneal@1210:         ])