Added quota service tests

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