def test_delay_extraction():
polling = LROBasePolling()
headers = {}
response = Response()
response.headers = headers
polling._pipeline_response = PipelineResponse(
None,
RequestsTransportResponse(
None,
response,
),
None # context
)
headers['retry-after'] = "10"
assert polling._extract_delay() == 10
# Test that I need to retry exactly one hour after, by mocking "now"
headers['retry-after'] = "Mon, 20 Nov 1995 19:12:08 -0500"
from datetime import datetime as basedatetime
now_mock_datetime = datetime.datetime(1995, 11, 20, 18, 12, 8, tzinfo=_FixedOffset(-5*60))
with mock.patch('datetime.datetime') as mock_datetime:
mock_datetime.now.return_value = now_mock_datetime
mock_datetime.side_effect = lambda *args, **kw: basedatetime(*args, **kw)
assert polling._extract_delay() == 60*60 # one hour in seconds
assert str(mock_datetime.now.call_args[0][0]) == "<FixedOffset -5.0>"
def polling_response():
polling = LROBasePolling()
headers = {}
response = Response()
response.headers = headers
response.status_code = 200
polling._pipeline_response = PipelineResponse(
None, RequestsTransportResponse(
None,
response,
), PipelineContext(None))
polling._initial_response = polling._pipeline_response
return polling, headers
def polling_response():
polling = LROBasePolling()
headers = {}
response = Response()
response.headers = headers
polling._pipeline_response = PipelineResponse(
None,
RequestsTransportResponse(
None,
response,
),
None # context
)
return polling, headers
def _callback(http_response, headers={}):
polling = LROBasePolling()
response = Response()
response.headers = headers
response.status_code = 200
response = create_transport_response(
http_response,
None,
response,
)
polling._pipeline_response = PipelineResponse(None, response,
PipelineContext(None))
polling._initial_response = polling._pipeline_response
return polling
