如何仅使用python unittest2在测试失败时执行代码？

Question

我在python的unittest2框架中运行了一些基于类的单元测试。我们正在使用Selenium
WebDriver，它具有便捷的save_screenshot()方法。我想在每次测试失败时在tearDown（）中抓取一个屏幕截图，以减少调试_为什么_ 测试失败的时间。

但是，我找不到仅在测试失败时运行代码的任何方法。tearDown()不管测试是否成功，都会调用，并且我不想用成百上千的浏览器屏幕快照来使我们的文件系统混乱，以确保测试成功。

您将如何处理？

Answer 1

找到了解决方案-我可以覆盖failureException：

@property
def failureException(self):
    class MyFailureException(AssertionError):
        def __init__(self_, *args, **kwargs):
            self.b.save_screenshot('%s.png' % self.id())
            return super(MyFailureException, self_).__init__(*args, **kwargs)
    MyFailureException.__name__ = AssertionError.__name__
    return MyFailureException

这似乎难以置信，但到目前为止，它似乎仍然有效。