def test_alert_aggregation_summary_table_one_row():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'aggregation': 1,
'summary_table_fields': ['field', 'abc'],
'summary_table_max_rows': 1,
}
matches = [
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'cde from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'cde from match', },
]
alert = Alerter(rule)
summary_table = str(alert.get_aggregation_summary_text(matches))
assert "+-------------+----------------+-------+" in summary_table
assert "| field | abc | count |" in summary_table
assert "+=============+================+=======+" in summary_table
assert "| field_value | abc from match | 3 |" in summary_table
assert "| field_value | cde from match | 2 |" not in summary_table
assert "Showing top 1 rows" in summary_table
def test_alert_get_aggregation_summary_text__maximum_width():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'alert_subject_max_len': 5
}
alert = Alerter(rule)
assert 80 == alert.get_aggregation_summary_text__maximum_width()
def test_alert_subject_size_limit_no_args():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'alert_subject_max_len': 5
}
alert = Alerter(rule)
alertSubject = alert.create_custom_title([{'test_term': 'test_value', '@timestamp': '2014-10-31T00:00:00'}])
assert 5 == len(alertSubject)
def test_alert_getinfo():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'alert_subject_max_len': 5
}
alert = Alerter(rule)
actual_data = alert.get_info()
expected_data = {'type': 'Unknown'}
assert expected_data == actual_data
def test_alert_subject_size_limit_with_args(ea):
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'Test alert for {0} {1}',
'alert_subject_args': ['test_term', 'test.term'],
'alert_subject_max_len': 6
}
alert = Alerter(rule)
alertSubject = alert.create_custom_title([{'test_term': 'test_value', '@timestamp': '2014-10-31T00:00:00'}])
assert 6 == len(alertSubject)
def test_resolving_rule_references(ea):
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'list_of_things': [
'1',
'$owner$',
[
'11',
'$owner$',
],
],
'nested_dict': {
'nested_one': '1',
'nested_owner': '$owner$',
},
'resolved_string_reference': '$owner$',
'resolved_int_reference': '$priority$',
'unresolved_reference': '$foo$',
}
alert = Alerter(rule)
assert 'the_owner' == alert.rule['resolved_string_reference']
assert 2 == alert.rule['resolved_int_reference']
assert '$foo$' == alert.rule['unresolved_reference']
assert 'the_owner' == alert.rule['list_of_things'][1]
assert 'the_owner' == alert.rule['list_of_things'][2][1]
assert 'the_owner' == alert.rule['nested_dict']['nested_owner']
def test_alert_error():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'alert_subject_max_len': 5
}
match = {
'@timestamp': '2021-01-01T00:00:00',
'name': 'datadog-test-name'
}
alert = Alerter(rule)
try:
alert.alert([match])
except NotImplementedError:
assert True
def test_alert_subject_with_jinja():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'Test alert for {{owner}}; field {{field}}; Abc: {{_data["abc"]}}',
'alert_text_type': "alert_text_jinja",
'jinja_root_name': "_data"
}
match = {
'@timestamp': '2016-01-01',
'field': 'field_value',
'abc': 'abc from match',
}
alert = Alerter(rule)
alertsubject = alert.create_custom_title([match])
assert "Test alert for the_owner;" in alertsubject
assert "field field_value;" in alertsubject
assert "Abc: abc from match" in alertsubject
def test_alert_aggregation_summary_table_suffix_prefix():
rule = {
'name': 'test_rule',
'type': mock_rule(),
'owner': 'the_owner',
'priority': 2,
'alert_subject': 'A very long subject',
'aggregation': 1,
'summary_table_fields': ['field', 'abc'],
'summary_prefix': 'This is the prefix',
'summary_suffix': 'This is the suffix',
}
matches = [
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'abc from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'cde from match', },
{'@timestamp': '2016-01-01', 'field': 'field_value', 'abc': 'cde from match', },
]
alert = Alerter(rule)
summary_table = str(alert.get_aggregation_summary_text(matches))
assert "This is the prefix" in summary_table
assert "This is the suffix" in summary_table
def __init__(self, rule):
Alerter.__init__(self, rule)
self.template_id = os.environ['GOVUK_NOTIFY_TEMPLATE_ID']
self.email_addresses = os.environ['NOTIFICATION_EMAILS'].split(',')
api_key = os.environ['GOVUK_NOTIFY_API_KEY']
self.notifications_client = NotificationsAPIClient(api_key)
