def test_rate_limiter_exceptions():
with pytest.raises(ValueError):
r = RateLimiter(0)
with pytest.raises(ValueError):
r = RateLimiter(-1)
with pytest.raises(ValueError):
r = RateLimiter(10)
r.rate_limit = 0
with pytest.raises(ValueError):
r = RateLimiter(10)
r.rate_limit = -1
def test_basic_rate_limiting_large_increment():
mock_time = MockTime()
r = RateLimiter(0.1, mock_time)
# 100 operations
for i in range(0, 100):
r.acquire()
# Simulates an operation that takes 2 second
mock_time.increment_time(11)
r.consume(1)
# The operation takes longer than the minimum wait, so rate limiting should have no effect
assert mock_time.time() == 1100.0
def test_basic_rate_limiting_small_increment():
mock_time = MockTime()
r = RateLimiter(0.1, mock_time)
# 100 operations
for i in range(0, 100):
r.acquire()
# Simulates an operation that takes 2 second
mock_time.increment_time(2)
r.consume(1)
# Since the first acquire doesn't take time, thus we should be expecting (100-1) * 10 seconds = 990 delay
# plus 2 for the last increment_time(2) operation
assert mock_time.time() == 992.0