def test_clear_issues(self, mock_os):
logger = Logger()
logger._issues = {'1': 1}
logger._data_msg_count = {'any': 'value'}
logger._offload_file = 'any file'
logger._offload_filename = 'any filename'
# Assert all variables have been cleared
logger.clear_issues()
self.assertEqual(logger._issues, {})
self.assertEqual(logger._offload_file, None)
mock_os.remove.assert_called_with('any filename')
# Check an OSError is passed without error
mock_os.remove.side_effect = OSError()
logger._offload_filename = 'any non existing filename'
logger.clear_issues()
def test_get_log_counts(self):
logger = Logger()
logger._data_msg_count = MagicMock()
self.assertEqual(logger._data_msg_count, logger.get_log_counts())
