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 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')