Python unittest单元测试框架的使用
一、测试模型
下面这部分来自于某书籍资料，拿过来，按需参考一下：
测试模型
（1）线性测试
1、概念：
通过录制或编写对应应用程序的操作步骤产生的线性脚本。单纯的来模拟用户完整的操作场景。（操作，重复操作，数据）都混合在一起。
2、优点：每个脚本相对独立，且不产生其他依赖和调用。任何一个测试用例脚本拿出来都可以单独执行。
3、缺点：开发成本高，用例之间存在重复的操作。比如重复的用户登录和退出。
维护成本高，由于重复的操作，当重复的操作发生改变时，则需要逐一进行脚本的修改。
4.线性测试实例：用户登录
（2）模块化驱动测试
1、概念：
将重复的操作独立成功共模块，当用例执行过程中需要用到这一模块操作时则被调用。
操作+（重复操作，数据）混合在一起。例如，自动化测试的执行需要保持测试用例的独立性和完整性，所以每一条用例在执行时都需要登录和退出操作，so可以把登录和退出的操作封装为公共函数。
2、优点：由于最大限度消除了重复，从而提高了开发效率和提高测试用例的可维护性。
3、缺点：虽然模块化的步骤相同，但是测试数据不同。比如说重复的登录模块，如果登录用户不同，依旧要重复编写登录脚本。
4.实例：对公共模块，例如登陆和退出进行模块化封装
（3）数据驱动测试
1、概念：它将测试中的测试数据和操作分离，数据存放在另外一个文件中单独维护。
通过数据的改变从而驱动自动化测试的执行，最终引起测试结果的改变。
操作+重复操作+数据分开。
2、优点：
通过这种方式，将数据和重复操作分开，可以快速增加相似测试，完成不同数据情况下的测试。
3、实例从excel表格读取用户名密码，登录邮箱。
二、unittest框架
用Python搭建自动化测试框架，需要组织用例以及测试执行，大部分推荐的是unittest。现在用的也是这个，随着了解，也有其他的框架，有时间再多去学习，保持持续学习哦~
附上官方文档地址：链接描述https://docs.python.org/2.7/library/unittest.html
unittest是Python自带的单元测试框，可以用来作自动化测试框架的用例组织执行框架。优点：提供用例组织与执行方法；提供比较方法；提供丰富的日志、清晰的报告。
大致流程：
- 写好TestCase
- 由TestLoader加载TestCase到TestSuite
- 然后由TextTestRunner来运行TestSuite，运行的结果保存在TextTestResult中。
通过命令行或者unittest.main()执行时，main会调用TextTestRunner中的run()来执行，或者可以直接通过TextTestRunner来执行用例。
在Runner执行时，默认将执行结果输出到控制台，我们可以设置其输出到文件，在文件中查看结果。
unittest中最核心的部分是：TestFixture、TestCase、TestSuite、TestRunner。
1、Test fixture
用于一个测试环境的准备和销毁还原。
当测试用例每次执行之前需要准备测试环境，每次测试完成后还原测试环境，比如执行前连接数据库、打开浏览器等，执行完成后需要还原数据库、关闭浏览器等操作。这时候就可以启用testfixture。
- setUp()：准备环境，执行每个测试用例的前置条件；
- tearDown()：环境还原，执行每个测试用例的后置条件；
- setUpClass()：必须使用@classmethod装饰器，所有case执行的前置条件，只运行一次；
- tearDownClass()：必须使用@classmethod装饰器，所有case运行完后只运行一次；
例如：
# 重写TestCase的setUp() tearDown()方法：在每个测试方法执行前以及执行后各执行一次
def setUp(self): # 钩子方法
print("do something before test : prepare environment")
def tearDown(self): # 钩子方法
print("do something after test : clean up ")
2、TestCase
类，unittest.TestCase
一个类class继承 unittest.TestCase，就是一个测试用例。一个TestCase的实例就是一个测试用例，就是一个完整的测试流程。
包括测试前环境准备setUp()|setUpClass()、执行代码run()、测试环境后的还原tearDown()|tearDownClass()。
继承自unittest.TestCase的类中，测试方法的名称要以test开头。且只会执行以test开头定义的方法（测试用例）。
例如：【先准备待测试的方法function.py】
#!/usr/bin/python3 # -*- coding:utf-8 -*- def add(a,b): return a+b def minus(a,b): return a-b def multi(a,b): return a*b def divide(a,b): return a/b
【测试脚本】：
import unittest
from A_UnitTest_basicDemo_ok.function import *
class TestFunc(unittest.TestCase):
# 继承自unittest.TestCase
# 重写TestCase的setUp()、tearDown()方法：在每个测试方法执行前以及执行后各执行一次
def setUp(self):
print("do something before test : prepare environment")
def tearDown(self):
print("do something after test : clean up ")
# 测试方法均已test开头，否则是不被unittest识别的
def test_add(self):
print("add:")
self.assertEqual(3,add(1,2))
def test_minus(self):
print("minus")
self.assertEqual(3,minus(5,2))
def test_multi(self):
print("multi")
self.assertEqual(6,multi(2 ,3))
def test_divide(self):
print("divide")
self.assertEqual(2,divide(4,2))
if __name__ == "__main__":
# 在main()中加verbosity参数，可以控制输出的错误报告的详细程度
# verbosity=*：默认是1；设为0，则不输出每一个用例的执行结果；2-输出详细的执行结果
unittest.main(verbosity=2)
或者也可以使用setUpClass() & tearDownClass()方法：
# 如果想在所有case执行之前准备一次测试环境，并在所有case执行结束后再清理环境
@classmethod
def setUpClass(cls):
print("this setupclass() method only called once")
@classmethod
def tearDownClass(cls):
print("this teardownclass() method only called once too")
【verbosity】
在测试用例文件的末尾加上如下代码：
if __name__ == "__main__": unittest.main(verbosity=2) # 输出详细的错误报告
在unittest.main()中加参数verbosity可以控制错误报告的详细程度：默认为1。0，表示不输出每一个用例的执行结果；2表示详细的执行报告结果。
【执行结果】：输出到控制台
this setupclass() method only called once
test_add (__main__.TestFunc) ... ok
add:
test_divide (__main__.TestFunc) ... ok
divide
test_minus (__main__.TestFunc) ... ok
minus
test_multi (__main__.TestFunc) ... ok
multithis teardownclass() method only called once too
----------------------------------------------------------------------
Ran 4 tests in 0.000sOK
3、TestSuite
上述简单的测试会产生两个问题，可不可以控制test测试用例的执行顺序？若不想执行某个测试用例，有没有办法可以跳过？
对于执行顺序，默认按照test的 A-Z、a-z的方法执行。若要按自己编写的用例的先后关系执行，需要用到testSuite。
把多个测试用例集合起来，一起执行，就是testSuite。testsuite还可以包含testsuite。
一般通过addTest()或者addTests()向suite中添加。case的执行顺序与添加到Suite中的顺序是一致的。
例如：run_main.py
if __name__ == "__main__":
suite = unittest.TestSuite()
# 定义list，按照list里的顺序执行测试用例
tests=[TestFunc("test_add"),TestFunc("test_minus"),TestFunc("test_multi"),TestFunc("test_divide")]
suite.addTests(tests)
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)
TestSuite可以再包含testsuite，示例如下：
suite1 = module.TheTestSuite() suite2=module.TheTestSuite() alltests=unittest.TestSuite([suite1],[suite2])
跳过某个case：skip装饰器
若想让某个测试用例不执行，有没有办法呢？当然是有的，可以使用skip装饰器。
例如：
@unittest.skip("i don't want to run this case -> test_minus() ... ")
def test_minus(self):
print("minus")
self.assertEqual(3,minus(5,2))
加上“@unittest.skip()”后，执行看看，对比控制台的输出结果就可以明显看出区别了。
Skip装饰器有如下几种情况：
（1）skip():无条件跳过
@unittest.skip("i don't want to run this case. ")
（2）skipIf(condition,reason):如果condition为true，则 skip
@unittest.skipIf(condition,reason)
（3）skipUnless(condition,reason):如果condition为False,则skip
@unittest.skipUnless(condition,reason)
（4）还可以使用TestCase.skipTest(reason)。例如：
def test_divide(self):
self.skipTest('do not run test_divide()')
print("divide")
self.assertEqual(2,divide(4,2))
控制台输出（部分）：test_divide (__main__.TestFunc) ... skipped 'do not run test_divide()'
4、TestLoader
TestLoadder用来加载TestCase到TestSuite中。
loadTestsFrom*()方法从各个地方寻找testcase，创建实例，然后addTestSuite，再返回一个TestSuite实例。
该类提供以下方法：
unittest.TestLoader().loadTestsFromTestCase(testCaseClass) unittest.TestLoader().loadTestsFromModule(module) unittest.TestLoader().loadTestsFromName(name,module=None) unittest.TestLoader().loadTestsFromNames(names,module=None) unittest.TestLoader().getTestCaseNames(testCaseclass) unittest.TestLoader().discover()
TestLoader 之discover：
用discover()来加载测试多个测试用例，再用TextRunner类的run()方法去一次执行多个脚本的用例，达到批量执行的效果。
discover方法里面有三个参数：
- -case_dir:这个是待执行用例的目录。
- -pattern：这个是匹配脚本名称的规则，test*.py意思是匹配test开头的所有脚本。
- -top_level_dir：这个是顶层目录的名称，一般默认等于None就行了。
TextTestRunner()：执行测试用例。runner.run(test)会执行TestSuite、TestCase中的run(result)方法。
如下：run_main.py示例
import unittest import os # 用例的路径 case_path = os.path.join(os.getcwd(),"case") # 报告存放的路径 report_path = os.path.join(os.getcwd(),"report") def all_cases(): discover= unittest.defaultTestLoader.discover(case_path,pattern="test*.py",top_level_dir=None) print(discover) return discover if __name__ == "__main__": runner = unittest.TextTestRunner(verbosity=2) runner.run(all_cases())
5、生成测试报告
生成TXT测试报告
代码示例：
if __name__ == "__main__":
suite = unittest.TestSuite()
# 生成.txt的测试报告（控制台的输出写入到文件中）
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestFunc))
with open('UnittestTextReport.txt','a') as f:
runner = unittest.TextTestRunner(stream=f,verbosity=2)
runner.run(suite)
可以看到在目录下，生成了UnittestTextReport.txt文件。
但是txt格式的文件太过于简陋。我们可以借助与第三方提供的库来输出更加形象的html报告，也可以自定义输出自己想要格式的html格式的报告。
- 生成HTML测试报告
- 先下载HTMLTestRunner.py（注意Python的版本），http://tungwaiyip.info/software/HTMLTestRunner.html。然后放在Python的Lib目录下；
- 在run_main.py文件中加入：from HTMLTestRunner import HTMLTestRunner
HTMLTestRunner()方法有三个参数：
- --stream:测试报告写入文件的存储区域
- --title:测试报告的主题
- --description：测试报告的描述
代码示例：
if __name__ == "__main__":
suite = unittest.TestSuite()
# 生成HTML格式的具体测试报告
with open('HtmlReport.html','wb') as f: # 在python3，要写成'wb' or 'wr'
runner = HTMLTestRunner(stream=f,title='function test
report',description='generated by HTMLTestRunner',verbosity=2)
runner.run(suite)
三、代码示例
function.py
#!/usr/bin/python3 # -*- coding:utf-8 -*- def add(a,b): return a+b def minus(a,b): return a-b def multi(a,b): return a*b def divide(a,b): return a/b
Test_function.py
#!/usr/bin/python3
# -*- coding:utf-8 -*-
import unittest
from UnitTest_1.function import * # from..import ... :要指定文件的路径
class TestFunc(unittest.TestCase): # unittest.TestCase
# 如果想在所有case执行之前准备一次测试环境，并在所有case执行结束后再清理环境
@classmethod
def setUpClass(cls):
print("this setupclass() method only called once")
@classmethod
def tearDownClass(cls):
print("this teardownclass() method only called once too")
# 测试方法均已test开头，否则是不被unittest识别的
def test_add(self):
print("add:")
self.assertEqual(3,add(1,2))
def test_minus(self):
print("minus")
self.assertEqual(3,minus(5,2))
# 如果想临时跳过某个case：skip装饰器
@unittest.skip("i don't want to run this case. ")
def test_multi(self):
print("multi")
self.assertEqual(6,multi(2,3))
def test_divide(self):
print("divide")
self.assertEqual(2,divide(5,2))
if __name__ == "__main__":
# 在main()中加verbosity参数，可以控制输出的错误报告的详细程度
# verbosity=*：默认是1；设为0，则不输出每一个用例的执行结果；2-输出详细的执行结果
unittest.main(verbosity=2)
Test_suite.py
#!/usr/bin/python3
# -*- coding:utf-8 -*-
import unittest
from UnitTest_1.test_function import TestFunc
from HTMLTestRunner import HTMLTestRunner
# 在Python3中已经没有 StringIO，所以引用的时候要注意
from io import StringIO
if __name__ == "__main__":
suite = unittest.TestSuite()
# 定义list，按照list里的顺序执行测试用例
tests = [TestFunc("test_add"),TestFunc("test_minus"),TestFunc("test_multi"),TestFunc("test_divide")]
suite.addTests(tests)
'''
runner = unittest.TextTestRunner(verbosity=2)
runner.run(suite)
'''
# 生成.txt的测试报告（控制台的输出写入到文件中）
'''
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestFunc))
with open('UnittestTextReport.txt','a') as f:
runner = unittest.TextTestRunner(stream=f,verbosity=2)
runner.run(suite)
'''
# 生成HTML格式的具体测试报告
with open('HtmlReport.html','wb') as f: # 在python3，要写成'wb' or 'wr'
runner = HTMLTestRunner(stream=f,title='function test report',description='generated by
HTMLTestRunner',verbosity=2)
runner.run(suite)
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持【听图阁-专注于Python设计】。
