69 lines
3.0 KiB
Python
69 lines
3.0 KiB
Python
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
|