自定义 sys.excepthook 不适用于 pytest

Question

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

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

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 会在它传递给函数时进行评估，而错误 msg 是

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”消息。而且我的日志文件中也没有错误消息。所有的作品都与内置的 sys.excepthook 一样。

我查看了 pytest 源，但 sys.excepthook 并没有改变。 但是如果我用 Cntrl-C 中断我的代码执行，我会在标准输出中收到“Hook”消息。

主要问题是为什么内置 sys.excepthook 调用而不是我的自定义函数，我该如何解决这个问题。 但如果存在另一种记录断言错误的方法，我也很感兴趣。

我在 64 位 windows 8.1 上使用 python3.2（32 位）。

Answer 1

excepthook 仅在存在 未处理 异常时触发，即正常终止程序的异常。测试中的任何异常都由测试框架处理。

请参阅Asserting with the assert statement - pytest documentation，了解如何使用该功能。自定义消息以标准方式指定：assert condition, failure_message。

如果您对pytest 处理asserts 的方式不满意，您需要这样做

• 使用包装器，或
• hook the assert statement。

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