Commit dd11bd42 by Artem Muterko Committed by Tom Christie

Add remaining tests for SimpleRateThrottle (#4803)

parent 14baf7eb
...@@ -397,3 +397,20 @@ class SimpleRateThrottleTests(TestCase): ...@@ -397,3 +397,20 @@ class SimpleRateThrottleTests(TestCase):
throttle.rate = 'some rate' throttle.rate = 'some rate'
throttle.get_cache_key = lambda *args: None throttle.get_cache_key = lambda *args: None
assert throttle.allow_request(request={}, view={}) is True assert throttle.allow_request(request={}, view={}) is True
def test_wait_returns_correct_waiting_time_without_history(self):
throttle = SimpleRateThrottle()
throttle.num_requests = 1
throttle.duration = 60
throttle.history = []
waiting_time = throttle.wait()
assert isinstance(waiting_time, float)
assert waiting_time == 30.0
def test_wait_returns_none_if_there_are_no_available_requests(self):
throttle = SimpleRateThrottle()
throttle.num_requests = 1
throttle.duration = 60
throttle.now = throttle.timer()
throttle.history = [throttle.timer() for _ in range(3)]
assert throttle.wait() is None
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment