扩展 Python 的 unittest 框架

目录

前言：

如何控制 unittest 用例执行的顺序呢？

方式 1，通过 TestSuite 类的 addTest 方法，按顺序加载测试用例

方式 2，通过修改函数名的方式

那就造个轮子吧

---

前言：

Python 的 unittest 模块提供了一个标准的单元测试框架，但它有一些限制，例如无法处理多线程测试和模拟网络请求等。在这篇文章中，我们将介绍如何扩展 Python 的 unittest 框架，以便能够进行多线程测试和模拟网络请求等操作。

unittest 是 Python 标准库自带的单元测试框架，是 Python 版本的 JUnit，关于 unittest 框架的使用，官方文档非常详细，网上也有不少好的教程，这里就不多说了。

本文主要分享在使用 unittest 的过程中，做的一些扩展尝试。先上一个例子。

import unittest

class TestLegion(unittest.TestCase):
    def test_create_legion(self):
        """创建军团

        :return:
        """

    def test_bless(self):
        """ 公会祈福

        :return:
        """

    def test_receive_bless_box(self):
        """ 领取祈福宝箱

        :return:
        """

    def test_quit_legion(self):
        """退出军团

        :return:
        """

这是一个标准的使用 unittest 进行测试的例子，写完后心里美滋滋，嗯，就按照这个顺序测就可以了。结果一运行。

执行的顺序乱了！！第一个执行的测试用例并不是创建军团，而是公会祈福，此时玩家还没创建军团，进行公会祈福的话会直接报错，导致用例失败。

到这里有些同学会想说，为什么要让测试用例之间有所依赖呢？

的确，如果完全没依赖，测试用例的执行顺序是不需要关注的。但是这样对于用例的设计和实现，要求就高了许多。而对游戏来说，一个系统内的操作，是有很大的关联性的。以军团为例，军团内的每个操作都有一个前提，你需要加入一个军团。所以要实现用例之间的完全解耦，需要每个用例开始之前，检测玩家的军团状态。

如果可以控制测试用例的执行顺序，按照功能玩法流程一遍走下来，节省的代码量是非常可观的，阅读测试用例也会清晰许多。

如何控制 unittest 用例执行的顺序呢？

我们先看看，unittest 是怎么样对用例进行排序的。在loader.py的loadTestsFromTestCase方法里边，调用了getTestCaseNames方法来获取测试用例的名称

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=functools.cmp_to_key(self.sortTestMethodsUsing))
    return testFnNames