cartodb4/dataservices-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve tests and add more unit tests for the quota functionality

Browse Source
This commit is contained in:
Mario de Frutos
2016-12-20 16:17:41 +01:00
parent 6c71d73498
commit 0672f2752b
5 changed files with 493 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/lib/python/cartodb_services/test/test_user_service.py
+11 -10
View File
@@ -108,7 +108,7 @@ class TestUserService(TestCase):
def zscore_counter(self):
return self._zscore_counter
self.redis_conn = MockRedisWithCounter()
us = self.__build_user_service('test_user', end_date=date.today())
us = self.__build_user_service('test_user', end_date=datetime.today())
us.used_quota(self.NOKIA_GEOCODER, date(2015, 6, 15))
#('user:test_user:geocoder_here:success_responses:201506', 15)
@@ -132,16 +132,17 @@ class TestUserService(TestCase):
assert self.redis_conn.zscore('org:test_org:geocoder_here:success_responses:201506', '1') == None
def __build_user_service(self, username, quota=100, service='heremaps',
orgname=None, soft_limit=False,
end_date=date.today()):
build_redis_user_config(self.redis_conn, username,
quota=quota, service=service,
soft_limit=soft_limit,
end_date=end_date)
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,
quota=quota, end_date=end_date)
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)