Python Selenium 之數(shù)據(jù)驅動測試的實現(xiàn)
數(shù)據(jù)驅動模式的測試好處相比普通模式的測試就顯而易見了吧!使用數(shù)據(jù)驅動的模式,可以根據(jù)業(yè)務分解測試數(shù)據(jù),只需定義變量,使用外部或者自定義的數(shù)據(jù)使其參數(shù)化,從而避免了使用之前測試腳本中固定的數(shù)據(jù)??梢詫y試腳本與測試數(shù)據(jù)分離,使得測試腳本在不同數(shù)據(jù)集合下高度復用。不僅可以增加復雜條件場景的測試覆蓋,還可以極大減少測試腳本的編寫與維護工作。
下面將使用Python下的數(shù)據(jù)驅動模式(ddt)庫,結合unittest庫以數(shù)據(jù)驅動模式創(chuàng)建百度搜索的測試。
ddt庫包含一組類和方法用于實現(xiàn)數(shù)據(jù)驅動測試??梢詫y試中的變量進行參數(shù)化。
可以通過python自帶的pip命令進行下載并安裝:pip install ddt . 更多關于ddt的信息可以參考:
一個簡單的數(shù)據(jù)驅動測試
為了創(chuàng)建數(shù)據(jù)驅動測試,需要在測試類上使用@ddt裝飾符,在測試方法上使用@data裝飾符。@data裝飾符把參數(shù)當作測試數(shù)據(jù),參數(shù)可以是單個值、列表、元組、字典。對于列表,需要用@unpack裝飾符把元組和列表解析成多個參數(shù)。
下面實現(xiàn)百度搜索測試,傳入搜索關鍵詞和期望結果,代碼如下:
import unittest
from selenium import webdriver
from ddt import ddt, data, unpack
@ddt
class SearchDDT(unittest.TestCase):
'''docstring for SearchDDT'''
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.driver.maximize_window()
self.driver.get("https://www.baidu.com")
# specify test data using @data decorator
@data(('python', 'PyPI'))
@unpack
def test_search(self, search_value, expected_result):
search_text = self.driver.find_element_by_id('kw')
search_text.clear()
search_text.send_keys(search_value)
search_button = self.driver.find_element_by_id('su')
search_button.click()
tag = self.driver.find_element_by_link_text("PyPI").text
self.assertEqual(expected_result, tag)
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
在test_search()方法中,search_value與expected_result兩個參數(shù)用來接收元組解析的數(shù)據(jù)。當運行腳本時,ddt把測試數(shù)據(jù)轉換為有效的python標識符,生成名稱為更有意義的測試方法。結果如下:

使用外部數(shù)據(jù)的數(shù)據(jù)驅動測試
如果外部已經存在了需要的測試數(shù)據(jù),如一個文本文件、電子表格或者數(shù)據(jù)庫,那也可以用ddt來直接獲取數(shù)據(jù)并傳入測試方法進行測試。
下面將借助外部的CSV(逗號分隔值)文件和EXCLE表格數(shù)據(jù)來實現(xiàn)ddt。
通過CSV獲取數(shù)據(jù)
同上在@data裝飾符使用解析外部的CSV(testdata.csv)來作為測試數(shù)據(jù)(代替之前的測試數(shù)據(jù))。其中數(shù)據(jù)如下:

接下來,先要創(chuàng)建一個get_data()方法,其中包括路徑(這里默認使用當前路徑)、CSV文件名。調用CSV庫去讀取文件并返回一行數(shù)據(jù)。再使用@ddt及@data實現(xiàn)外部數(shù)據(jù)驅動測試百度搜索,代碼如下:
import csv, unittest
from selenium import webdriver
from ddt import ddt, data, unpack
def get_data(file_name):
# create an empty list to store rows
rows = []
# open the CSV file
data_file = open(file_name, "r")
# create a CSV Reader from CSV file
reader = csv.reader(data_file)
# skip the headers
next(reader, None)
# add rows from reader to list
for row in reader:
rows.append(row)
return rows
@ddt
class SearchCSVDDT(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.driver.maximize_window()
self.driver.get("https://www.baidu.com")
# get test data from specified csv file by using the get_data funcion
@data(*get_data('testdata.csv'))
@unpack
def test_search(self, search_value, expected_result):
search_text = self.driver.find_element_by_id('kw')
search_text.clear()
search_text.send_keys(search_value)
search_button = self.driver.find_element_by_id('su')
search_button.click()
tag = self.driver.find_element_by_link_text("PyPI").text
self.assertEqual(expected_result, tag)
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
測試執(zhí)行時,@data將調用get_data()方法讀取外部數(shù)據(jù)文件,并將數(shù)據(jù)逐行返回給@data。執(zhí)行的結果也同上~
通過Excel獲取數(shù)據(jù)
測試中經常用Excle存放測試數(shù)據(jù),同上在也可以使用@data裝飾符來解析外部的CSV(testdata.csv)來作為測試數(shù)據(jù)(代替之前的測試數(shù)據(jù))。其中數(shù)據(jù)如下:

接下來,先要創(chuàng)建一個get_data()方法,其中包括路徑(這里默認使用當前路徑)、EXCEL文件名。調用xlrd庫去讀取文件并返回數(shù)據(jù)。再使用@ddt及@data實現(xiàn)外部數(shù)據(jù)驅動測試百度搜索,代碼如下:
import xlrd, unittest
from selenium import webdriver
from ddt import ddt, data, unpack
def get_data(file_name):
# create an empty list to store rows
rows = []
# open the CSV file
book = xlrd.open_workbook(file_name)
# get the frist sheet
sheet = book.sheet_by_index(0)
# iterate through the sheet and get data from rows in list
for row_idx in range(1, sheet.nrows): #iterate 1 to maxrows
rows.append(list(sheet.row_values(row_idx, 0, sheet.ncols)))
return rows
@ddt
class SearchEXCLEDDT(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.driver.maximize_window()
self.driver.get("https://www.baidu.com")
# get test data from specified excle spreadsheet by using the get_data funcion
@data(*get_data('TestData.xlsx'))
@unpack
def test_search(self, search_value, expected_result):
search_text = self.driver.find_element_by_id('kw')
search_text.clear()
search_text.send_keys(search_value)
search_button = self.driver.find_element_by_id('su')
search_button.click()
tag = self.driver.find_element_by_link_text("PyPI").text
self.assertEqual(expected_result, tag)
def tearDown(self):
self.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
與上面讀取CVS文件一樣,測試執(zhí)行時,@data將調用get_data()方法讀取外部數(shù)據(jù)文件,并將數(shù)據(jù)逐行返回給@data。執(zhí)行的結果也同上~
如果想從數(shù)據(jù)庫的庫表中獲取數(shù)據(jù),同樣也需要一個get_data()方法,并且通過DB相關的庫來連接數(shù)據(jù)庫、SQL查詢來獲取測試數(shù)據(jù)。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
Python利用字典和列表實現(xiàn)學生信息管理系統(tǒng)
這篇文章主要為大家詳細介紹了Python如何利用字典和列表實現(xiàn)一個簡易的學生信息管理系統(tǒng),文中詳細注釋,方便理解,需要的可以參考一下2022-06-06
Python創(chuàng)建7種不同的文件格式的方法總結
今天的這篇文章呢,小編來介紹一下如何通過Python來創(chuàng)建各種形式的文件,這里包括了:文本文件、CSV文件、Excel文件、壓縮文件、XML文件、JSON文件和PDF文件,需要的可以參考一下2023-01-01
一文帶你精通Python中*args和**kwargs的應用技巧
如果能在Python中創(chuàng)建適應不同場景的函數(shù),而無需每次都重寫它們,會使得操作簡潔方便,這就是*args和**kwargs的魔力所在,下面我們就來看看它們的具體一些應用技巧吧2024-03-03
python-django中的APPEND_SLASH實現(xiàn)方法
這篇文章主要介紹了python-django中的APPEND_SLASH實現(xiàn)方法,本文給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-06-06

