Fix non-zero-padded days in tests

See https://github.com/CartoDB/dataservices-api/issues/204 and related
tickets.
This commit is contained in:
Rafa de la Torre
2017-07-07 15:11:23 +02:00
parent 074234b8da
commit 49a53e4cd7
2 changed files with 5 additions and 5 deletions

View File

@@ -35,7 +35,7 @@ class TestQuotaChecker(TestCase):
soft_limit = False
)
redis_conn = MockRedis()
redis_conn.zincrby(self.redis_key, self.period_end_date.day, 999)
redis_conn.zincrby(self.redis_key, self.period_end_date.strftime('%d'), 999)
assert QuotaChecker(user_service_config, redis_conn).check() == True
def test_routing_quota_check_fails_when_quota_exhausted(self):
@@ -48,7 +48,7 @@ class TestQuotaChecker(TestCase):
soft_limit = False
)
redis_conn = MockRedis()
redis_conn.zincrby(self.redis_key, self.period_end_date.day, 1001)
redis_conn.zincrby(self.redis_key, self.period_end_date.strftime('%d'), 1001)
checker = QuotaChecker(user_service_config, redis_conn)
assert checker.check() == False
@@ -66,7 +66,7 @@ class TestQuotaChecker(TestCase):
soft_limit = False
)
redis_conn = MockRedis()
redis_conn.zincrby(self.redis_key, self.period_end_date.day, 1000)
redis_conn.zincrby(self.redis_key, self.period_end_date.strftime('%d'), 1000)
checker = QuotaChecker(user_service_config, redis_conn)
assert checker.check() == False
@@ -80,6 +80,6 @@ class TestQuotaChecker(TestCase):
soft_limit = True
)
redis_conn = MockRedis()
redis_conn.zincrby(self.redis_key, self.period_end_date.day, 1001)
redis_conn.zincrby(self.redis_key, self.period_end_date.strftime('%d'), 1001)
checker = QuotaChecker(user_service_config, redis_conn)
assert checker.check() == True

View File

@@ -73,7 +73,7 @@ def increment_service_uses(redis_conn, username, orgname=None,
yearmonth = date.strftime('%Y%m')
redis_name = "{0}:{1}:{2}:{3}:{4}".format(prefix, entity_name,
service, metric, yearmonth)
redis_conn.zincrby(redis_name, date.day, amount)
redis_conn.zincrby(redis_name, date.strftime('%d'), amount)
def plpy_mock_config():