Ejemplo n.º 1
0
 def test_moving_window_clear(self):
     limiter = MovingWindowRateLimiter(self.storage)
     per_min = RateLimitItemPerMinute(1)
     limiter.hit(per_min)
     self.assertFalse(limiter.hit(per_min))
     limiter.clear(per_min)
     self.assertTrue(limiter.hit(per_min))
Ejemplo n.º 2
0
 def test_redis_cluster_clear(self):
     storage = RedisClusterStorage("redis+cluster://localhost:7000")
     limiter = MovingWindowRateLimiter(storage)
     per_min = RateLimitItemPerMinute(1)
     limiter.hit(per_min)
     self.assertFalse(limiter.hit(per_min))
     limiter.clear(per_min)
     self.assertTrue(limiter.hit(per_min))