def test_get_retry_after_no_header():
"""
Test the _get_retry_after method with no Retry-After header.
"""
responses.add(responses.GET, BASE_URL, status=503)
response = requests.get(BASE_URL)
retry_handler = RetryHandler()
assert retry_handler._get_retry_after(response) is None
def test_get_retry_after():
"""
Test the _get_retry_after method with an integer value for retry header.
"""
responses.add(responses.GET,
BASE_URL,
headers={'Retry-After': "120"},
status=503)
response = requests.get(BASE_URL)
retry_handler = RetryHandler()
assert retry_handler._get_retry_after(response) == 120
def test_get_retry_after_http_date():
"""
Test the _get_retry_after method with a http date as Retry-After value.
"""
timevalue = time() + 120
http_date = formatdate(timeval=timevalue, localtime=False, usegmt=True)
responses.add(responses.GET,
BASE_URL,
headers={'retry-after': f'{http_date}'},
status=503)
response = requests.get(BASE_URL)
retry_handler = RetryHandler()
assert retry_handler._get_retry_after(response) < 120