6

有没有一些简单的方法来捕获和断言记录的消息nose？

例如，我希望能够执行以下操作：

cook_eggs()
assert_logged("eggs are ready!")
4

3 回答 3

12

您可以创建一个自定义处理程序，该处理程序可以检查通过日志记录发送的消息。BufferingHandler非常适合这项工作。

您可能还希望在测试中将处理程序附加到您在代码中使用的任何记录器，例如logging.getLogger('foo').addHandler(...). 您最终可以将处理程序附加到测试用例的setUp和方法中。tearDown

import logging
import logging.handlers

class AssertingHandler(logging.handlers.BufferingHandler):

    def __init__(self,capacity):
        logging.handlers.BufferingHandler.__init__(self,capacity)

    def assert_logged(self,test_case,msg):
        for record in self.buffer:
            s = self.format(record)
            if s == msg:
                return
        test_case.assertTrue(False, "Failed to find log message: " + msg)


def cook_eggs():
    logging.warn("eggs are ready!")


import unittest

class TestLogging(unittest.TestCase):

    def test(self):
        asserting_handler = AssertingHandler(10)
        logging.getLogger().addHandler(asserting_handler)
        cook_eggs() 
        asserting_handler.assert_logged(self,"eggs are ready!")
        logging.getLogger().removeHandler(asserting_handler)


unittest.main()
于 2011-02-23T03:07:03.207 回答
2

这就是“模拟对象”的用途。

您可以使用模拟版本的日志记录，它将正确缓冲日志消息，以便您以后可以对它们进行断言。

于 2011-02-22T23:27:10.450 回答
0

只是 FWIW，在 datalad 项目中，我们需要类似的功能，但也需要吞下日志（可能还有内省）。所以这里出现了解决方案——swallow_logs 上下文处理程序：https ://github.com/datalad/datalad/blob/master/datalad/utils.py#L296 （目前在 b633c9da46ab9cccde3d4767928d167a91857153）。所以现在在测试中我们像

def test_swallow_logs():
    lgr = logging.getLogger('datalad')
    with swallow_logs(new_level=9) as cm:
        eq_(cm.out, '')
        lgr.log(8, "very heavy debug")
        eq_(cm.out, '')  # not even visible at level 9
        lgr.log(9, "debug1")
        eq_(cm.out, 'debug1\n')  # not even visible at level 9
        lgr.info("info")
        eq_(cm.out, 'debug1\ninfo\n')  # not even visible at level 9
于 2015-09-18T04:24:06.493 回答