def test_one_rule_no_persistence(self):
name = 'ut'
config = self._test_config()
config[name].pop('persistence')
rate_limiter = RateLimiter.get(name, config=config, user='******')
for i in range(5):
self.assertEqual(0, rate_limiter())
wait_time = rate_limiter(waits=False)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def test_one_rule_no_persistence(self):
name = 'ut'
config = self._test_config()
config[name].pop('persistence')
rate_limiter = RateLimiter.get(name, config=config, user='******')
for i in range(5):
self.assertEqual(0, rate_limiter())
wait_time = rate_limiter(waits=False)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def test_single_rate_limit_loaded_data(self):
name = 'ut'
config = self._test_config()
index = RamKV()
index.set_kv('rl:docido:ut-user:ut', '1462177808756|5')
rate_limiter = RateLimiter.get(name, config=config, user='******',
persistence_kwargs=dict(index=index))
for i in range(1, 6):
self.assertEqual(0, rate_limiter())
self.assertEqual(str(i),
index._kv['rl:docido:ut-user:ut'].split('|')[1])
wait_time = rate_limiter(waits=False)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def _single_rate_limit(self, runtime=False):
name = 'ut'
config = self._test_config()
index = RamKV()
ctor_kwargs, call_kwargs = self._runtime_expr(runtime)
rate_limiter = RateLimiter.get(name, config=config,
persistence_kwargs=dict(index=index),
**ctor_kwargs)
for i in range(1, 6):
self.assertEqual(0, rate_limiter(**call_kwargs))
self.assertEqual(str(i),
index._kv['rl:docido:ut-user:ut'].split('|')[1])
wait_time = rate_limiter(waits=False, **call_kwargs)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def _single_rate_limit(self, runtime=False):
name = 'ut'
config = self._test_config()
index = RamKV()
ctor_kwargs, call_kwargs = self._runtime_expr(runtime)
rate_limiter = RateLimiter.get(name,
config=config,
persistence_kwargs=dict(index=index),
**ctor_kwargs)
for i in range(1, 6):
self.assertEqual(0, rate_limiter(**call_kwargs))
self.assertEqual(str(i),
index._kv['rl:docido:ut-user:ut'].split('|')[1])
wait_time = rate_limiter(waits=False, **call_kwargs)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def test_single_rate_limit_loaded_data(self):
name = 'ut'
config = self._test_config()
index = RamKV()
index.set_kv('rl:docido:ut-user:ut', '1462177808756|5')
rate_limiter = RateLimiter.get(name,
config=config,
user='******',
persistence_kwargs=dict(index=index))
for i in range(1, 6):
self.assertEqual(0, rate_limiter())
self.assertEqual(str(i),
index._kv['rl:docido:ut-user:ut'].split('|')[1])
wait_time = rate_limiter(waits=False)
self.assertLess(0.9, wait_time)
self.assertGreater(1.1, wait_time)
def test_global_config(self):
self.assertIsInstance(RateLimiter.get_configs(), dict)
def test_global_config(self):
self.assertIsInstance(RateLimiter.get_configs(), dict)
