老李分享：走读unittest源码

老李分享：走读unittest源码

 

poptest是国内唯一一家培养测试开发工程师的培训机构，以学员能胜任自动化测试，性能测试，测试工具开发等工作为目标。poptest测试开发工程师就业培训感兴趣，请大家咨询qq：908821478，最近学员的就业推荐开始，帮助学员梳理学习的知识点，其中涉及到我们在学习中的单元测试框架unittest，在下面和大家分享下里面涉及到的内容，以及源码情况。

一，用法

unittest模块的用法很简单，定义个 TestCase的子类，然后根据需要重载Setup()和TearDown()两个方法。并且在类中定义一些测试用例条目的“方法” ，以test开头命名即可

import laoli

import unittest

class ProductTestCase(unittest.TestCase):

def setUp(self):

print 'units test begin'+'#'*20

def tearDown(self):

print 'unittest end'+'#'*20

def testxx(self):

print 'this is just test'

def testIntegerProduct(self):

for x in xrange(-10,10):

for y in xrange(-10,10):

p = kk.product(x,y)

self.failUnless(p == x*y,'integer multiplication failture')

def testFractalProduct(self):

for x in xrange(-10,10):

for y in xrange(-10,10):

x = x/10.0

y = y/10.0

p = kk.product(x,y)

self.failUnless(p == x*y,'fractal multiplication failture')

if __name__ == '__main__':

unittest.main()

其中laoli为被测模块

注意到代码中的testIntegerProduct(self) 及 testFractalProduct(self) 两个以test开头命名的方法即是测试用例条目，整个类为TestCase的子类，unittest通过这两个信息（红色标注部分）来识别测试用例和用例下的条目。

通过调用unitest.main()方法来启动单元测试。

二.unittest 模块流程概述

首先明确unittest中对测试用例的封装，即TestSuit  是 TestCase类的一个“集合”：

源码中作者对TestSuit的注释：

class TestSuite(object):

"""A test suite is a composite test consisting of a number of TestCases.

For use, create an instance of TestSuite, then add test case instances.

When all tests have been added, the suite can be passed to a test

runner, such as TextTestRunner. It will run the individual test cases

in the order in which they were added, aggregating the results. When

subclassing, do not forget to call the base class constructor.

"""

往TestSuit内增加TestCase的代码片段为：

def addTest(self, test):

# sanity checks

if not hasattr(test, '__call__'):

raise TypeError("the test to add must be callable")

if isinstance(test, type) and issubclass(test, (TestCase, TestSuite)):

raise TypeError("TestCases and TestSuites must be instantiated "

"before passing them to addTest()")

self._tests.append(test)

def addTests(self, tests):

if isinstance(tests, str):

raise TypeError("tests must be an iterable of tests, not a string")

for test in tests:

self.addTest(test)

其中形参test 是TestCase的实例。

明确这一点后，unittest的大体流程其实是比较好理解的：

1.调用unittest.main（）

调用unittest中的main方法其实是创建TestProgram类的实例

main = TestProgram

##############################################################################

# Executing this module from the command line

##############################################################################

if __name__ == "__main__":

main(module=None)

TestProgram类“实施”了unittest的全部流程（创建测试用例，执行测试用例，输出结果）：

2.解析参数和构造TestSuit

unittest支持命令行参数，TestProgram通过调用parseArgs()方法完成参数的解析

并且根据传入参数来对TestSuit进行构建，传递的参数可以包含到具体的测试用例或测试条目 例如支持这样的命令：python Myunittest.py Mytestcase.testsomething 等

所以paresArgs()方法会根据不同的参数调用不同的加载测试用例的方法：有loadTestFromMoudle() 、loadTestFromName()、loadTestFromCase()等，这些方法都属于TestLoader类。

需要提前知道的是，这些Load测试用例的方法返回的都是TestSuit类。

def parseArgs(self, argv):

import getopt

long_opts = ['help','verbose','quiet']

try:

options, args = getopt.getopt(argv[1:], 'hHvq', long_opts)

for opt, value in options:

if opt in ('-h','-H','--help'):

self.usageExit()

if opt in ('-q','--quiet'):

self.verbosity = 0

if opt in ('-v','--verbose'):

self.verbosity = 2

if len(args) == 0 and self.defaultTest is None:

self.test = self.testLoader.loadTestsFromModule(self.module)# 通过调用loadTestFromModule加载测试用例

return

if len(args) > 0:

self.testNames = args

else:

self.testNames = (self.defaultTest,)

self.createTests()#若指定测试用例参数，则执行createTest（），即调用loadTestFromName

except getopt.error as msg:

self.usageExit(msg)

def createTests(self):

self.test = self.testLoader.loadTestsFromNames(self.testNames,

self.module)

3.加载完测试用例即获得TestSuit后，则执行runtest方法开始执行测试

def runTests(self):

if isinstance(self.testRunner, type):

try:

testRunner = self.testRunner(verbosity=self.verbosity)

except TypeError:

# didn't accept the verbosity argument

testRunner = self.testRunner()

else:

# it is assumed to be a TestRunner instance

testRunner = self.testRunner

self.result = testRunner.run(self.test)

if self.exit:

sys.exit(not self.result.wasSuccessful())

随后将分为“加载测试用例” “执行测试用例”记录

概念：

TestCase：testcase实例就是一个测试用例。

什么是测试用例呢？就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码(run)，以及测试后环境的还原(tearDown)。元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。

TestSuite: 多个测试用例集合在一起，就是TestSuite，TestSuite可以嵌套TestSuite。

TestLoader：作用是加载TestCase到TestSuite中的，其中有几个loadTestsFrom__()方法，就是从各个地方寻找TestCase，创建它们的实例，然后add到TestSuite中，再返回一个TestSuite实例。

TextTestRunner：是来执行测试用例的，其中的run(test)会执行TestSuite/TestCase中的run(result)方法。

TextTestResult：测试的结果会保存到TextTestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息。