def test_asking_for_tasks_when_it_exceed_rate_limit(self):
rate_limit = "60000/m"
t = Throttle(self.throttle_namespace, rate_limit)
timestamp = 600
redis_conn.redis.set(t.current_time_bucket_key(timestamp), 60001)
self.assertEquals(0, t.throttle_task_size(timestamp))
def test_asking_for_tasks_when_it_exceed_rate_limit(self):
rate_limit = "60000/m"
t = Throttle(self.throttle_namespace, rate_limit)
timestamp = 600
redis_conn.redis.set(t.current_time_bucket_key(timestamp), 60001)
self.assertEquals(0, t.throttle_task_size(timestamp))
def test_throttle_current_bucket_key(self):
rate_limit = "60000/m"
t = Throttle(self.throttle_namespace, rate_limit)
timestamp = 600
self.assertEquals("t/c/1002/d/ios/10",
t.current_time_bucket_key(timestamp))
def test_throttle_current_bucket_key(self):
rate_limit = "60000/m"
t = Throttle(self.throttle_namespace, rate_limit)
timestamp = 600
self.assertEquals("t/c/1002/d/ios/10", t.current_time_bucket_key(timestamp))
