unittest单元测试框架,主要由四部分组成:测试固件、测试用例、测试套件、测试执行器
测试固件(test fixture)
测试固件有两部分,执行测试前的准备部分setUp(),测试执行完后的清扫部分tearDown()
测试用例(test case)
测试用例中,包含测试固件,具体的测试代码的函数。测试固件可以不写,但是至少要有一个以test开头的函数。
unittest会自动识别test开头的函数是测试代码,一定要用小写的test开头!,下面看一个实例
在PyCharm中新建一个项目,在下面建一个包叫做lesson_unittest,然后在这个包下面新建一个.py文件叫做lesson_test_fixture1,然后写下面的代码
import unittest
from io import StringIO
class lesson(unittest.TestCase): def setUp(self):
self.f = StringIO()
print("创建StringIO") def tearDown(self):
s = self.f.getvalue()
print("读取StringIO:",s) def test_case1(self):
self.f.write("test case 1")
print("写入test case 1") def test_case2(self):
self.f.write("test case 2")
print("写入test case 2") if __name__ =='__main__':
unittest.main()
所有方法中一定要带self这个变量
unittest.main()在PyCharm可以不要,如果要在cmd中运行,一定要加上这行
setUp()执行成功之后,不管test中有没有异常发生,tearDown()都会执行
可以看出上面这个实例在执行每个test之前和之后都会去调用setUp()和tearDown()这个方法
很多时候我们只想执行一次测试固件:比如setUp()中放着连接数据库操作,多个test case放着不同数据库操作,然后tearDown()断开数据连接
使用setUpClass()和tearDownClass()可以的得到这样的效果,必须被装饰成classmethod
使用装饰器@unittest.skip可以跳过某个测试用例不执行
在刚才lesson_unittest包下,再新建一个.py文件叫做lesson_test_fixture2,然后写下面的代码
import unittest
from io import StringIO
class lesson(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.f = StringIO()
print("创建StringIO") @classmethod
def tearDownClass(cls):
s = cls.f.getvalue()
print("读取StringIO:",s) def test_case1(self):
self.f.write("test case 1")
print("写入test case 1") def test_case2(self):
self.f.write("test case 2")
print("写入test case 2") if __name__ =='__main__':
unittest.main()
测试执行器(test runner)
test runner用来执行加载的测试用例,可以是单个测试用例,也可以是测试套件
一种就是上方实例中的直接使用unittest.main()去执行,会搜索所有以test开头的测试用例,按照ASCII的顺序执行多个用例
另一种就是使用TextTestRunner()方法初始化一个测试执行器,然后使用run()去执行测试套件
runner = unittest.TextTestRunner() runner.run(suite)
这里suite是指测试套件的实例
测试套件(test suite)
很多测试用例的集合就是测试套件,通过测试套件来管理多个测试用例,会根据用例的加载顺序执行用例
使用TestSuite()初始化一个Suite实例
使用addTest()方法加载测试方法到测试套件中,addTest()如果要在传入别的包下的测试方法: 包名.类名(测试方法函数)
suite = unittest.TestSuite()
suite.addTest(lesson('test_case1'))#lesson是类名,test_case1是测试函数
suite.addTest(lesson('test_case2'))#如果是lesson是其他包下的,就是包名.lesson(‘test_case2’)
runner = unittest.TextTestRunner()
runner.run(suite)
将文章第一个示例中最后一行unittest.main()换成上方代码执行
使用makeSuite()可以把一个类下面所有测试方法都加载到测试套件中
目前项目路径应该是这样的,最上面是项目,然后包,最后是.py文件。一定要有__init__.py文件,不然引用不到
在项目下新建一个.py文件,名字随便不要带test,或lesson就行。在项目下新建,不要在包里,不要在包里,不要在包里
import unittest
from lesson_unittest.lesson_test_fixture1 import lesson
suite = unittest.TestSuite(unittest.makeSuite(lesson))
if __name__=='__main__':
runner = unittest.TextTestRunner()
runner.run(suite)
可以看到把lesson两个测试用例都执行了,但是使用makeSuite()还是有些繁琐。要把写的所有的类都加到测试套件中
最常用的是使用discover()方法可以把指定路径下所有测试方法都加载到测试套件中
使用刚才实验makeSuite()方法的那个.py文件,清空代码输入以下代码
import unittest
test_dir = 'D:/PycharmProjects/PythonLesson/lesson_unittest'
discover = unittest.defaultTestLoader.discover(test_dir, pattern='lesson*.py')
runner = unittest.TextTestRunner()
runner.run(discover)
test_dir是指定的路径,pattern是指定的文件名。pattern参数是定义文件名匹配规则的,可以随便修改
比如这里lesson*.py是指文件名以lesson开头的文件
也可以写成*test*文件带有test的文件
如果是*lesson.py,那就是指文件名以lesson.py结尾的文件
*起到跟通配符差不多的作用