unittest流程

本文详细探讨了Python的unittest框架,从main.py入口开始,解析TestProgram类如何处理命令行参数,进行测试发现。通过_xdo_discovery()函数,unittest遍历指定目录下的模块,寻找以test开头的py文件。TestLoader类负责加载测试用例,它通过loadTestsFromModule()等方法实例化TestCase子类,确保每个测试方法以test开头并可调用。TestCase类是测试用例的核心,定义了执行顺序及异常处理。最后,TestSuite类将测试用例组织成测试套件,TextTestRunner类以文本形式展示测试结果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近研究了下unittest的代码,画了一个xmind图,如下所示

unittest类图
入口,main.py中的main=TestProgram
Class TestProgram(object) 为一个类,实现了总体流程
初始化默认参数和解析命令行参数
def parseArgs(argv)
若参数中有discover,则需要调用内部函数 def _do_discovery(argv)去搜索指定的py下的测试用例
默认搜索开始目录为当前,匹配模式为test开头的py文件,顶级目录为空
层层递归,导入,加载测试用例,加载方法使用下面介绍的loadTestsFromModule(module)
运行测试用例
def createTests()创建测试用例
def runTests()运行测试用例
创建测试用例 Class TestLoader(object)
def loadTestsFromModule(module) 从模块中创建
若果py文件中的类是case.TestCase类型的子类,则调用def loadTestsFromTestCase(obj)加载该类下面的具体以test开头的方法
def loadTestsFromTestCase(obj) 从具体的测试函数中加载测试用例
def getTestCaseNames(testCaseClass)获取具体需要测试的函数,要求是以test开头,并且可以调用的函数,然后返回一个列表
使用map(testCaseClass, testCaseNames)去实例化每一个测试函数为一个测试用例
这里testCaseClass虽然是客户自定义的测试类的名称,但都是继承自类 class TestCase(object),该类传入测试方法名称进行实例化
Class TestCase(object) 单个测试用例的具体实现(核心类)
def run(result)传入结果处理类class TestResult(object) 的实例化,该函数定义了setUp, 测试方法, tearDown的执行顺序,以及执行异常时的操作
接着使用suite.py下的类 class TestSuite(BaseTestSuite) 将前面实例化的所有测试用例实例化为一个测试套件
可能一个模块中存在多个测试类,每一个类都对应着一个测试套件,这里对loadTestsFromTestCase返回回来的结果再使用类class TestSuite(BaseTestSuite) 进行实例化,最后返回测试套件列表
def loadTestsFromNames(testNames, module) 根据名字创建
def loadTestsFromName(name, module) 根据名字创建单个套件
首先将该名字以模块方式导入
导入的是模块类型,则调用def loadTestsFromModule(obj)
导入的是测试用例类型,调用def loadTestsFromTestCase(obj)
导入的是未绑定的方法类型并且模块上一级是测试用例类型,则调用suiteClass套件类型
导入的是TestSuite类型,则直接返回
若是可调用,则实例化后再如上判断,否则报错
运行测试用例 def runTests()
Class TextTestRunner(object) 测试运行类,以文本形式显示测试结果
实例化TextTestRunner后执行run方法,这样会层层执行,分别从套件TestSuite到TestCase中的run方法

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值