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

from test_helper import *
from unittest import TestCase
from nose.tools import assert_raises
from mockredis import MockRedis
from datetime import datetime, timedelta
from cartodb_services.metrics import GeocoderConfig, ObservatorySnapshotConfig, ConfigException


class TestConfig(TestCase):

    def setUp(self):
        self.redis_conn = MockRedis()
        plpy_mock_config()

    def test_should_return_list_of_nokia_geocoder_config_if_its_ok(self):
        build_redis_user_config(self.redis_conn, 'test_user')
        geocoder_config = GeocoderConfig(self.redis_conn, plpy_mock,
                                         'test_user', None)
        assert geocoder_config.heremaps_geocoder is True
        assert geocoder_config.geocoding_quota == 100
        assert geocoder_config.soft_geocoding_limit is False

    def test_should_return_list_of_nokia_geocoder_config_ok_for_org(self):
        yesterday = datetime.today() - timedelta(days=1)
        build_redis_user_config(self.redis_conn, 'test_user')
        build_redis_org_config(self.redis_conn, 'test_org',
                               quota=200, end_date=yesterday)
        geocoder_config = GeocoderConfig(self.redis_conn, plpy_mock,
                                         'test_user', 'test_org')
        assert geocoder_config.heremaps_geocoder is True
        assert geocoder_config.geocoding_quota == 200
        assert geocoder_config.soft_geocoding_limit is False
        assert geocoder_config.period_end_date.date() == yesterday.date()

    def test_should_return_config_for_obs_snapshot(self):
        yesterday = datetime.today() - timedelta(days=1)
        build_redis_user_config(self.redis_conn, 'test_user',
                                do_quota=100, soft_do_limit=True,
                                end_date=yesterday)
        do_config = ObservatorySnapshotConfig(self.redis_conn, plpy_mock,
                                          'test_user')
        assert do_config.monthly_quota == 100
        assert do_config.soft_limit is True
        assert do_config.period_end_date.date() == yesterday.date()

    def test_should_return_db_quota_if_not_redis_quota_config_obs_snapshot(self):
        yesterday = datetime.today() - timedelta(days=1)
        build_redis_user_config(self.redis_conn, 'test_user',
                                end_date=yesterday)
        do_config = ObservatorySnapshotConfig(self.redis_conn, plpy_mock,
                                          'test_user')
        assert do_config.monthly_quota == 0
        assert do_config.soft_limit is False
        assert do_config.period_end_date.date() == yesterday.date()

    def test_should_raise_exception_when_missing_parameters(self):
        plpy_mock._reset()
        build_redis_user_config(self.redis_conn, 'test_user')
        assert_raises(ConfigException, GeocoderConfig, self.redis_conn,
                      plpy_mock, 'test_user', None)