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)