码迷,mamicode.com
首页 > 编程语言 > 详细

第二种方式,修改python unittest的执行顺序,使用猴子补丁

时间:2018-07-25 20:09:04      阅读:204      评论:0      收藏:0      [点我收藏+]

标签:alt   文件中   应该   round   fir   self   unit   cas   猴子补丁   

1、按照测试用例的上下顺序,而不是按方法的名称的字母顺序来执行测试用例。

之前的文章链接 

之前写的,不是猴子补丁,而是要把Test用例的类名传到run里面去执行,与原生的使用有一点区别。现在修改成,用的时候代码与原生unittest完全一模一样,但运行时候使用与原生不同的逻辑。

 

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()

 

 

 

 这次的unitetest测试用例写法还是继承原生的Testcase类,执行生成测试用例也是原生的方法。

可以看到,正常情况下应该打印3 2 1 ,但使用了猴子技术后打印是 1 2  3。

 

运行结果是:

技术分享图片

 

 

 

 

2、可以把调用patch_unittest函数写在最常用的工具包里面,就不用亲自去调用他了。

之前上篇验证过,python的模块只会导入一次,不管你在不同的文件中不同地方import 几百次,都是只会导入一次的。别的地方import 的unitetest的行为都会发生变化。

第二种方式,修改python unittest的执行顺序,使用猴子补丁

标签:alt   文件中   应该   round   fir   self   unit   cas   猴子补丁   

原文地址:https://www.cnblogs.com/ydf0509/p/9367643.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!