def setUp(self):
self.s3_event = MagicMock()
self.s3_event.objects = [('a-bucket', 'an-object')]
self.handler = ElbLogEventHandler(self.s3_event)
mock_actor = MagicMock()
self.handler.actors = [mock_actor]
class TestParseElbLog(unittest.TestCase):
def setUp(self):
self.handler = ElbLogEventHandler("{}")
@patch(module + ".Logstash")
def test_call(self, mock_logstash_cls):
mock_logstash_cls.return_value = mock_logstash_obj = MagicMock()
self.handler.parse_elb_log("/somewhere")
mock_logstash_obj.parse.assert_called_with("/somewhere")
class TestParseElbLog(unittest.TestCase):
def setUp(self):
self.handler = ElbLogEventHandler('{}')
@patch(module + '.Logstash')
def test_call(self, mock_logstash_cls):
mock_logstash_cls.return_value = mock_logstash_obj = MagicMock()
self.handler.parse_elb_log('/somewhere')
mock_logstash_obj.parse.assert_called_with('/somewhere')
class TestDownloadElbLog(unittest.TestCase):
def setUp(self):
self.handler = ElbLogEventHandler(MagicMock())
@patch(module + ".S3")
def test_call(self, mock_s3_cls):
mock_s3_cls.return_value = mock_s3_obj = MagicMock()
self.handler.download_elb_log("a-bucket", "an-object", "/somewhere")
mock_s3_cls.assert_called_with("a-bucket")
mock_s3_obj.download.assert_called_with("an-object", "/somewhere")
class TestDownloadElbLog(unittest.TestCase):
def setUp(self):
self.handler = ElbLogEventHandler(MagicMock())
@patch(module + '.S3')
def test_call(self, mock_s3_cls):
mock_s3_cls.return_value = mock_s3_obj = MagicMock()
self.handler.download_elb_log('a-bucket', 'an-object', '/somewhere')
mock_s3_cls.assert_called_with('a-bucket')
mock_s3_obj.download.assert_called_with('an-object', '/somewhere')
def setUp(self):
self.s3_event = MagicMock()
self.s3_event.objects = [("a-bucket", "an-object")]
self.handler = ElbLogEventHandler(self.s3_event)
mock_actor = MagicMock()
self.handler.actors = [mock_actor]
class TestHandle(unittest.TestCase):
def setUp(self):
self.s3_event = MagicMock()
self.s3_event.objects = [("a-bucket", "an-object")]
self.handler = ElbLogEventHandler(self.s3_event)
mock_actor = MagicMock()
self.handler.actors = [mock_actor]
def test_call(self):
self.handler.download_elb_log = mock_download_elb_log = MagicMock()
self.handler.parse_elb_log = mock_parse_elb_log = MagicMock()
mock_api_record = MagicMock()
mock_parse_elb_log.return_value = [mock_api_record]
self.handler.handle()
self.assertEqual(mock_download_elb_log.call_args[0][0], "a-bucket")
self.assertEqual(mock_download_elb_log.call_args[0][1], "an-object")
self.assertEqual(mock_download_elb_log.call_args[0][2], mock_parse_elb_log.call_args[0][0])
self.handler.actors[0].notify.assert_called_with(mock_api_record)
class TestHandle(unittest.TestCase):
def setUp(self):
self.s3_event = MagicMock()
self.s3_event.objects = [('a-bucket', 'an-object')]
self.handler = ElbLogEventHandler(self.s3_event)
mock_actor = MagicMock()
self.handler.actors = [mock_actor]
def test_call(self):
self.handler.download_elb_log = mock_download_elb_log = MagicMock()
self.handler.parse_elb_log = mock_parse_elb_log = MagicMock()
mock_api_record = MagicMock()
mock_parse_elb_log.return_value = [mock_api_record]
self.handler.handle()
self.assertEqual(mock_download_elb_log.call_args[0][0], 'a-bucket')
self.assertEqual(mock_download_elb_log.call_args[0][1], 'an-object')
self.assertEqual(mock_download_elb_log.call_args[0][2],
mock_parse_elb_log.call_args[0][0])
self.handler.actors[0].notify.assert_called_with(mock_api_record)
def watch_sqs():
queue = get_queue()
if queue is None:
msg = 'Cannot find queue'
logger.error(msg)
raise ValueError(msg)
message = queue.read(wait_time_seconds=20)
if message is None:
logger.info('No message available now')
return
logger.info(message.get_body())
s3_event = S3Event(message.get_body())
try:
ElbLogEventHandler(s3_event).handle()
finally:
queue.delete_message(message)
logger.info('Delete SQS message')
def setUp(self):
self.handler = ElbLogEventHandler("{}")
def setUp(self):
self.handler = ElbLogEventHandler(MagicMock())
def setUp(self):
self.handler = ElbLogEventHandler('{}')
def setUp(self):
self.handler = ElbLogEventHandler(MagicMock())
