unittest跳過測試用例進階版-----上一個測試用例失敗後跳過下一個測試用例

實際寫代碼的時候會遇到這樣的問題，以登錄登出爲例，登錄失敗後我們怎麼跳過登出的用例，因爲登錄失敗後測試登出沒有意義結果裏的報錯也沒有意義。這裏主要使用兩個知識點，裝飾器和 TestCase中的_outcome.result （python2中爲_resultForDoCleanups），裝飾器怎麼用具體原理大家自己百度哦，具體示例代碼如下

import unittest
from functools import wraps


def skip_dependon(depend=""):
    """
    :param depend: 依賴的用例函數名，默認爲空
    :return: wraper_func
    """
    def wraper_func(test_func):
        @wraps(test_func)  # @wraps：避免被裝飾函數自身的信息丟失
        def inner_func(self):
            if depend == test_func.__name__:
                raise ValueError("{} cannot depend on itself".format(depend))
            # print("self._outcome", self._outcome.__dict__)
            # 此方法適用於python3.4 +
            # 如果是低版本的python3，請將self._outcome.result修改爲self._outcomeForDoCleanups
            # 如果你是python2版本，請將self._outcome.result修改爲self._resultForDoCleanups
            failures = str([fail[0] for fail in self._outcome.result.failures])
            print("111111111111111111111", failures)
            errors = str([error[0] for error in self._outcome.result.errors])
            print("2222222222222222222222", errors)
            skipped = str([error[0] for error in self._outcome.result.skipped])
            print("333333333333333333333", skipped)
            flag = (depend in failures) or (depend in errors) or (depend in skipped)
            if failures.find(depend) != -1:
                # 輸出結果 [<__main__.TestDemo testMethod=test_login>]
                # 如果依賴的用例名在failures中，則判定爲失敗，以下兩種情況同理
                # find()方法：查找子字符串，若找到返回從0開始的下標值，若找不到返回 - 1
                test = unittest.skipIf(flag, "{} failed".format(depend))(test_func)
            elif errors.find(depend) != -1:
                test = unittest.skipIf(flag, "{} error".format(depend))(test_func)
            elif skipped.find(depend) != -1:
                test = unittest.skipIf(flag, "{} skipped".format(depend))(test_func)
            else:
                test = test_func
            return test(self)
        return inner_func
    return wraper_func


class TestDemo(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        print("setupclass")

    def setUp(self):
        print("setup")

    def test_login(self):
        print("test_login")
        self.assertEqual(1, 2)  # 這裏讓登錄判斷爲失敗

    @skip_dependon(depend="test_login")
    def test_logout(self):
        print("test_logout")
        self.assertEqual(1, 1)

    @skip_dependon(depend="test_logout")
    def test_1(self):
        print("test1")

    @skip_dependon(depend="test_1")
    def test_2(self):
        print("test2")

    def tearDown(self):
        print("teardown")

    @classmethod
    def tearDownClass(cls):
        print("teardownclass")

if __name__ == '__main__':
    testsuite = unittest.TestSuite()
    testsuite.addTest(TestDemo("test_login"))
    testsuite.addTest(TestDemo("test_logout"))
    testsuite.addTest(TestDemo("test_1"))
    testsuite.addTest(TestDemo("test_2"))
    runner = unittest.TextTestRunner()
    runner.run(testsuite)

結果如下：