Python中有一個(gè)自帶的單元測(cè)試框架是unittest模塊，用它來做單元測(cè)試，它里面封裝好了一些校驗(yàn)返回的結(jié)果方法和一些用例執(zhí)行前的初始化操作。
unittest是python的標(biāo)準(zhǔn)測(cè)試庫，相比于其他測(cè)試框架是python目前使用最廣的測(cè)試框架。
unittest有四個(gè)比較重要的概念是test fixture, test case, test suite, test runner, 。
在說unittest之前，先說幾個(gè)概念：
TestCase 也就是測(cè)試用例
TestSuite 多個(gè)測(cè)試用例集合在一起，就是TestSuite
TestLoader是用來加載TestCase到TestSuite中的
TestRunner是來執(zhí)行測(cè)試用例的,測(cè)試的結(jié)果會(huì)保存到TestResult實(shí)例中，包括運(yùn)行了多少測(cè)試用例，成功了多少，失敗了多少等信息
通過dir(unittest)，我們可以看到unittest全部的屬性和方法，他們的關(guān)系如下圖所示。
unittest主要類關(guān)系
正常調(diào)用unittest的流程是，TestLoader 自動(dòng)將測(cè)試用例TestCase中加載到TestSuite里，TextTestRunner調(diào)用TestSuite的run方法，順序執(zhí)行里面的TestCase中以test開頭的方法，并得到測(cè)試結(jié)果TestResult。在執(zhí)行TestCase過程中，先進(jìn)行SetUp()環(huán)境準(zhǔn)備，執(zhí)行測(cè)試代碼，最后tearDown()進(jìn)行測(cè)試的還原。
其中TestLoader在加載過程中，進(jìn)行添加的TestCase是沒有順序的。一個(gè)TestCase里如果存在多個(gè)驗(yàn)證方法的話，會(huì)按照方法中test后方首字母的ascii碼從小到大排序后執(zhí)行。
可以通過手動(dòng)調(diào)用TestSuite的addTest、addTests方法來動(dòng)態(tài)添加TestCase，這樣既可以確定添加用例的執(zhí)行順序，也可避免TestCase中的驗(yàn)證方法一定要用test開頭。
下面寫一個(gè)簡單的單元測(cè)試用例
import unittestclass MyTest(unittest.TestCase): # 繼承unittest.TestCase def tearDown(self): # 每個(gè)測(cè)試用例執(zhí)行之后做操作 print('111') def setUp(self): # 每個(gè)測(cè)試用例執(zhí)行之前做操作 print('22222') @classmethod def tearDownClass(self): # 必須使用 @ classmethod裝飾器, 所有test運(yùn)行完后運(yùn)行一次 print('4444444') @classmethod def setUpClass(self): # 必須使用@classmethod 裝飾器,所有test運(yùn)行前運(yùn)行一次 print('33333') def test_a_run(self): self.assertEqual(1, 1) # 測(cè)試用例 def test_b_run(self): self.assertEqual(2, 2) # 測(cè)試用例 if __name__ == '__main__': unittest.main()#運(yùn)行所有的測(cè)試用例下面是一些常用的斷言，也就是校驗(yàn)結(jié)果
assertEqual(a, b) a == b assertNotEqual(a, b) a != b assertTrue(x) bool(x) is True assertFalse(x) bool(x) is False assertIsNone(x) x is None assertIsNotNone(x) x is not None assertIn(a, b) a in b assertNotIn(a, b) a not in b
那如何生成一個(gè)測(cè)試報(bào)告呢，需要加入另外一個(gè)模塊了，HTMLTestRunner，這個(gè)模塊需要自己安裝，使用執(zhí)行測(cè)試用例就會(huì)生成一個(gè)html的測(cè)試報(bào)告，里面會(huì)有每個(gè)測(cè)試用例的執(zhí)行結(jié)果，代碼如下：
import HTMLTestRunner import unittest class MyTest(unittest.TestCase):#繼承unittest.TestCase def tearDown(self): #每個(gè)測(cè)試用例執(zhí)行之后做操作 print('111') def setUp(self): #每個(gè)測(cè)試用例執(zhí)行之前做操作 print(22222) def test_run(self): # self.assertEqual(1,1) self.assertIs(1,1) #測(cè)試用例 def test_run2(self): # self.assertEqual(1,1) self.assertIs(1,1) #測(cè)試用例 def test_run3(self): # self.assertEqual(1,1) self.assertIs(1,1) #測(cè)試用例 def test_run1(self): # self.assertEqual(1,1) self.assertIs(1,1) #測(cè)試用例 if __name__ == '__main__': test_suite = unittest.TestSuite()#創(chuàng)建一個(gè)測(cè)試集合 test_suite.addTest(MyTest('test_run1'))#測(cè)試套件中添加測(cè)試用例 #test_suite.addTest(unittest.makeSuite(MyTest))#使用makeSuite方法添加所有的測(cè)試方法 fp = open('res.html','wb')#打開一個(gè)保存結(jié)果的html文件 runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='api測(cè)試報(bào)告',description='測(cè)試情況') #生成執(zhí)行用例的對(duì)象 runner.run(test_suite) #執(zhí)行測(cè)試套件如果我們有很多個(gè)模塊，每個(gè)模塊下面都寫了很多python文件，每個(gè)python文件里面都有測(cè)試用例，那怎么把這個(gè)目錄下的用例都執(zhí)行了呢，就要先找到這個(gè)目錄下的所有python文件，然后找到里面的測(cè)試用例，逐個(gè)執(zhí)行，代碼如下：
import unittest,HTMLTestRunner suite = unittest.TestSuite()#創(chuàng)建測(cè)試套件 all_cases = unittest.defaultTestLoader.discover('.','test_*.py') #找到某個(gè)目錄下所有的以test開頭的Python文件里面的測(cè)試用例 for case in all_cases: suite.addTests(case)#把所有的測(cè)試用例添加進(jìn)來 fp = open('res.html','wb') runner = HTMLTestRunner.HTMLTestRunner(stream=fp,title='all_tests',description='所有測(cè)試情況') runner.run(suite) #運(yùn)行測(cè)試我們?cè)诤罄m(xù)進(jìn)行持續(xù)集成的時(shí)候，要讓代碼自動(dòng)運(yùn)行，就會(huì)用到Jenkins了，但是上面產(chǎn)生的測(cè)試報(bào)告都是html格式的，Jenkins不認(rèn)識(shí)，就在Jenkins里面顯示不出來。那咱們就要產(chǎn)生一些Jenkins認(rèn)識(shí)的測(cè)試報(bào)告，Jenkins認(rèn)識(shí)xml格式的報(bào)告，那咱們就產(chǎn)生xml格式的唄，就需要用一個(gè)新的模塊，xmlrunner，安裝直接 pip install xmlrunner即可，代碼如下：
import unittestimport xmlrunner#導(dǎo)入這個(gè)模塊class My(unittest.TestCase): def test1(self,a,b,c): self.assertEqual(a+b,c) if __name__=='__main__': test_suite = unittest.TestSuite() test_suite.addTest(unittest.makeSuite(My)) runner = xmlrunner.XMLTestRunner(output='report')#指定報(bào)告放的目錄 runner.run(test_suite)
然后咱們運(yùn)行，可以看到在report目錄下已經(jīng)產(chǎn)生了xml格式的報(bào)告了，而且還自動(dòng)把日期加上了
以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持VEVB武林網(wǎng)。
新聞熱點(diǎn)
疑難解答
圖片精選
