def test_session_headers_and_request_headers_combine():
"""Should merge headers directly specified over any headers in a specified Session"""
session = requests.Session()
session.headers = {
'session-header': 'mySession',
'conflict-header': 'session-loses'
}
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200},
session=session,
headers={
'req-header': 'myRequest',
'conflict-header': 'req-wins'
})
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert args[0].headers == {
'conflict-header': 'req-wins',
'req-header': 'myRequest',
'session-header': 'mySession'
}
def test_multiple_alarms():
error_msg = {'status': 'ERROR'}
responses = [
httpretty.Response(body=json.dumps(error_msg), status=500),
httpretty.Response(body='Internal Server Error', status=500),
httpretty.Response(body='success', status=200)
]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
alarms = [{
'text': 'Internal Server Error',
'status_code': 500
}, {
'json': error_msg,
'status_code': 500
}]
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'status_code': 200}, alarms=alarms)
try:
httsleep.run()
except Alarm as e:
assert e.response.status_code == 500
assert e.response.json() == error_msg
assert e.alarm == {'json': error_msg, 'status_code': 500}
else:
pytest.fail("No exception raised!")
def test_run_alarm():
"""Should raise an Alarm when a failure criteria has been reached"""
httpretty.register_uri(httpretty.GET,
URL,
body='<html></html>',
status=400)
httsleep = HttSleeper(URL, {'status_code': 200},
alarms={'status_code': 400})
with pytest.raises(Alarm):
httsleep.run()
def test_run_max_retries():
"""Should raise an exception when max_retries is reached"""
responses = [httpretty.Response(body="Internal Server Error", status=500),
httpretty.Response(body="Internal Server Error", status=500),
httpretty.Response(body="Internal Server Error", status=500)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'status_code': 200}, max_retries=2)
with pytest.raises(StopIteration):
httsleep.run()
def test_default_sends_verify_true():
"""Should not send a value for 'verify' to requests by default"""
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200})
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert 'verify' in kwargs
assert kwargs['verify'] == True
def test_propagate_verify():
"""Should tell requests to skip SSL verification if verify==False"""
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200}, verify=False)
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert 'verify' in kwargs
assert kwargs['verify'] == False
def test_default_uses_default_session():
"""Should use a default Session object unless one is specified"""
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200})
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert args[0].headers == requests.utils.default_headers()
assert 'verify' in kwargs
assert kwargs['verify'] == True
def test_request_verify_overrules_session_verify():
"""Should give precedence to the 'verify' setting in the request, over that in a specified Session"""
session = requests.Session()
session.verify = '/path/to/ca/bundle'
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200},
session=session,
verify='/override/path')
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert 'verify' in kwargs
assert kwargs['verify'] == '/override/path'
def test_run_success():
"""Should return response when a success criteria has been reached"""
httpretty.register_uri(httpretty.GET, URL, body='<html></html>', status=200)
with mock.patch('httsleep.main.sleep') as mock_sleep:
httsleep = HttSleeper(URL, {'status_code': 200})
resp = httsleep.run()
assert resp.status_code == 200
assert not mock_sleep.called
def test_jsonpath_condition():
payload = {'status': 'SUCCESS'}
responses = [httpretty.Response(body=json.dumps(payload), status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'jsonpath': [{'expression': 'status', 'value': 'SUCCESS'}]})
resp = httsleep.run()
assert resp.status_code == 200
assert resp.json() == payload
def test_ignore_exceptions():
responses = [httpretty.Response(body=ConnectionError),
httpretty.Response(body="{}", status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'json': {}}, ignore_exceptions=[ConnectionError])
resp = httsleep.run()
assert resp.status_code == 200
assert resp.text == "{}"
def test_jsonpath_condition_multiple_values():
payload = {'foo': [{'bar': 1}, {'bar': 2}]}
responses = [httpretty.Response(body=json.dumps(payload), status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'jsonpath': [{'expression': 'foo[*].bar', 'value': [1, 2]}]})
resp = httsleep.run()
assert resp.status_code == 200
assert resp.json() == payload
def test_json_condition():
expected = {'my_key': 'my_value'}
responses = [httpretty.Response(body=json.dumps({'error': 'not found'}), status=404),
httpretty.Response(body=json.dumps(expected), status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'json': expected})
resp = httsleep.run()
assert resp.status_code == 200
assert resp.json() == expected
def test_text_condition():
expected = 'you got it!'
responses = [httpretty.Response(body='not found', status=404),
httpretty.Response(body=expected, status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, {'text': expected})
resp = httsleep.run()
assert resp.status_code == 200
assert resp.text == expected
def test_propagate_session():
"""Should propagate a Session's headers, verify setting, cookies etc. when specified"""
session = requests.Session()
session.cookies = {'tasty-cookie': 'chocolate'}
session.headers = {'Content-Type': 'test/type'}
session.verify = '/session/verify'
resp = Response()
resp.status_code = 200
httsleep = HttSleeper(URL, {'status_code': 200}, session=session)
with mock.patch('requests.adapters.HTTPAdapter.send') as mock_adapter_send:
mock_adapter_send.return_value = resp
httsleep.run()
assert mock_adapter_send.called
args, kwargs = mock_adapter_send.call_args
assert args[0].headers == {
'Content-Type': 'test/type',
'Cookie': 'tasty-cookie=chocolate'
}
assert 'verify' in kwargs
assert kwargs['verify'] == '/session/verify'
def test_multiple_success_conditions():
responses = [httpretty.Response(body='first response', status=200),
httpretty.Response(body='second response', status=200),
httpretty.Response(body='third response', status=200)]
httpretty.register_uri(httpretty.GET, URL, responses=responses)
conditions = [{'text': 'third response', 'status_code': 200},
{'text': 'second response', 'status_code': 200}]
with mock.patch('httsleep.main.sleep'):
httsleep = HttSleeper(URL, conditions)
resp = httsleep.run()
assert resp.status_code == 200
assert resp.text == 'second response'
def test_run_success_alarm():
"""Make sure failure criteria takes precedence over success criteria (if httsleep is being used incorrectly)"""
httpretty.register_uri(httpretty.GET, URL, body='', status=200)
httsleep = HttSleeper(URL, {'status_code': 200}, alarms={'text': ''})
with pytest.raises(Alarm):
httsleep.run()
