def test_filter_silenced(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.filter_silenced(), True)
sentinel.Response.status_code = 404
self.assertEqual(handler.filter_silenced(), False)
def test_stash_exists(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.stash_exists('/foo'), True)
sentinel.Response.status_code = 404
self.assertEqual(handler.stash_exists('/foo'), False)
def setUp(self):
self.cfg_file_1 = 'tests/config.json'
self.cfg_file_2 = 'tests/checks.json'
self.sample_event_file = 'tests/event.json'
os.environ['SENSU_CONFIG_FILES'] = '{0}:{1}'.format(self.cfg_file_1,
self.cfg_file_2)
self.evt_json = open(self.sample_event_file, 'r').read()
os.environ['SENSU_EVENT'] = self.evt_json
os.environ['SENSU_TEST'] = 'True'
self.handler = Handler()
def test_filter_dependencies(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
# check hit
sentinel.Response.status_code = 200
handler.event['check']['dependencies'] = ['foo', 'bar']
mock_method.return_value = sentinel.Response
self.assertEqual(handler.filter_dependencies(), True)
# check miss
sentinel.Response.status_code = 404
del handler.event['check']['dependencies']
self.assertEqual(handler.filter_dependencies(), False)
def test_event_exists(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
# check hit
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.event_exists('sensu.test', 'cron'), True)
mock_method.assert_called_with('/event/sensu.test/cron')
# check miss
sentinel.Response.status_code = 404
self.assertEqual(handler.event_exists('sensu.test', 'cron'), False)
mock_method.assert_called_with('/event/sensu.test/cron')
class HandlerTestCase(unittest.TestCase):
def setUp(self):
self.cfg_file_1 = 'tests/config.json'
self.cfg_file_2 = 'tests/checks.json'
self.sample_event_file = 'tests/event.json'
os.environ['SENSU_CONFIG_FILES'] = '{0}:{1}'.format(self.cfg_file_1,
self.cfg_file_2)
self.evt_json = open(self.sample_event_file, 'r').read()
os.environ['SENSU_EVENT'] = self.evt_json
os.environ['SENSU_TEST'] = 'True'
self.handler = Handler()
def tearDown(self):
pass
def test_handler_settings(self):
self.assertTrue(self.handler.settings.has_key('api'))
self.assertTrue(self.handler.settings.has_key('handlers'))
self.assertTrue(self.handler.settings.has_key('checks'))
def test_handler_event(self):
self.assertNotEqual(self.handler.event, None)
self.assertTrue(self.handler.event.has_key('check'))
self.assertTrue(self.handler.event.has_key('action'))
def test_handle(self):
with self.assertRaises(HandlerException):
self.handler.handle()
def test_filter_disabled(self):
self.assertEqual(self.handler.filter_disabled(), False)
self.handler.event['check']['alert'] = False
self.assertEqual(self.handler.filter_disabled(), True)
def test_filter_repeated(self):
self.handler.event['occurrences'] = 2
self.assertEqual(self.handler.filter_repeated(), True)
self.handler.event['occurrences'] = 60
self.assertEqual(self.handler.filter_repeated(), False)
def test_stash_exists(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.stash_exists('/foo'), True)
sentinel.Response.status_code = 404
self.assertEqual(handler.stash_exists('/foo'), False)
def test_filter_silenced(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.filter_silenced(), True)
sentinel.Response.status_code = 404
self.assertEqual(handler.filter_silenced(), False)
def test_event_exists(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
# check hit
sentinel.Response.status_code = 200
mock_method.return_value = sentinel.Response
self.assertEqual(handler.event_exists('sensu.test', 'cron'), True)
mock_method.assert_called_with('/event/sensu.test/cron')
# check miss
sentinel.Response.status_code = 404
self.assertEqual(handler.event_exists('sensu.test', 'cron'), False)
mock_method.assert_called_with('/event/sensu.test/cron')
def test_filter_dependencies(self):
handler = Handler()
with patch.object(Handler, '_api_request') as mock_method:
# check hit
sentinel.Response.status_code = 200
handler.event['check']['dependencies'] = ['foo', 'bar']
mock_method.return_value = sentinel.Response
self.assertEqual(handler.filter_dependencies(), True)
# check miss
sentinel.Response.status_code = 404
del handler.event['check']['dependencies']
self.assertEqual(handler.filter_dependencies(), False)
