mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-10 19:56:59 +03:00
Add remaining tests for SimpleRateThrottle (#4803)
This commit is contained in:
parent
14baf7eb7a
commit
dd11bd42a2
|
@ -397,3 +397,20 @@ class SimpleRateThrottleTests(TestCase):
|
|||
throttle.rate = 'some rate'
|
||||
throttle.get_cache_key = lambda *args: None
|
||||
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
|
||||
|
|
Loading…
Reference in New Issue
Block a user