python - 使用 unittest 测试记录器消息

Question

我想在我的单元测试中测试记录器消息而不将它们打印到屏幕上。鉴于此代码:

 import logging
 logging.basicConfig(level=logging.ERROR)
 logger = logging.getLogger('test')
 logger.warning('this is a warning')

 # How do I see that there was a warning?

如何查看记录器中的日志记录以查看是否有警告？我在 Logger 中找不到可以完成这项工作的迭代器。

Answer 1

您可能需要使用 TestCase.assertLogs()在这种情况下的上下文管理器。该文档提供了一个很好的示例，说明可以用它做什么:

with self.assertLogs('foo', level='INFO') as cm:
   logging.getLogger('foo').info('first message')
   logging.getLogger('foo.bar').error('second message')
self.assertEqual(cm.output, ['INFO:foo:first message',
                             'ERROR:foo.bar:second message'])

在上下文管理器中，您可以访问 cm.records 获取 LogRecord 实例列表，或访问 cm.output 获取列表格式化消息