def test_es_log_extra_argument_insertion(self):
self.log.info("About to test elasticsearch insertion")
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
es_index_name="pythontest",
es_additional_fields={'App': 'Test', 'Environment': 'Dev'},
raise_on_indexing_exceptions=True)
es_test_server_is_up = handler.test_es_source()
self.log.info("ES services status is: {0!s}".format(es_test_server_is_up))
self.assertEqual(True, es_test_server_is_up)
log = logging.getLogger("PythonTest")
log.addHandler(handler)
log.warning("Extra arguments Message", extra={"Arg1": 300, "Arg2": 400})
self.assertEqual(1, len(handler._buffer))
self.assertEqual(handler._buffer[0]['Arg1'], 300)
self.assertEqual(handler._buffer[0]['Arg2'], 400)
self.assertEqual(handler._buffer[0]['App'], 'Test')
self.assertEqual(handler._buffer[0]['Environment'], 'Dev')
handler.flush()
self.assertEqual(0, len(handler._buffer))
def test_buffered_log_insertion_flushed_when_buffer_full(self):
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name="pythontest",
use_ssl=True,
verify_ssl=True,
buffer_size=2,
flush_frequency_in_sec=1000,
es_additional_fields={'App': 'Test', 'Environment': 'Dev'},
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
raise_on_indexing_exceptions=True)
es_test_server_is_up = handler.test_es_source()
self.log.info("ES services status is: {0!s}".format(es_test_server_is_up))
self.assertEqual(True, es_test_server_is_up)
log = logging.getLogger("PythonTest")
log.setLevel(logging.DEBUG)
log.addHandler(handler)
log.warning("First Message")
log.info("Seccond Message")
self.assertEqual(0, len(handler._buffer))
handler.close()
def test_ping(self):
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name="pythontest",
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
raise_on_indexing_exceptions=True)
es_test_server_is_up = handler.test_es_source()
self.assertEqual(True, es_test_server_is_up)
def test_fast_insertion_of_hundred_logs(self):
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
buffer_size=500,
flush_frequency_in_sec=0.5,
es_index_name="pythontest",
raise_on_indexing_exceptions=True)
log = logging.getLogger("PythonTest")
log.setLevel(logging.DEBUG)
log.addHandler(handler)
for i in range(100):
log.info("Logging line {0:d}".format(i), extra={'LineNum': i})
handler.flush()
self.assertEqual(0, len(handler._buffer))
def test_index_name_frequency_functions(self):
index_name = "pythontest"
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name=index_name,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
index_name_frequency=CMRESHandler.IndexNameFrequency.DAILY,
raise_on_indexing_exceptions=True)
self.assertEqual(
handler._index_name_func.__func__(index_name),
CMRESHandler._get_daily_index_name(index_name)
)
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name=index_name,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
index_name_frequency=CMRESHandler.IndexNameFrequency.WEEKLY,
raise_on_indexing_exceptions=True)
self.assertEqual(
handler._index_name_func.__func__(index_name),
CMRESHandler._get_weekly_index_name(index_name)
)
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name=index_name,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
index_name_frequency=CMRESHandler.IndexNameFrequency.MONTHLY,
raise_on_indexing_exceptions=True)
self.assertEqual(
handler._index_name_func.__func__(index_name),
CMRESHandler._get_monthly_index_name(index_name)
)
handler = CMRESHandler(hosts=[{'host': self.getESHost(), 'port': self.getESPort()}],
auth_type=CMRESHandler.AuthType.HTTPS,
es_index_name=index_name,
use_ssl=True,
verify_ssl=True,
ca_certs="tests/certs/ca.pem",
client_cert="tests/certs/kirk.crtfull.pem",
client_key="tests/certs/kirk.key.pem",
auth_details=('admin', 'admin'),
index_name_frequency=CMRESHandler.IndexNameFrequency.YEARLY,
raise_on_indexing_exceptions=True)
self.assertEqual(
handler._index_name_func.__func__(index_name),
CMRESHandler._get_yearly_index_name(index_name)
)