自定义sys.excepthook不能与pytest一起使用

时间:2015-03-13 15:37:53

标签: python pytest

我想将pytest aserts的结果放入日志中。

首先我尝试了这个解决方案

def logged_assert(self, testval, msg=None):
    if not testval:
        if msg is None:
            try:
                assert testval
            except AssertionError as e:
                self.logger.exception(e)
                raise e
        self.logger.error(msg)
        assert testval, msg

它的工作正常但我需要为每个断言使用我自己的msg而不是内置。问题是testval评估它何时传入函数并且错误消息是

AssertionError: False

我在第一条评论中找到了解决问题http://code.activestate.com/recipes/577074-logging-asserts/的绝佳方法。

我在记录器包装器模块中编写了这个函数

def logged_excepthook(er_type, value, trace):
    print('HOOK!')
    if isinstance(er_type, AssertionError):
        current = sys.modules[sys._getframe(1).f_globals['__name__']]
        if 'logger' in sys.modules[current]:
            sys.__excepthook__(er_type, value, trace)
            sys.modules[current].error(exc_info=(er_type, value, trace))
        else:
            sys.__excepthook__(er_type, value, trace)
    else:
        sys.__excepthook__(er_type, value, trace)

然后

sys.excepthook = logged_excepthook

在测试模块中,我有断言输出

import sys
print(sys.excepthook, sys.__excepthook__, logged_excepthook)

<function logged_excepthook at 0x02D672B8> <built-in function excepthook> <function logged_excepthook at 0x02D672B8>

但我的输出中没有'Hook'消息。并且我的日志文件中也没有ERROR消息。所有的工作都与内置的sys.excepthook一样。

我查看了pytest源代码,但sys.excepthook在那里没有改变。 但如果我用Cntrl-C中断我的代码执行,我在stdout中收到了'Hook'消息。

主要的问题是为什么内置的sys.excepthook被称为isntead我的自定义函数,我该如何解决这个问题。 但是,如果存在记录断言错误的另一种方法,它对我也很有意义。

我在64位Windows 8.1上使用python3.2(32位)。

1 个答案:

答案 0 :(得分:2)

仅当存在未处理的异常(即通常终止您的程序的异常)时才会触发

excepthook。测试中的任何异常都由测试框架处理。

请参阅Asserting with the assert statement - pytest documentation有关如何使用该功能的信息。自定义消息以标准方式指定:assert condition, failure_message

如果您对pytest处理assert的方式不满意,则需要

pytest也使用assert钩子。其逻辑位于Lib\site-packages\_pytest\assertion(a stock plugin)。它可能足以包装/替换那里的一些功能。为了避免修补代码库,您可以使用自己的插件:在运行时修补exceptions插件,或者 禁用它并自己重用它的功能。