def test_pagerduty_ea_exception():
with pytest.raises(EAException) as ea:
rule = {
'name': 'Test PD Rule',
'type': 'any',
'alert_subject': '{0} kittens',
'alert_subject_args': ['somefield'],
'pagerduty_service_key': 'magicalbadgers',
'pagerduty_event_type': 'trigger',
'pagerduty_client_name': 'ponies inc.',
'pagerduty_incident_key': 'custom {0}',
'pagerduty_incident_key_args': ['someotherfield'],
'pagerduty_proxy': 'http://proxy.url',
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
match = {
'@timestamp': '2017-01-01T00:00:00',
'somefield': 'Stinkiest',
'someotherfield': 'foobarbaz'
}
mock_run = mock.MagicMock(side_effect=RequestException)
with mock.patch('requests.post',
mock_run), pytest.raises(RequestException):
alert.alert([match])
assert 'Error posting to pagerduty: ' in str(ea)
def test_pagerduty_alerter_custom_alert_subject():
rule = {
'name': 'Test PD Rule',
'type': 'any',
'alert_subject': 'Hungry kittens',
'pagerduty_service_key': 'magicalbadgers',
'pagerduty_client_name': 'ponies inc.',
'pagerduty_incident_key': 'custom {0}',
'pagerduty_incident_key_args': ['somefield'],
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'client': 'ponies inc.',
'description': 'Hungry kittens',
'details': {
'information':
'Test PD Rule\n\n@timestamp: 2017-01-01T00:00:00\nsomefield: foobarbaz\n'
},
'event_type': 'trigger',
'incident_key': 'custom foobarbaz',
'service_key': 'magicalbadgers',
}
mock_post_request.assert_called_once_with(
alert.url,
data=mock.ANY,
headers={'content-type': 'application/json'},
proxies=None)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])
def test_pagerduty_alerter_v2_payload_custom_details():
rule = {
'name': 'Test PD Rule',
'type': 'any',
'pagerduty_service_key': 'magicalbadgers',
'pagerduty_client_name': 'ponies inc.',
'pagerduty_api_version': 'v2',
'pagerduty_v2_payload_class': 'ping failure',
'pagerduty_v2_payload_component': 'mysql',
'pagerduty_v2_payload_group': 'app-stack',
'pagerduty_v2_payload_severity': 'error',
'pagerduty_v2_payload_source': 'mysql.host.name',
'pagerduty_v2_payload_custom_details': {
'a': 'somefield',
'c': 'f'
},
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'client': 'ponies inc.',
'payload': {
'class': 'ping failure',
'component': 'mysql',
'group': 'app-stack',
'severity': 'error',
'source': 'mysql.host.name',
'summary': 'Test PD Rule',
'custom_details': {
'a':
'foobarbaz',
'c':
None,
'information':
'Test PD Rule\n\n@timestamp: 2017-01-01T00:00:00\nsomefield: foobarbaz\n'
},
'timestamp': '2017-01-01T00:00:00'
},
'event_action': 'trigger',
'dedup_key': '',
'routing_key': 'magicalbadgers',
}
mock_post_request.assert_called_once_with(
'https://events.pagerduty.com/v2/enqueue',
data=mock.ANY,
headers={'content-type': 'application/json'},
proxies=None)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])
def test_pagerduty_getinfo():
rule = {
'name': 'Test PD Rule',
'type': 'any',
'pagerduty_service_key': 'magicalbadgers',
'pagerduty_client_name': 'ponies inc.',
'alert': []
}
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
expected_data = {
'type': 'pagerduty',
'pagerduty_client_name': 'ponies inc.'
}
actual_data = alert.get_info()
assert expected_data == actual_data
def test_pagerduty_required_error(pagerduty_service_key, pagerduty_client_name,
expected_data):
try:
rule = {'name': 'Test PD Rule', 'type': 'any', 'alert': []}
if pagerduty_service_key:
rule['pagerduty_service_key'] = pagerduty_service_key
if pagerduty_client_name:
rule['pagerduty_client_name'] = pagerduty_client_name
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
actual_data = alert.get_info()
assert expected_data == actual_data
except Exception as ea:
assert expected_data in str(ea)
def test_pagerduty_alerter_event_type(pagerduty_event_type,
excepted_pagerduty_event_type,
excepted_log_message, caplog):
caplog.set_level(logging.INFO)
rule = {
'name': 'Test PD Rule',
'type': 'any',
'pagerduty_service_key': 'magicalbadgers',
'pagerduty_client_name': 'ponies inc.',
'alert': []
}
if pagerduty_event_type:
rule['pagerduty_event_type'] = pagerduty_event_type
rules_loader = FileRulesLoader({})
rules_loader.load_modules(rule)
alert = PagerDutyAlerter(rule)
match = {'@timestamp': '2017-01-01T00:00:00', 'somefield': 'foobarbaz'}
with mock.patch('requests.post') as mock_post_request:
alert.alert([match])
expected_data = {
'client': 'ponies inc.',
'description': 'Test PD Rule',
'details': {
'information':
'Test PD Rule\n\n@timestamp: 2017-01-01T00:00:00\nsomefield: foobarbaz\n'
},
'event_type': excepted_pagerduty_event_type,
'incident_key': '',
'service_key': 'magicalbadgers',
}
mock_post_request.assert_called_once_with(
'https://events.pagerduty.com/generic/2010-04-15/create_event.json',
data=mock.ANY,
headers={'content-type': 'application/json'},
proxies=None)
assert expected_data == json.loads(
mock_post_request.call_args_list[0][1]['data'])
assert ('elastalert', logging.INFO,
excepted_log_message) == caplog.record_tuples[0]