def _update_progress(
self, match_method, items_to_complete, command_output
):
items_count = len(items_to_complete)
for item in items_to_complete:
if match_method(item, command_output):
self.items_processed += 1
if self.items_processed <= items_count:
Logger.progress(
self.items_processed, items_count,
'[ INFO ]: Processing'
)
def _stop_progress(self):
Logger.progress(
100, 100, '[ INFO ]: Processing'
)
def setup(self):
self.log = Logger('kiwi')
def _init_progress(self):
Logger.progress(
0, 100, '[ INFO ]: Processing'
)
class TestLogger:
def setup(self):
self.log = Logger('kiwi')
def setup_method(self, cls):
self.setup()
@patch('sys.stdout')
def test_progress(self, mock_stdout):
self.log.progress(50, 100, 'foo')
mock_stdout.write.assert_called_once_with(
'\rfoo: [#################### ] 50%'
)
mock_stdout.flush.assert_called_once_with()
def test_progress_raise(self):
assert self.log.progress(50, 0, 'foo') is None
@patch('logging.FileHandler')
def test_set_logfile(self, mock_file_handler):
self.log.set_logfile('logfile')
mock_file_handler.assert_called_once_with(
filename='logfile', encoding='utf-8'
)
assert self.log.get_logfile() == 'logfile'
@patch('logging.StreamHandler')
def test_set_logfile_to_stdout(self, mock_stream_handler):
self.log.set_logfile('stdout')
mock_stream_handler.assert_called_once_with(
sys.__stdout__
)
assert self.log.get_logfile() is None
@patch('kiwi.logger.ColorFormatter')
def test_set_color_format(self, mock_color_format):
self.log.set_color_format()
assert sorted(mock_color_format.call_args_list) == [
call(
'$COLOR[ %(levelname)-8s]: %(asctime)-8s | %(message)s',
'%H:%M:%S'
),
call(
'$COLOR[ %(levelname)-8s]: %(asctime)-8s | %(message)s',
'%H:%M:%S'
),
call(
'$LIGHTCOLOR[ %(levelname)-8s]: %(asctime)-8s | %(message)s',
'%H:%M:%S'
)
]
@patch('logging.FileHandler')
def test_set_logfile_raise(self, mock_file_handler):
mock_file_handler.side_effect = KiwiLogFileSetupFailed
with raises(KiwiLogFileSetupFailed):
self.log.set_logfile('logfile')
def test_getLogLevel(self):
self.log.setLogLevel(42)
assert self.log.getLogLevel() == 42
def test_getLogFlags(self):
assert self.log.getLogFlags().get('run-scripts-in-screen') is None
self.log.setLogFlag('run-scripts-in-screen')
assert self.log.getLogFlags().get('run-scripts-in-screen') is True