dataservices-api/server/lib/python/cartodb_services/test/metrics/test_config.py

from unittest import TestCase
from mockredis import MockRedis
from ..test_helper import *
from cartodb_services.metrics.config import RoutingConfig, ServicesRedisConfig


class TestRoutingConfig(TestCase):

    def setUp(self):
        self._redis_conn = MockRedis()
        self._db_conn = plpy_mock
        self._username = 'my_test_user'
        self._user_key = "rails:users:{0}".format(self._username)
        self._redis_conn.hset(self._user_key, 'period_end_date', '2016-10-10')

    def test_should_pick_quota_from_server_by_default(self):
        orgname = None
        config = RoutingConfig(self._redis_conn, self._db_conn, self._username, orgname)
        assert config.monthly_quota == 1500000

    def test_should_pick_quota_from_redis_if_present(self):
        self._redis_conn.hset(self._user_key, 'mapzen_routing_quota', 1000)
        orgname = None
        config = RoutingConfig(self._redis_conn, self._db_conn, self._username, orgname)
        assert config.monthly_quota == 1000

    def test_org_quota_overrides_user_quota(self):
        self._redis_conn.hset(self._user_key, 'mapzen_routing_quota', 1000)
        orgname = 'my_test_org'
        orgname_key = "rails:orgs:{0}".format(orgname)
        self._redis_conn.hset(orgname_key, 'period_end_date', '2016-05-31')
        self._redis_conn.hset(orgname_key, 'mapzen_routing_quota', 5000)

        # TODO: these are not too relevant for the routing config
        self._redis_conn.hset(orgname_key, 'geocoding_quota', 0)
        self._redis_conn.hset(orgname_key, 'here_isolines_quota', 0)

        config = RoutingConfig(self._redis_conn, self._db_conn, self._username, orgname)
        assert config.monthly_quota == 5000


    def test_should_have_soft_limit_false_by_default(self):
        orgname = None
        config = RoutingConfig(self._redis_conn, self._db_conn, self._username, orgname)
        assert config.soft_limit == False

    def test_can_set_soft_limit_in_user_conf(self):
        self._redis_conn.hset(self._user_key, 'soft_mapzen_routing_limit', True)
        orgname = None
        config = RoutingConfig(self._redis_conn, self._db_conn, self._username, orgname)
        assert config.soft_limit == True


class TestServicesRedisConfig(TestCase):
    def test_it_picks_mapzen_routing_quota_from_redis(self):
        redis_conn = MockRedis()
        redis_conn.hset('rails:users:my_username', 'mapzen_routing_quota', 42)
        redis_config = ServicesRedisConfig(redis_conn).build('my_username', None)
        assert 'mapzen_routing_quota' in redis_config
        assert int(redis_config['mapzen_routing_quota']) == 42

    def test_org_quota_overrides_user_quota(self):
        redis_conn = MockRedis()
        redis_conn.hset('rails:users:my_username', 'mapzen_routing_quota', 42)
        redis_conn.hset('rails:orgs:acme', 'mapzen_routing_quota', 31415)
        redis_config = ServicesRedisConfig(redis_conn).build('my_username', 'acme')
        assert 'mapzen_routing_quota' in redis_config
        assert int(redis_config['mapzen_routing_quota']) == 31415