unittest是一個單元測試框架,是Python編程的單元測試框架。有時候,也做叫做“PyUnit”,是Junit的Python語言版本。這里了解下,Junit是Java語言的單元測試框架,Java還有一個很好用的單元測試框架叫TestNG,本系列只學習Python,所以只需要unittest是Python里的一個單元測試框架可以了。
unittest支持測試自動化,共享測試用例中的初始化和關閉退出代碼,在unittest中小單元是test,也是一個測試用例。要了解unittest單元測試框架,先來了解以下幾個重要的概念。
測試固件(test fixture)
一個測試固件包括兩部分,執(zhí)行測試代碼之前的準備部分和測試結束之后的清掃代碼。這兩部分一般用函數(shù)setUp()和tearDown()表示。這里舉例以下,例如要測試百度搜索selenium這個場景,我們的測試固件可以這樣寫,setUp()里寫打開瀏覽器,瀏覽器大化,和打開百度首頁等腳本代碼;在tearDown()里寫結束搜索后,退出并關閉瀏覽器的代碼。
測試用例(test case)
unittest中管理的小單元是測試用例,一個測試用例,包括測試固件,和具體測試業(yè)務的函數(shù)或者方法。一個測試用例中,測試固件可以不寫,但是至少有一個已test開頭的函數(shù)。unittest會自動化識別test開頭的函數(shù)是測試代碼,如果你寫的函數(shù)不是test開頭,unittest是不會執(zhí)行這個函數(shù)里面的腳本的,這個千萬要記住,所有的測試函數(shù)都要test開頭,記住是小寫的哦。
測試套件 (test suite)
很簡單,是很多測試用例的集合,叫測試套件,一個測試套件可以隨意管理多個測試用例。如果測試用例比作單個學生,測試套件是好像是班級的概念。
測試執(zhí)行器 (test runner)
test runner是一個用來執(zhí)行加載測試用例,并執(zhí)行用例,且提供測試輸出的一個組建。test runner可以加載test case或者test suite進行執(zhí)行測試任務。
我們舉例來,練習一下test fixture和test case的使用,學習unittest的簡單用法:
1. 新建一個testbaidu.py的文件
2. 導入unittest模塊
3. 當前測試類繼承unittest.TestCase,相當于當前利用unittest創(chuàng)建了一個test case,這個test case是能夠被unittest直接識別。
4. 寫setUP(),主要是打開瀏覽器和打開站點
5. 寫一個test_search()用例寫搜索的代碼
6. 寫tearDown(),主要是瀏覽器退出操作
相關腳本代碼如下:
[python] view plain copy
# coding=utf-8
import time
import unittest
from selenium import webdriver
class BaiduSearch(unittest.TestCase):
def setUp(self):
"""
測試固件的setUp()的代碼,主要是測試的前提準備工作
:return:
"""
self.driver = webdriver.Chrome()
self.driver.maximize_window()
self.driver.implicitly_wait(8)
self.driver.get("https://www.baidu.com")
def tearDown(self):
"""
測試結束后的操作,這里基本上都是關閉瀏覽器
:return:
"""
self.driver.quit()
def test_baidu_search(self):
"""
這里一定要test開頭,把測試邏輯代碼封裝到一個test開頭的方法里。
:return:
"""
self.driver.find_element_by_id('kw').send_keys('selenium')
time.sleep(1)
try:
assert 'selenium' in self.driver.title
print ('Test Pass.')
except Exception as e:
print ('Test Fail.', format(e))
if __name__ == '__main__':
unittest.main()
解釋:
后結尾處的unittest.main(),添加這個是支持在cmd,里面,cd到這個腳本文件所在的目錄,然后Python 腳本名.py執(zhí)行,如果不添加這一段,是無法執(zhí)行cmd里面運行腳本的,在PyCharm中,不添加后一段,也可以通過,右鍵 Run "unittest xxx",來達到執(zhí)行效果。