def test_get_summary(self):
logger = Logger()
logger.get_warnings = MagicMock(return_value=['the warnings'])
logger.get_errors = MagicMock(return_value=['the errors'])
logger.get_log_counts = MagicMock(return_value={'the': 'log counts'})
self.assertEqual({
'warnings': ['the warnings'],
'errors': ['the errors'],
'log_counts': {'the': 'log counts'}
}, logger.get_summary())
def test_get_warnings(self):
logger = Logger()
logger.messages['warning'] = 'warning messages'
self.assertEqual('warning messages', logger.get_warnings())
def test_get_warnings(self):
logger = Logger()
logger.messages['warning'] = ['warning messages']
self.assertEqual(['warning messages'], logger.get_warnings())
logger.messages['data_warning'] = ['data warning messages']
self.assertEqual(['warning messages', 'data warning messages'], logger.get_warnings())