Some tests for the QuotaChecker #281

server/lib/python/cartodb_services/test/metrics/test_quota.py

@@ -0,0 +1,66 @@
+from unittest import TestCase
+from mockredis import MockRedis
+from ..test_helper import build_plpy_mock
+from cartodb_services.metrics.quota import QuotaChecker
+from cartodb_services.metrics import RoutingConfig
+from datetime import datetime
+
+
+class RoutingConfigMock(object):
+
+    def __init__(self, **kwargs):
+        self.__dict__ = kwargs
+
+
+class TestQuotaChecker(TestCase):
+
+    USERNAME = 'my_test_user'
+    PERIOD_END_DATE = datetime(2016,10,20)
+    SERVICE_TYPE = 'routing_mapzen'
+    REDIS_KEY = 'user:{0}:{1}:success_responses:{2}{3}'.format(
+        USERNAME,
+        SERVICE_TYPE,
+        PERIOD_END_DATE.year,
+        PERIOD_END_DATE.month
+    )
+
+    def test_routing_quota_check_passes_when_enough_quota(self):
+        user_service_config = RoutingConfigMock(
+            username = self.USERNAME,
+            organization = None,
+            service_type = self.SERVICE_TYPE,
+            monthly_quota = 1000,
+            period_end_date = datetime.today(),
+            soft_limit = False
+        )
+        redis_conn = MockRedis()
+        redis_conn.zincrby(self.REDIS_KEY, self.PERIOD_END_DATE.day, 999)
+        assert QuotaChecker(user_service_config, redis_conn).check() == True
+
+    def test_routing_quota_check_fails_when_quota_exhausted(self):
+        user_service_config = RoutingConfigMock(
+            username = self.USERNAME,
+            organization = None,
+            service_type = self.SERVICE_TYPE,
+            monthly_quota = 1000,
+            period_end_date = datetime.today(),
+            soft_limit = False
+        )
+        redis_conn = MockRedis()
+        redis_conn.zincrby(self.REDIS_KEY, self.PERIOD_END_DATE.day, 1001)
+        checker = QuotaChecker(user_service_config, redis_conn)
+        assert checker.check() == False
+
+    def test_routing_quota_check_passes_if_no_quota_but_soft_limit(self):
+        user_service_config = RoutingConfigMock(
+            username = self.USERNAME,
+            organization = None,
+            service_type = self.SERVICE_TYPE,
+            monthly_quota = 1000,
+            period_end_date = datetime.today(),
+            soft_limit = True
+        )
+        redis_conn = MockRedis()
+        redis_conn.zincrby(self.REDIS_KEY, self.PERIOD_END_DATE.day, 1001)
+        checker = QuotaChecker(user_service_config, redis_conn)
+        assert checker.check() == True