class TestCLILogging(unittest.TestCase):
def setUp(self):
self.mock_ctx = MockContext()
self.cli_logging = CLILogging('clitest', cli_ctx=self.mock_ctx)
def test_determine_log_level_default(self):
argv = []
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 2
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_verbose(self):
argv = ['--verbose']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 3
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_debug(self):
argv = ['--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_v_v_v_default(self):
argv = ['--verbose', '--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_only_show_errors(self):
argv = ['--only-show-errors']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 1
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_only_show_errors_config(self):
argv = []
self.mock_ctx.only_show_errors = True
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 1
self.assertEqual(actual_level, expected_level)
self.mock_ctx.only_show_errors = False
def test_determine_log_level_all_flags(self):
argv = ['--verbose', '--debug', '--only-show-errors']
with self.assertRaises(CLIError):
self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
def test_determine_log_level_other_args_verbose(self):
argv = ['account', '--verbose']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 3
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_other_args_debug(self):
argv = ['account', '--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_get_cli_logger(self):
logger = get_logger()
self.assertEqual(logger.name, CLI_LOGGER_NAME)
def test_get_module_logger(self):
module_logger = get_logger('a.module')
self.assertEqual(module_logger.name, 'cli.a.module')
def test_get_console_log_levels(self):
# CRITICAL
self.cli_logging.log_level = 0
levels = self.cli_logging._get_console_log_levels()
expected = {'cli': 50, 'root': 50}
self.assertEqual(levels, expected)
# ERROR
self.cli_logging.log_level = 1
levels = self.cli_logging._get_console_log_levels()
expected = {'cli': 40, 'root': 50}
self.assertEqual(levels, expected)
# WARNING
self.cli_logging.log_level = 2
levels = self.cli_logging._get_console_log_levels()
expected = {'cli': 30, 'root': 50}
self.assertEqual(levels, expected)
# INFO
self.cli_logging.log_level = 3
levels = self.cli_logging._get_console_log_levels()
expected = {'cli': 20, 'root': 50}
self.assertEqual(levels, expected)
# DEBUG
self.cli_logging.log_level = 4
levels = self.cli_logging._get_console_log_levels()
expected = {'cli': 10, 'root': 10}
self.assertEqual(levels, expected)
def test_get_console_log_formats(self):
# DEBUG level, color enabled
self.cli_logging.log_level = 4
self.cli_logging.cli_ctx.enable_color = True
formats = self.cli_logging._get_console_log_formats()
expected = {'cli': '%(name)s: %(message)s', 'root': '%(name)s: %(message)s'}
self.assertEqual(formats, expected)
# DEBUG level, color disabled
self.cli_logging.log_level = 4
self.cli_logging.cli_ctx.enable_color = False
formats = self.cli_logging._get_console_log_formats()
expected = {'cli': '%(levelname)s: %(name)s: %(message)s', 'root': '%(levelname)s: %(name)s: %(message)s'}
self.assertEqual(formats, expected)
# WARNING level, color enabled
self.cli_logging.log_level = 2
self.cli_logging.cli_ctx.enable_color = True
formats = self.cli_logging._get_console_log_formats()
expected = {'cli': '%(message)s', 'root': '%(message)s'}
self.assertEqual(formats, expected)
# WARNING level, color disabled
self.cli_logging.log_level = 2
self.cli_logging.cli_ctx.enable_color = False
formats = self.cli_logging._get_console_log_formats()
expected = {'cli': '%(levelname)s: %(message)s', 'root': '%(levelname)s: %(message)s'}
self.assertEqual(formats, expected)
class TestLoggingLevel(unittest.TestCase):
def setUp(self):
self.mock_ctx = MockContext()
self.cli_logging = CLILogging('clitest', cli_ctx=self.mock_ctx)
def test_determine_log_level_default(self):
argv = []
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 2
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_verbose(self):
argv = ['--verbose']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 3
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_debug(self):
argv = ['--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_v_v_v_default(self):
argv = ['--verbose', '--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_only_show_errors(self):
argv = ['--only-show-errors']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 1
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_only_show_errors_config(self):
argv = []
self.mock_ctx.only_show_errors = True
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 1
self.assertEqual(actual_level, expected_level)
self.mock_ctx.only_show_errors = False
def test_determine_log_level_all_flags(self):
argv = ['--verbose', '--debug', '--only-show-errors']
with self.assertRaises(CLIError):
self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
def test_determine_log_level_other_args_verbose(self):
argv = ['account', '--verbose']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 3
self.assertEqual(actual_level, expected_level)
def test_determine_log_level_other_args_debug(self):
argv = ['account', '--debug']
actual_level = self.cli_logging._determine_log_level(argv) # pylint: disable=protected-access
expected_level = 4
self.assertEqual(actual_level, expected_level)
def test_get_cli_logger(self):
logger = get_logger()
self.assertEqual(logger.name, CLI_LOGGER_NAME)
def test_get_module_logger(self):
module_logger = get_logger('a.module')
self.assertEqual(module_logger.name, 'cli.a.module')