From 53acd4d30ecd5276555aa468124ec6ac6c0ac9a0 Mon Sep 17 00:00:00 2001 From: Javier Goizueta Date: Tue, 20 Dec 2016 12:41:54 +0100 Subject: [PATCH] Add tests that reveal end-of-period date problem See #322 --- .../lib/python/cartodb_services/test/test_user_service.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/server/lib/python/cartodb_services/test/test_user_service.py b/server/lib/python/cartodb_services/test/test_user_service.py index 2905f8b..cd4b880 100644 --- a/server/lib/python/cartodb_services/test/test_user_service.py +++ b/server/lib/python/cartodb_services/test/test_user_service.py @@ -23,6 +23,10 @@ class TestUserService(TestCase): amount=400) assert us.used_quota(self.NOKIA_GEOCODER, date.today()) == 400 + def test_user_quota_for_a_month_shorter_than_end_day(self): + us = self.__build_user_service('test_user', end_date=date(2016,1,31)) + assert us.used_quota(self.NOKIA_GEOCODER, date(2016,2,10)) == 0 + def test_org_used_quota_for_a_day(self): us = self.__build_user_service('test_user', orgname='test_org') increment_service_uses(self.redis_conn, 'test_user', @@ -30,6 +34,10 @@ class TestUserService(TestCase): amount=400) assert us.used_quota(self.NOKIA_GEOCODER, date.today()) == 400 + def test_org_quota_quota_for_a_month_shorter_than_end_day(self): + us = self.__build_user_service('test_user', orgname='test_org', end_date=date(2016,1,31)) + assert us.used_quota(self.NOKIA_GEOCODER, date(2016,2,10)) == 0 + def test_user_not_amount_in_used_quota_for_month_should_be_0(self): us = self.__build_user_service('test_user') assert us.used_quota(self.NOKIA_GEOCODER, date.today()) == 0