访问单元拆卸中的测试状态

2022-05-17 00:00:00 python unit-testing python-unittest

问题描述

我需要在测试套件中的每一个pyunit测试通过或失败后调用Web API,因此我基本上需要访问tearDown方法中的测试状态。

但我找不到(或我完全错过了)任何有关谁访问此数据的文档。

有什么想法吗?


解决方案

使用TestResult。

import unittest

class TestFoo(unittest.TestCase):
    def test_ok(self):
        self.assertEqual(1+2, 3)
    def test_fail(self):
        self.assertEqual(1+2, 4)
    def test_error(self):
        1/0
    @unittest.skip('blah')
    def test_skip(self):
        self.assertEqual(42, 42)

class MyResult(unittest.TextTestResult):
    def addError(self, test, err):
        self.call_web_api(test, err)
        return super(MyResult, self).addError(test, err)
    def addFailure(self, test, err):
        self.call_web_api(test, err)
        return super(MyResult, self).addFailure(test, err)
    def call_web_api(self, test, err):
        print('***', test, err)

if __name__ == '__main__':
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(TestFoo))

    runner = unittest.TextTestRunner(resultclass=MyResult)
    runner.run(suite)

您也可以按如下方式使用:

...
    runner = unittest.TextTestRunner()
    result = runner.run(suite)
    print(result.errors)
    print(result.failures)
    print(result.skipped)

相关文章