class TestFileLogger(unittest.TestCase): def setUp(self): robottime.TIMESTAMP_CACHE = _FakeTimeCache() FileLogger._get_writer = lambda *args: StringIO() self.logger = FileLogger('whatever', 'INFO') def tearDown(self): robottime.TIMESTAMP_CACHE = TimestampCache() def test_write(self): self.logger.write('my message', 'INFO') expected = '20060613 08:37:42.123 | INFO | my message\n' self._verify_message(expected) self.logger.write('my 2nd msg\nwith 2 lines', 'ERROR') expected += '20060613 08:37:42.123 | ERROR | my 2nd msg\nwith 2 lines\n' self._verify_message(expected) def test_write_helpers(self): self.logger.info('my message') expected = '20060613 08:37:42.123 | INFO | my message\n' self._verify_message(expected) self.logger.warn('my 2nd msg\nwith 2 lines') expected += '20060613 08:37:42.123 | WARN | my 2nd msg\nwith 2 lines\n' self._verify_message(expected) def test_set_level(self): self.logger.write('msg', 'DEBUG') self._verify_message('') self.logger.set_level('DEBUG') self.logger.write('msg', 'DEBUG') self._verify_message('20060613 08:37:42.123 | DEBUG | msg\n') def _verify_message(self, expected): assert_equals(self.logger._writer.getvalue(), expected)
class TestFileLogger(unittest.TestCase): def setUp(self): FileLogger._get_writer = lambda *args: StringIO() self.logger = FileLogger('whatever', 'INFO') utils.robottime._CURRENT_TIME = (2006, 6, 13, 8, 37, 42, 123) def tearDown(self): utils.robottime._CURRENT_TIME = None def test_write(self): self.logger.write('my message', 'INFO') expected = '20060613 08:37:42.123 | INFO | my message\n' assert_equals(self.logger._writer.getvalue(), expected) self.logger.write('my 2nd msg\nwith 2 lines', 'ERROR') expected += '20060613 08:37:42.123 | ERROR | my 2nd msg\nwith 2 lines\n' assert_equals(self.logger._writer.getvalue(), expected) def test_write_helpers(self): self.logger.info('my message') expected = '20060613 08:37:42.123 | INFO | my message\n' assert_equals(self.logger._writer.getvalue(), expected) self.logger.warn('my 2nd msg\nwith 2 lines') expected += '20060613 08:37:42.123 | WARN | my 2nd msg\nwith 2 lines\n' assert_equals(self.logger._writer.getvalue(), expected) def test_set_level(self): self.logger.write('msg', 'DEBUG') assert_equals(self.logger._writer.getvalue(), '') self.logger.set_level('DEBUG') self.logger.write('msg', 'DEBUG') assert_equals(self.logger._writer.getvalue(), '20060613 08:37:42.123 | DEBUG | msg\n')