使用TestCase中的addCleanup()方法清理测试结束后的资源

addCleanup() 方法是 Python unittest 模块中 TestCase 类的一个方法，用于在测试结束后清理资源。它会在每个测试方法运行结束后自动调用。这个方法可以用来执行一些清理工作，比如关闭数据库连接、删除临时文件等。

addCleanup() 方法的语法如下：

addCleanup(function, *args, **kwargs)

其中，“function” 是需要执行的函数名，“*args” 和 “**kwargs” 是传递给该函数的参数。

下面是一个示例，演示如何使用 addCleanup() 方法进行清理工作。

import unittest

class MyTestCase(unittest.TestCase):
    def setUp(self):
        # 创建测试用例的准备工作
        self.file_path = 'temp.txt'
        with open(self.file_path, 'w') as f:
            f.write('Hello, World!')

    def tearDown(self):
        # 清理测试用例的资源
        if self.file_path:
            os.remove(self.file_path)

    def test_file_exist(self):
        # 检查文件是否存在
        self.assertTrue(os.path.isfile(self.file_path))

    def test_file_content(self):
        # 检查文件内容是否正确
        with open(self.file_path, 'r') as f:
            content = f.read()
        self.assertEqual(content, 'Hello, World!')

if __name__ == "__main__":
    unittest.main()

在这个例子中，我们创建了一个测试用例类 MyTestCase，其中包含了两个测试方法 test_file_exist() 和 test_file_content()。

在 setUp() 方法中，我们创建了一个临时文件 'temp.txt' 并写入内容 'Hello, World!'。我们将文件路径保存在实例变量 self.file_path 中，在 tearDown() 方法中检查该变量是否存在，并删除文件。

当运行这个测试用例时，每个测试方法在运行完毕后，都会自动调用 tearDown() 方法进行资源清理。这样，我们就可以确保每个测试方法都能够在结束后进行必要的清理工作，以保持测试环境的干净和稳定。

总结：addCleanup() 方法是 unittest 模块中 TestCase 类的一个方法，用于在每个测试方法运行结束后执行清理工作。它接受一个函数名和参数，可以用来关闭数据库连接、删除临时文件等。通过 addCleanup() 方法，我们可以确保每个测试方法都能在运行结束后进行资源清理，保持测试环境的干净和稳定。