一文搞懂Python Unittest测试方法执行顺序

淡淡的烟草味﹌ 2023-10-08 13:52 18阅读 0赞

![在这里插入图片描述][5fb87c883155482198a70b4dca710d26.gif_pic_center]

### Unittest ###

unittest大家应该都不陌生。它作为一款博主在5-6年前最常用的单元测试框架，现在正被pytest，nose慢慢蚕食。

渐渐地，看到大家更多的讨论的内容从unittest+HTMLTestRunner变为pytest+allure2等后起之秀。

不禁感慨，终究是自己落伍了，跟不上时代的大潮了。

### 回到主题 ###

感慨完了，回到正文。虽然unittest正在慢慢被放弃，但是它仍然是一款很全面的测试框架。

今天在群里看到有个群友的一番言论，激起了我的一番回忆。

自己以前是知道unittest的执行顺序并不是按照编写test方法的顺序执行，而是按照字典序执行的。但遗憾的是我都是投机取巧去解决的问题（后面会讲）。

下面我们就来探讨下unittest类的test方法的执行顺序问题。

### 源码初窥 ###

研究一下源码(unittest.TestLoader)可以发现，在加载一个class下面的test方法的时候，原生Loader进行了排序，并且根据functools.cmp\_to\_key方法对测试方法列表进行了排序。

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16]  
我们知道，unittest是不需要我们指定对应的方法，说白了，它是从类里面自动获取到咱们的方法，并约定了以test开头的方法都会被视为测试方法。

![可以看到testMethodPrefix，即测试方法前缀，如果不是test开头则直接return False][testMethodPrefix_test_return False]

查询一下self.sortTestMethodsUsing（这个是一个排序的方式）。  
![在这里插入图片描述][7511df89a9df4ee3acfa4ee9b8f207f1.png]

可以看到这个比较方法写的很明确了，如果x < y那么返回-1，x = y则返回0，x > y返回1。

其实大家可能不知道Python里面的字符串也是可以比较的，在此必须说明一下字典序。我们来看看这个例子:

a = "abc"
    b = "abcd"
    c = "abce"
    print(a > b)
    print(b > c)

猜猜看执行结果，很显然，字典序的比较，是按A-Z的顺序来比较的，如果前缀一样但长度不一样，那么长度长的那个，字典序靠后。  
![在这里插入图片描述][6a7c2627d1a0480899e308a6b9e57bb2.png]

了解了字典序以后，我们就不难知道，在unittest里面它寻找case的过程可以这样简化:

*  找到对应类下面以test开头的测试方法
 *  对他们进行字典序排序
 *  依次执行
    
    这样就不难解释**为什么我们有时候写的case不按照自己想的顺序来。**

### 回到问题的本质 ###

搞清楚为什么用例会乱，那就想到对应的解决方案。由于修改源码是不太合适的，那我们有2个策略去达成目的。

比如我有多个test方法:

class Testcase(unittest.TestCase):
    
        def setUp(self) -> None:
            pass
    
        def test_1(self):
            print("执行第一个")
    
        def test_2(self):
            print("第二个")
    
        def test_3(self):
            print("第三个")
    
        def test_10(self):
            print("第四个")
    
        def test_11(self):
            print("第五个")
    
        def tearDown(self) -> None:
            pass
            
    if __name__ == "__main__":
        unittest.main()

执行起来，按照字典序，其实是1 10 11 2 3的顺序。

![可以看到现在还是不对的][df252dd44d1a4387a345f8cbfe51ec02.png]

### 1. 以字典序的方式编写test方法 ###

我们可以手动修改test方法的名称，这也是我早前的处理方式。也就是说把想要先执行的case字典序排到前面:

class Testcase(unittest.TestCase):
    
        def setUp(self) -> None:
            pass
    
        def test_0_1(self):
            print("执行第一个")
    
        def test_0_2(self):
            print("第二个")
    
        def test_0_3(self):
            print("第三个")
    
        def test_1_0(self):
            print("第四个")
    
        def test_1_1(self):
            print("第五个")
    
        def tearDown(self) -> None:
            pass

我们可以把数字按位数拆开，个位数就把10位补0，这样就能达到效果，如果会写100个case，我们就需要补2个0，比如0\_0\_1，当然一个文件里面也不会有太多case。

如果遇到test\_login这种怎么办呢，不是数字结尾的方法。

其实是一样的，可以写成test\_数字\_业务的模式。番货写了一个装饰器专门解决这样的问题，大家可以去参考下。

### 2. 回归本质，从根本解决问题 ###

方案1用了番货的装饰器，好是好，但是改变了方法本身的名称，我们其实可以针对他的排序方式入手，按照我们编写case的顺序排序测试方法，就能达到想要的目的。

说说思路:

1.  手写一个loader继承自TestLoader类，改写里面的排序方法
2.  在unittest运行的时候传入这个新的loader

来看看完整代码，注释里面写的很完善了。

import unittest
    
    
    class MyTestLoader(unittest.TestLoader):
        def getTestCaseNames(self, testcase_class):
            # 调用父类的获取“测试方法”函数
            test_names = super().getTestCaseNames(testcase_class)
            # 拿到测试方法list
            testcase_methods = list(testcase_class.__dict__.keys())
            # 根据list的索引对testcase_methods进行排序
            test_names.sort(key=testcase_methods.index)
            # 返回测试方法名称
            return test_names
    
    
    class Testcase(unittest.TestCase):
    
        def setUp(self) -> None:
            pass
    
        def test_1(self):
            print("执行第一个")
    
        def test_2(self):
            print("第二个")
    
        def test_3(self):
            print("第三个")
    
        def test_10(self):
            print("第四个")
    
        def test_11(self):
            print("第五个")
    
        def tearDown(self) -> None:
            pass
    
    
    if __name__ == "__main__":
        unittest.main(testLoader=MyTestLoader())

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 1]  
执行了一下还是不对，是不是哪里出了什么问题呢？

是因为pycharm有一种默认的unittest的调试方法，我们要改成普通的方法去执行。  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 2]![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_15_color_FFFFFF_t_70_g_se_x_16]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 3]  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 4]  
试试用控制台执行:

![也没什么毛病][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_16_color_FFFFFF_t_70_g_se_x_16]

今天的内容就讲到这里了，看懂的记得给个赞哦~

![在这里插入图片描述][ddf4282d4b9d4dc8ba703bc646a4f72d.png_pic_center]

**最后：** 可以关注公众号：伤心的辣条 ！ 进去有许多资料共享！资料都是面试时面试官必问的知识点，也包括了很多测试行业常见知识，其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

**如果我的博客对你有帮助、如果你喜欢我的博客内容，请 “点赞” “评论” “收藏” 一键三连哦！推荐软件测试交流学习群：914172719 里面会分享一些资深架构师录制的视频录像**

--------------------

### 好文推荐 ###

[转行面试，跳槽面试，软件测试人员都必须知道的这几种面试技巧！][Link 1]

[面试经：一线城市搬砖！又面软件测试岗，5000就知足了…][5000]

[面试官：工作三年，还来面初级测试？恐怕你的软件测试工程师的头衔要加双引号…][Link 2]

[什么样的人适合从事软件测试工作？][Link 3]

[那个准点下班的人，比我先升职了…][Link 4]

[测试岗反复跳槽，跳着跳着就跳没了…][Link 5]

[5fb87c883155482198a70b4dca710d26.gif_pic_center]: https://img-blog.csdnimg.cn/5fb87c883155482198a70b4dca710d26.gif#pic_center
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/bc55c35f8da84deabc4656ec3553504e.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_20,color_FFFFFF,t_70,g_se,x_16
[testMethodPrefix_test_return False]: https://img-blog.csdnimg.cn/96557b7e946a4eb393063532aa929f21.png
[7511df89a9df4ee3acfa4ee9b8f207f1.png]: https://img-blog.csdnimg.cn/7511df89a9df4ee3acfa4ee9b8f207f1.png
[6a7c2627d1a0480899e308a6b9e57bb2.png]: https://img-blog.csdnimg.cn/6a7c2627d1a0480899e308a6b9e57bb2.png
[df252dd44d1a4387a345f8cbfe51ec02.png]: https://img-blog.csdnimg.cn/df252dd44d1a4387a345f8cbfe51ec02.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 1]: https://img-blog.csdnimg.cn/93002bcb94e0410eaa3d4cb0c6e164ac.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 2]: https://img-blog.csdnimg.cn/c24bbcea72cb420ba72c93c37431fc28.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_15_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/eda7f1cf98794c38ab6a40f8c3168349.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_15,color_FFFFFF,t_70,g_se,x_16
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 3]: https://img-blog.csdnimg.cn/58c1de43d1b44feeb841b6769e9766fb.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_20_color_FFFFFF_t_70_g_se_x_16 4]: https://img-blog.csdnimg.cn/ce9314be993b4bc1a9a3e57fbd78adf9.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_20,color_FFFFFF,t_70,g_se,x_16
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb_size_16_color_FFFFFF_t_70_g_se_x_16]: https://img-blog.csdnimg.cn/907c35a99a9a4e4f9cdef82a736c2a0c.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA6L2v5Lu25rWL6K-V5ZCb,size_16,color_FFFFFF,t_70,g_se,x_16
[ddf4282d4b9d4dc8ba703bc646a4f72d.png_pic_center]: https://img-blog.csdnimg.cn/ddf4282d4b9d4dc8ba703bc646a4f72d.png#pic_center
[Link 1]: https://blog.csdn.net/m0_56736369/article/details/117909132
[5000]: https://blog.csdn.net/m0_56736369/article/details/117915399
[Link 2]: https://blog.csdn.net/weixin_50829653/article/details/117422433
[Link 3]: https://blog.csdn.net/weixin_50271247/article/details/108421434
[Link 4]: https://blog.csdn.net/weixin_50829653/article/details/117856900
[Link 5]: https://blog.csdn.net/m0_56736369/article/details/117923266