def set_logging(log_path='logs/api.log', foreground_mode=False, debug_mode='info'):
for logger_name in ('connexion.aiohttp_app', 'connexion.apis.aiohttp_api', 'wazuh-api'):
api_logger = alogging.APILogger(log_path=log_path, foreground_mode=foreground_mode,
debug_level='info' if logger_name != 'wazuh-api'
and debug_mode != 'debug2' else debug_mode,
logger_name=logger_name)
api_logger.setup_logger()
def test_apilogger_setup_logger(mock_logger, debug_level, expected_level):
"""Check loggin level is as expected"""
logger = alogging.APILogger(log_path='test_path',
foreground_mode=False,
debug_level=debug_level,
logger_name='wazuh')
logger.setup_logger()
assert mock_logger.call_args == call(expected_level)
def test_apilogger_setup_logger(mock_logger, debug_level, expected_level):
"""Check loggin level is as expected"""
current_logger_path = os.path.join(os.path.dirname(__file__), 'testing.log')
logger = alogging.APILogger(log_path=current_logger_path, foreground_mode=False, debug_level=debug_level,
logger_name='wazuh')
logger.setup_logger()
assert mock_logger.call_args == call(expected_level)
os.path.exists(current_logger_path) and os.remove(current_logger_path)
def test_apilogger_init(mock_wazuhlogger):
"""Check parameters are as expected when calling __init__ method"""
alogging.APILogger(log_path='test_path',
foreground_mode=False,
debug_level='info',
logger_name='wazuh')
assert mock_wazuhlogger.call_args.kwargs['log_path'] == 'test_path'
assert not mock_wazuhlogger.call_args.kwargs['foreground_mode']
assert mock_wazuhlogger.call_args.kwargs['debug_level'] == 'info'
assert mock_wazuhlogger.call_args.kwargs['logger_name'] == 'wazuh'
assert mock_wazuhlogger.call_args.kwargs[
'tag'] == '{asctime} {levelname}: {message}'
def test_apilogger_init(mock_wazuhlogger):
"""Check parameters are as expected when calling __init__ method"""
current_logger_path = os.path.join(os.path.dirname(__file__), 'testing.log')
alogging.APILogger(log_path=current_logger_path, foreground_mode=False, debug_level='info',
logger_name='wazuh')
assert mock_wazuhlogger.call_args.kwargs['log_path'] == current_logger_path
assert not mock_wazuhlogger.call_args.kwargs['foreground_mode']
assert mock_wazuhlogger.call_args.kwargs['debug_level'] == 'info'
assert mock_wazuhlogger.call_args.kwargs['logger_name'] == 'wazuh'
assert mock_wazuhlogger.call_args.kwargs['tag'] == '{asctime} {levelname}: {message}'
os.path.exists(current_logger_path) and os.remove(current_logger_path)
