def setUp(self):
self.alert_key = 'ALERT KEY'
self.description = 'ALERT DESCRIPTION'
self.html_description = 'HTML ALERT DESCRIPTION'
self.mock_redis_storage = MagicMock(RedisStorage)
self.mock_hipchat_client = MagicMock(HipChat)
self.hcn = HipchatNotifier(self.mock_hipchat_client, self.mock_redis_storage)
def setUp(self):
self.alert_key = 'ALERT KEY'
self.description = 'ALERT DESCRIPTION'
self.html_description = 'HTML ALERT DESCRIPTION'
self.mock_redis_storage = MagicMock(RedisStorage)
self.mock_hipchat_client = MagicMock(HipChat)
self.hcn = HipchatNotifier(self.mock_hipchat_client, self.mock_redis_storage)
class TestHipChatNotifier(TestCase):
def setUp(self):
self.alert_key = 'ALERT KEY'
self.description = 'ALERT DESCRIPTION'
self.html_description = 'HTML ALERT DESCRIPTION'
self.mock_redis_storage = MagicMock(RedisStorage)
self.mock_hipchat_client = MagicMock(HipChat)
self.hcn = HipchatNotifier(self.mock_hipchat_client,
self.mock_redis_storage)
def test_should_not_notify_hipchat_if_no_rooms_have_been_added(self):
self.hcn.notify(self.alert_key, Level.WARNING, self.description,
self.html_description)
self.assertEqual(self.mock_hipchat_client.message_room.mock_calls, [])
def test_should_not_notify_hipchat_if_warning_and_already_notified(self):
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = True
self.hcn.notify(self.alert_key, Level.WARNING, self.description,
self.html_description)
self.assertEqual(self.mock_hipchat_client.mock_calls, [])
def test_should_notify_hipchat_resolved_if_nominal_and_had_notified(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = True
self.hcn.notify(self.alert_key, Level.NOMINAL, self.description,
self.html_description)
self.mock_redis_storage.is_locked_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='green',
)
self.mock_redis_storage.remove_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
def test_should_notify_room_of_warning_if_had_not_notified_before(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = False
self.hcn.notify(self.alert_key, Level.WARNING, self.description,
self.html_description)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='yellow')
self.mock_redis_storage.set_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
def test_should_notify_room_of_critical_if_had_not_notified_before(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = False
self.hcn.notify(self.alert_key, Level.CRITICAL, self.description,
self.html_description)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='red')
self.mock_redis_storage.set_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
class TestHipChatNotifier(TestCase):
def setUp(self):
self.alert_key = 'ALERT KEY'
self.description = 'ALERT DESCRIPTION'
self.html_description = 'HTML ALERT DESCRIPTION'
self.mock_redis_storage = MagicMock(RedisStorage)
self.mock_hipchat_client = MagicMock(HipChat)
self.hcn = HipchatNotifier(self.mock_hipchat_client, self.mock_redis_storage)
def test_should_not_notify_hipchat_if_no_rooms_have_been_added(self):
self.hcn.notify(self.alert_key, Level.WARNING, self.description, self.html_description)
self.assertEqual(self.mock_hipchat_client.message_room.mock_calls, [])
def test_should_not_notify_hipchat_if_warning_and_already_notified(self):
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = True
self.hcn.notify(self.alert_key, Level.WARNING, self.description, self.html_description)
self.assertEqual(self.mock_hipchat_client.mock_calls, [])
def test_should_notify_hipchat_resolved_if_nominal_and_had_notified(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = True
self.hcn.notify(self.alert_key, Level.NOMINAL, self.description, self.html_description)
self.mock_redis_storage.is_locked_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='green',
)
self.mock_redis_storage.remove_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
def test_should_notify_room_of_warning_if_had_not_notified_before(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = False
self.hcn.notify(self.alert_key, Level.WARNING, self.description, self.html_description)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='yellow'
)
self.mock_redis_storage.set_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
def test_should_notify_room_of_critical_if_had_not_notified_before(self):
room_name = 'ROOM NAME'
self.hcn.add_room(room_name)
self.mock_redis_storage.is_locked_for_domain_and_key.return_value = False
self.hcn.notify(self.alert_key, Level.CRITICAL, self.description, self.html_description)
self.mock_hipchat_client.message_room.assert_called_once_with(
room_name,
'Graphite-Pager',
self.html_description,
message_format='html',
color='red'
)
self.mock_redis_storage.set_lock_for_domain_and_key.assert_called_once_with(
'HipChat', self.alert_key)
