设置失败时，如何清理Python Unitest？

Question

假设我有以下python Unittest：

import unittest

def Test(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # Get some resources
        ...

        if error_occurred:
            assert(False)

    @classmethod
    def tearDownClass(cls):
        # release resources
        ...

如果设置级呼叫失败，则不会调用拆除式级别，因此资源永远不会发布。如果下一个测试需要资源，则在测试过程中是一个问题。

当SetupClass调用失败时，有没有办法进行清理？

Answer 1

您可以在SetupClass方法中尝试一个尝试捕获，并直接调用除外的拆卸。

def setUpClass(cls):
     try: 
         # setUpClassInner()
     except Exception, e:
        cls.tearDownClass()
        raise # to still mark the test as failed.

要求外部资源运行您的UNITDEST是不好的做法。如果这些资源不可用，您需要测试代码的一部分是否有一个奇怪的错误，您将无法快速运行它。尝试将集成测试与单位测试区分开。

Answer 2

您在其他地方保护资源的方式相同。 try-except：

def setUpClass(cls):
    # ... acquire resources 
    try:
        # ... some call that may fail
    except SomeError, e:
        # cleanup here

清理可以像调用cls.teardownclass（）中的中一样简单。 然后您可以调用servert（false）或任何您喜欢尽早退出测试的方法。

Answer 3

我有一大堆测试助手功能，可以使用测试实例，并使用addCleanup清洁设置 /拆除线程，临时文件等，因此我也需要AddCleanup API来用于课堂级灯具。我重新完成了一些Unittest Docleanup功能来帮助我，并在课堂设置中使用模拟来修补AddCleanup（）

import unittest
import logging
import mock

LOGGER = logging.getLogger(__name__)

class ClassCleanupTestCase(unittest.TestCase):
    _class_cleanups = []

    @classmethod
    def setUpClassWithCleanup(cls):
        def cleanup_fn():
            """Do some cleanup!"""
        # Do something that requires cleanup
        cls.addCleanup(cleanup_fn)

    @classmethod
    def addCleanupClass(cls, function, *args, **kwargs):
        cls._class_cleanups.append((function, args, kwargs))

    @classmethod
    def doCleanupsClass(cls):
        results = []
        while cls._class_cleanups:
            function, args, kwargs = cls._class_cleanups.pop()
            try:
                function(*args, **kwargs)
            except Exceptions:
                LOGGER.exception('Exception calling class cleanup function')
                results.append(sys.exc_info())

        if results:
            LOGGER.error('Exception(s) raised during class cleanup, re-raising '
                         'first exception.')
            raise results[0]

    @classmethod
    def setUpClass(cls):
        try:
            with mock.patch.object(cls, 'addCleanup') as cls_addCleanup:
                cls_addCleanup.side_effect = cls.addCleanupClass
                cls.setUpClassWithCleanup()
        except Exception:
            cls.doCleanupsClass()
            raise

    @classmethod
    def tearDownClass(cls):
        cls.doCleanupsClass()

Answer 4

同时，addClassCleanup类方法已添加到unittest.testcase的确切目的： https://docs.python.org/3/library/library/unittest.html#unittest.test.testcase.testcase.addclasscleanup