def setUp(self):
user_geocoder_config = UserGeocoderConfig(
username='******',
organization=None,
period_end_date=datetime.date.today())
redis_conn = MockRedis()
self.user_metrics_service = UserMetricsService(user_geocoder_config,
redis_conn)
def __build_user_service(self,
username,
service='geocoding',
quota=100,
provider='heremaps',
orgname=None,
soft_limit=False,
end_date=datetime.today()):
build_redis_user_config(self.redis_conn,
username,
service,
quota=quota,
provider=provider,
soft_limit=soft_limit,
end_date=end_date)
if orgname:
build_redis_org_config(self.redis_conn,
orgname,
service,
provider=provider,
quota=quota,
end_date=end_date)
geocoder_config = GeocoderConfig(self.redis_conn, plpy_mock, username,
orgname)
return UserMetricsService(geocoder_config, self.redis_conn)
def setUp(self):
user_geocoder_config = UserGeocoderConfig(
username = '******',
organization = None,
period_end_date = datetime.date.today()
)
redis_conn = MockRedis()
self.user_metrics_service = UserMetricsService(user_geocoder_config, redis_conn)
class TestUserMetricsService(TestCase):
def setUp(self):
user_geocoder_config = UserGeocoderConfig(
username = '******',
organization = None,
period_end_date = datetime.date.today()
)
redis_conn = MockRedis()
self.user_metrics_service = UserMetricsService(user_geocoder_config, redis_conn)
def test_routing_used_quota_zero_when_no_usage(self):
assert self.user_metrics_service.used_quota(UserMetricsService.SERVICE_MAPZEN_ROUTING, datetime.date.today()) == 0
def test_routing_used_quota_counts_usages(self):
self.user_metrics_service.increment_service_use(UserMetricsService.SERVICE_MAPZEN_ROUTING, 'success_responses')
self.user_metrics_service.increment_service_use(UserMetricsService.SERVICE_MAPZEN_ROUTING, 'empty_responses')
assert self.user_metrics_service.used_quota('routing_mapzen', datetime.date.today()) == 2
class TestUserMetricsService(TestCase):
def setUp(self):
user_geocoder_config = UserGeocoderConfig(
username='******',
organization=None,
period_end_date=datetime.date.today())
redis_conn = MockRedis()
self.user_metrics_service = UserMetricsService(user_geocoder_config,
redis_conn)
def test_routing_used_quota_zero_when_no_usage(self):
assert self.user_metrics_service.used_quota(
UserMetricsService.SERVICE_MAPZEN_ROUTING,
datetime.date.today()) == 0
def test_routing_used_quota_counts_usages(self):
self.user_metrics_service.increment_service_use(
UserMetricsService.SERVICE_MAPZEN_ROUTING, 'success_responses')
self.user_metrics_service.increment_service_use(
UserMetricsService.SERVICE_MAPZEN_ROUTING, 'empty_responses')
assert self.user_metrics_service.used_quota('routing_mapzen',
datetime.date.today()) == 2
