使用unittestTestResult()来扩展测试框架功能

unittest是Python的一个标准库，用于编写和运行单元测试。unittest提供了一系列的类和方法，方便我们编写和执行测试用例。其中，unittest.TestResult类是unittest框架的一个核心组件，用于存储测试结果并提供一些扩展的功能。

要扩展unittest的功能，我们可以通过继承unittest.TestResult类来自定义自己的测试结果类。下面将介绍如何使用unittest的TestResult类来扩展测试框架，并给出一个具体的例子。

首先，我们可以创建一个新的类，如MyTestResult，来继承unittest.TestResult类。在该类中，我们可以重写一些方法来扩展unittest的功能。以下是一个例子：

import unittest

class MyTestResult(unittest.TestResult):
    def __init__(self):
        super().__init__()  # 调用父类的构造函数
    
    def addSuccess(self, test):
        super().addSuccess(test)  # 调用父类的addSuccess方法
        # 在测试通过时，打印一条自定义的成功信息
        print(f"Test {test} passed!")

    def addFailure(self, test, err):
        super().addFailure(test, err)  # 调用父类的addFailure方法
        # 在测试失败时，打印一条自定义的失败信息
        print(f"Test {test} failed: {err}")

# 定义一个简单的测试类
class MyTestCase(unittest.TestCase):
    def test_add(self):
        self.assertEqual(1+1, 2)

if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(MyTestCase('test_add'))
    
    runner = unittest.TextTestRunner(resultclass=MyTestResult)
    runner.run(suite)

在上面的例子中，我们定义了一个新的测试结果类MyTestResult，该类继承自unittest.TestResult类。在MyTestResult中，我们重写了addSuccess方法和addFailure方法。addSuccess方法在测试通过时调用，addFailure方法在测试失败时调用。我们在这两个方法中添加了自定义的打印信息。

然后，我们定义了一个简单的测试类MyTestCase，其中包含了一个测试方法test_add。这个方法测试了1+1是否等于2。

在主程序中，我们首先创建一个测试套件suite，并添加了MyTestCase类中的test_add方法。然后，创建了一个TextTestRunner对象runner，并将resultclass参数设置为我们自定义的测试结果类MyTestResult。最后，调用runner的run方法运行测试。

当我们运行上述代码时，输出如下：

Test <MyTestCase testMethod=test_add> passed!

可以看到，当测试通过时，addSuccess方法中的自定义信息被打印出来。

这只是一个简单的例子，仅仅展示了如何扩展unittest框架的功能。我们可以根据自己的需求，在自定义的测试结果类中添加更多的功能，比如记录测试结果到日志文件、发送测试报告等。

总结起来，使用unittest的TestResult类来扩展测试框架的功能，可以通过继承该类并重写相应的方法来实现。通过添加自定义的逻辑，我们可以在测试通过或失败时执行一些额外的操作，从而满足特定需求。