33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
import logging
|
|
|
|
from flexmock import flexmock
|
|
|
|
from borgmatic import verbosity as module
|
|
|
|
|
|
def insert_logging_mock(log_level):
|
|
'''
|
|
Mock the isEnabledFor from Python logging.
|
|
'''
|
|
logging = flexmock(module.logging.Logger)
|
|
logging.should_receive('isEnabledFor').replace_with(lambda level: level >= log_level)
|
|
logging.should_receive('getEffectiveLevel').replace_with(lambda: log_level)
|
|
|
|
|
|
def test_verbosity_to_log_level_maps_known_verbosity_to_log_level():
|
|
flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
|
|
flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
|
|
flexmock(module.logging).DISABLED = module.borgmatic.logger.DISABLED
|
|
|
|
assert module.verbosity_to_log_level(module.VERBOSITY_ERROR) == logging.ERROR
|
|
assert module.verbosity_to_log_level(module.VERBOSITY_ANSWER) == module.borgmatic.logger.ANSWER
|
|
assert module.verbosity_to_log_level(module.VERBOSITY_SOME) == logging.INFO
|
|
assert module.verbosity_to_log_level(module.VERBOSITY_LOTS) == logging.DEBUG
|
|
assert module.verbosity_to_log_level(module.VERBOSITY_DISABLED) == logging.DISABLED
|
|
|
|
|
|
def test_verbosity_to_log_level_maps_unknown_verbosity_to_warning_level():
|
|
flexmock(module.borgmatic.logger).should_receive('add_custom_log_levels')
|
|
flexmock(module.logging).ANSWER = module.borgmatic.logger.ANSWER
|
|
|
|
assert module.verbosity_to_log_level('my pants') == logging.WARNING
|