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())