import time

import unittest

from unittest.main import TestProgram

from app.utils.utils_ydf import LogManager

class CustomTestCase(unittest.TestCase):

    @classmethod

    def setUpClass(cls):

        cls.logger = LogManager('unittest用例的日志').get_logger_without_handlers()

    @classmethod

    def tearDownClass(cls):

        time.sleep(1)

        cls.logger.debug('---------------------------------所有用例完成-----------------------------')

        time.sleep(200000)

    def tearDown(self):

        time.sleep(0.1)

    def setUp(self):

        time.sleep(0.1)

class CustomLoader(unittest.TestLoader):

    def getTestCaseNames(self, testCaseClass):

        """Return a sorted sequence of method names found within testCaseClass

        """

        def isTestMethod(attrname, testCaseClass=testCaseClass,

                         prefix=self.testMethodPrefix):

            return attrname.startswith(prefix) and \

                   callable(getattr(testCaseClass, attrname))

        testFnNames = list(filter(isTestMethod, dir(testCaseClass)))

        if self.sortTestMethodsUsing:

            testFnNames.sort(key=lambda fun_name: getattr(testCaseClass, fun_name).__code__.co_firstlineno)

        return testFnNames

custom_loader = CustomLoader()

class CustomTestTestProgram(TestProgram):

    def __init__(self, *args, testLoader=custom_loader, **kwargs):

        super().__init__(*args, testLoader=testLoader, **kwargs)

def patch_unittest():

    unittest.TestCase = CustomTestCase

    unittest.defaultTestLoader = custom_loader

    unittest.main = CustomTestTestProgram

if __name__ == '__main__':

    patch_unittest()

    class _Test(unittest.TestCase):

        def test_3(self):

            print(1)

        def test_2(self):

            print(2)

        def test_1(self):

            print(3)

    unittest.main()