def test_raises_error_on_invalid_init_arguments(self):
with pytest.raises(ValueError) as exc_info:
bidi._Throttle(access_limit=10,
time_window=datetime.timedelta(seconds=0.0))
assert "time_window" in str(exc_info.value)
assert "must be a positive timedelta" in str(exc_info.value)
with pytest.raises(ValueError) as exc_info:
bidi._Throttle(access_limit=0,
time_window=datetime.timedelta(seconds=10))
assert "access_limit" in str(exc_info.value)
assert "must be positive" in str(exc_info.value)
def test_delays_entry_attempts_above_threshold(self):
throttle = bidi._Throttle(access_limit=3,
time_window=datetime.timedelta(seconds=1))
entries = []
for _ in range(6):
with throttle as time_waited:
entry_info = {
"entered_at": datetime.datetime.now(),
"reported_wait": time_waited,
}
entries.append(entry_info)
# For each group of 4 consecutive entries the time difference between
# the first and the last entry must have been greater than time_window,
# because a maximum of 3 are allowed in each time_window.
for i, entry in enumerate(entries[3:], start=3):
first_entry = entries[i - 3]
delta = entry["entered_at"] - first_entry["entered_at"]
assert delta.total_seconds() > 1.0
# check the reported wait times
# (NOTE: not using assert all(...), b/c the coverage check would complain)
for i, entry in enumerate(entries):
if i != 3:
assert entry["reported_wait"] == 0.0
# The delayed entry is expected to have been delayed for a significant
# chunk of the full second, and the actual and reported delay times
# should reflect that.
assert entries[3]["reported_wait"] > 0.7
delta = entries[3]["entered_at"] - entries[2]["entered_at"]
assert delta.total_seconds() > 0.7
def test_does_not_delay_entry_attempts_under_threshold(self):
throttle = bidi._Throttle(access_limit=3,
time_window=datetime.timedelta(seconds=1))
entries = []
for _ in range(3):
with throttle as time_waited:
entry_info = {
"entered_at": datetime.datetime.now(),
"reported_wait": time_waited,
}
entries.append(entry_info)
# check the reported wait times ...
assert all(entry["reported_wait"] == 0.0 for entry in entries)
# .. and the actual wait times
delta = entries[1]["entered_at"] - entries[0]["entered_at"]
assert delta.total_seconds() < 0.1
delta = entries[2]["entered_at"] - entries[1]["entered_at"]
assert delta.total_seconds() < 0.1
def test_repr(self):
delta = datetime.timedelta(seconds=4.5)
instance = bidi._Throttle(access_limit=42, time_window=delta)
assert repr(
instance) == "_Throttle(access_limit=42, time_window={})".format(
repr(delta))