如何在setUpClass失败时清理Python UnitTest？

rhfm7lfc 于 2024-01-05 发布在 Python

关注(0)|答案(4)|浏览(178)

假设我有以下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调用失败，则不会调用tearDownClass，因此资源永远不会释放。如果下一个测试需要资源，则在测试运行期间会出现问题。
当setUpClass调用失败时，有没有一种方法可以进行清理？

python

来源：https://stackoverflow.com/questions/17538198/how-can-you-cleanup-a-python-unittest-when-setupclass-fails

4条答案

按热度按时间

nqwrtyyt1#

你可以在setUpClass方法中放置一个try catch，并直接调用except中的tearDown。

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

字符串
需要外部资源来运行单元测试是不好的做法。如果这些资源不可用，而你需要测试代码的一部分，以发现奇怪的错误，你将无法快速运行它。尝试区分集成测试和单元测试。

赞(0）回复(0）举报 2024-01-05

tyky79it2#

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

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

字符串
测试可以简单到在except块中调用cls.tearDownClass()。* 然后 * 您可以调用assert(False)或任何您喜欢的方法来提前退出测试。

赞(0）回复(0）举报 2024-01-05

isr3a4wc3#

我有一大堆测试助手函数，它们接受一个测试示例，并使用addfocus来干净地设置/拆除线程、临时文件等，所以我也需要addfocus API来为类级别的fixtures工作。

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()