Python日志记录到StringIO处理程序
问题描述:
我有一个python测试,我想测试日志记录是否正常工作。例如,我有一个创建用户的函数,最后日志记录将响应写入日志文件。Python日志记录到StringIO处理程序
logger = logging.getLogger('mylogger')
logger.setLevel(logging.DEBUG)
handler = logging.handlers.WatchedFileHandler('mylogfile.log')
formatter = logging.Formatter('%(asctime)s: %(message)s',
'%d/%b/%Y:%H:%M:%S %z')
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.info('Some log text')
在我的测试案例中,我想将日志输出发送到StringIO。
class MyTest(unittest.TestCase):
def setUp(self):
stream = StringIO()
self.handler = logging.StreamHandler(stream)
log = logging.getLogger('mylogger')
log.removeHandler(log.handlers[0])
log.addHandler(self.handler)
def tearDown(self):
log = logging.getLogger('mylogger')
log.removeHandler(self.handler)
self.handler.close()
的问题是,我不知道我应该如何测试wheter或不是我的记录工作。
答
下面是一个可以工作的例子,确保你设置了你的日志级别并刷新缓冲区。
class MyTest(unittest.TestCase):
def setUp(self):
self.stream = StringIO()
self.handler = logging.StreamHandler(self.stream)
self.log = logging.getLogger('mylogger')
self.log.setLevel(logging.INFO)
for handler in self.log.handlers:
self.log.removeHandler(handler)
self.log.addHandler(self.handler)
def testLog(self):
self.log.info("test")
self.handler.flush()
print '[', self.stream.getvalue(), ']'
self.assertTrue(self.stream.getvalue(), 'test')
def tearDown(self):
self.log.removeHandler(self.handler)
self.handler.close()
if __name__=='__main__':
unittest.main()
进一步阅读,这里是一个示例Temporily Capturing Python Logging to a string buffer,它显示了你如何也可以做格式化。
对不起,这可能只是超越我,但你不只是做'stream.getvalue()'? – macduff 2012-03-02 14:25:04
相关:[PyDev单元测试:如何捕获记录到“Captured Output”中的logging.Logger的文本](http://stackoverflow.com/q/7472863/321973) – 2013-02-26 11:17:43