From a18bfa955451cccf24c2a7741b29e348e4710b19 Mon Sep 17 00:00:00 2001 From: Mario de Frutos Date: Wed, 11 Nov 2015 12:28:24 +0100 Subject: [PATCH] Added quota service tests --- .../test/test_quota_service.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/server/lib/python/cartodb_geocoder/test/test_quota_service.py b/server/lib/python/cartodb_geocoder/test/test_quota_service.py index e69de29..d9c5a50 100644 --- a/server/lib/python/cartodb_geocoder/test/test_quota_service.py +++ b/server/lib/python/cartodb_geocoder/test/test_quota_service.py @@ -0,0 +1,27 @@ +from mockredis import MockRedis +from cartodb_geocoder import quota_service +from unittest import TestCase +from nose.tools import assert_raises + + +class TestQuotaService(TestCase): + + def setUp(self): + self.fake_redis_connection = MockRedis() + self.qs = quota_service.QuotaService('user_id', 'tx_id', redis_connection = self.fake_redis_connection) + + def test_should_return_true_if_quota_with_no_use(self): + self.fake_redis_connection.hset('geocoder:user_id','geocoding_quota', 100) + assert self.qs.check_user_quota() == True + + def test_should_return_true_if_quota_is_not_completely_used(self): + self.fake_redis_connection.hset('geocoder:user_id','geocoding_quota', 100) + self.fake_redis_connection.hset('geocoder:user_id:201511','tx_id', 10) + self.fake_redis_connection.hset('geocoder:user_id:201511','tx_id_2', 10) + assert self.qs.check_user_quota() == True + + def test_should_return_false_if_quota_is_surpassed(self): + self.fake_redis_connection.hset('geocoder:user_id','geocoding_quota', 1) + self.fake_redis_connection.hset('geocoder:user_id:201511','tx_id', 10) + self.fake_redis_connection.hset('geocoder:user_id:201511','tx_id_2', 10) + assert self.qs.check_user_quota() == False \ No newline at end of file