def test_http_alerter_headers():
rule = {
'name': 'Test HTTP Post Alerter Without Payload',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'http_post2_headers': {
'authorization': 'Basic 123dr3234'
},
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'@timestamp': '2017-01-01T00:00:00',
'somefield': 'foobarbaz',
}
mock_post_request.assert_called_once_with(
rule['http_post2_url'],
data=mock.ANY,
headers={
'Content-Type': 'application/json',
'Accept': 'application/json;charset=utf-8',
'authorization': 'Basic 123dr3234'
},
proxies=None,
timeout=10,
verify=True)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])
def test_http_alerter_with_payload_nested(caplog):
caplog.set_level(logging.INFO)
rule = {
'name': 'Test HTTP Post Alerter With Payload',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'http_post2_payload': {
'posted_name': '{{ toto.tata }}'
},
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'toto': {'tata': 'titi'}}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'posted_name': 'titi',
}
mock_post_request.assert_called_once_with(
rule['http_post2_url'],
data=mock.ANY,
headers={
'Content-Type': 'application/json',
'Accept': 'application/json;charset=utf-8'
},
proxies=None,
timeout=10,
verify=True)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])
assert ('elastalert', logging.INFO,
'HTTP Post 2 alert sent.') == caplog.record_tuples[0]
def test_http_alerter_with_header_args_key_not_found(caplog):
caplog.set_level(logging.INFO)
rule = {
'name': 'Test HTTP Post Alerter With Headers args key if not found',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'http_post2_headers': {
'header_{{titi1}}': 'tata'
},
'http_post2_payload': {
'posted_name': 'toto'
},
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'titi': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_headers = {
'Content-Type': 'application/json',
'Accept': 'application/json;charset=utf-8',
'header_': 'tata'
}
mock_post_request.assert_called_once_with(rule['http_post2_url'],
data=mock.ANY,
headers=expected_headers,
proxies=None,
timeout=10,
verify=True)
assert ('elastalert', logging.INFO,
'HTTP Post 2 alert sent.') == caplog.record_tuples[0]
def test_http_getinfo():
rule = {
'name': 'Test HTTP Post Alerter Without Payload',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
expected_data = {
'type': 'http_post2',
'http_post2_webhook_url': ['http://test.webhook.url']
}
actual_data = alert.get_info()
assert expected_data == actual_data
def test_http_alerter_post_ea_exception():
with pytest.raises(EAException) as ea:
rule = {
'name': 'Test HTTP Post Alerter Without Payload',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'http_post2_ca_certs': False,
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
mock_run = mock.MagicMock(side_effect=RequestException)
with mock.patch('requests.post',
mock_run), pytest.raises(RequestException):
alert.alert([match])
assert 'Error posting HTTP Post 2 alert: ' in str(ea)
def test_http_required_error(http_post2_url, expected_data):
try:
rule = {
'name': 'Test HTTP Post Alerter Without Payload',
'type': 'any',
'alert': []
}
if http_post2_url:
rule['http_post2_url'] = http_post2_url
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
actual_data = alert.get_info()
assert expected_data == actual_data
except Exception as ea:
assert expected_data in str(ea)
def test_http_alerter_post_ca_certs(ca_certs, ignore_ssl_errors,
excpet_verify):
rule = {
'name': 'Test HTTP Post Alerter Without Payload',
'type': 'any',
'http_post2_url': 'http://test.webhook.url',
'alert': []
}
if ca_certs:
rule['http_post2_ca_certs'] = ca_certs
if ignore_ssl_errors:
rule['http_post2_ignore_ssl_errors'] = ignore_ssl_errors
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = HTTPPost2Alerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'@timestamp': '2017-01-01T00:00:00',
'somefield': 'foobarbaz',
}
mock_post_request.assert_called_once_with(
rule['http_post2_url'],
data=mock.ANY,
headers={
'Content-Type': 'application/json',
'Accept': 'application/json;charset=utf-8'
},
proxies=None,
timeout=10,
verify=excpet_verify)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])