测试只会python_Python unittest,只有在测试失败时才能执行某些操作

我最近尝试做类似的事情,我找到了出路：

import unittest

class MyTestResult(unittest.TestResult):

def addFailure(self, test, err):

# here you can do what you want to do when a test case fails

print('test failed!')

super(MyTestResult, self).addFailure(test, err)

def addError(self, test, err):

# here you can do what you want to do when a test case raises an error

super(MyTestResult, self).addError(test, err)

class MyUT(unittest.TestCase):

def test_fail(self):

self.assertEqual(1, 2, '123')

self.assertTrue("ABc".isupper())

if __name__ == '__main__':

unittest.main(testRunner=unittest.TextTestRunner(resultclass=MyTestResult))

如果你想根据不同的测试用例类做不同的工作,你可以这样做：

import unittest

class MyUT(unittest.TestCase):

class TestResult(unittest.TestResult):

def addFailure(self, test, err):

print('do something when test case failed')

super(MyUT.TestResult, self).addFailure(test, err)

def addError(self, test, err):

print('test case error')

super(MyUT.TestResult, self).addError(test, err)

def test_fail(self):

self.assertEqual(1, 2, "1=2")

class MyUT2(unittest.TestCase):

class TestResult(unittest.TestResult):

def addFailure(self, test, err):

print('do something else when test case failed')

super(MyUT2.TestResult, self).addFailure(test, err)

def addError(self, test, err):

print('test case error')

super(MyUT2.TestResult, self).addError(test, err)

def test_fail(self):

self.assertEqual(1, 2, "1=2")

if __name__ == '__main__':

classes = [MyUT, MyUT2]

for c in classes:

suite = unittest.TestLoader().loadTestsFromTestCase(c)

unittest.TextTestRunner(resultclass=c.TestResult).run(suite)