python自動化測試中裝飾器@ddt與@data源碼深入解析
一、使用ddt和data裝飾器的大致框架如下,每個test_開頭的方法,代表一條測試用例
from ddt import ddt,data import unittest test_datas=[ {'id':1,'title':'測試用例1'}, {'id':2,'title':'測試用例2'}, {'id':3,'title':'測試用例3'}, {'id':4,'title':'測試用例4'} ] @ddt class TestDemo(unittest.TestCase): @data(*test_datas) def test_demo1(self,item): print('測試用例執行',item)
unittest中的測試用例:
測試類中每一個test開頭的方法就是一條測試用例
ddt根據用例數據生成測試用例的思路:
1、利用data裝飾器:傳入測試數據,在裝飾器中將測試數據保存起來
2、ddt這個裝飾器:遍歷測試數據,每遍歷出一條數據,往測試類中添加一個test開頭的方法
setattr(類,方法名,方法)
二、給類動態的增加方法
案例1
setattr(對象/類,屬性名/方法名,屬性值/方法)
特別註意:
給類動態增加方法一定要加self
class Demo: def test_1(self): print("這個是方法test_1") def kobe(self,item): print("kobe-----執行瞭",item) datas=[2,8,23,22,24] #根據數據動態給測試類中增加5個方法 for i in datas: name='test_1_{}'.format(i) #給類動態增加方法 setattr(Demo,name,kobe) print(Demo.__dict__)
案例2:
調用動態執行的5個方法,執行結果都為kobe—–執行瞭 24,有bug
class Demo: def test_1(self): print("這個是方法test_1") def kobe(self,item): print("kobe-----執行瞭",item) datas=[2,8,23,22,24] #根據數據動態給測試類中增加5個方法 for i in datas: name='test_1_{}'.format(i) def wrapper(self): kobe(self,i) #給類動態增加方法 setattr(Demo,name,wrapper) #print(Demo.__dict__) Demo().test_1_2() Demo().test_1_8() Demo().test_1_22() Demo().test_1_23() Demo().test_1_24()
執行結果:
kobe—–執行瞭 24
kobe—–執行瞭 24
kobe—–執行瞭 24
kobe—–執行瞭 24
kobe—–執行瞭 24
原因分析
案例3:
解決案例2的bug
定義閉包create_method:進行數據鎖定,鎖定的是datas=[2,8,23,22,24]
class Demo: def test_1(self): print("這個是方法test_1") def kobe(self,item): print("kobe-----執行瞭",item) datas=[2,8,23,22,24] #todo 使用閉包進行數據鎖定 def create_method(i): def wrapper(self): kobe(self,i) return wrapper #根據數據動態給測試類中增加5個方法 for i in datas: name='test_1_{}'.format(i) wrapper=create_method(i) #給類動態增加方法 setattr(Demo,name,wrapper) Demo().test_1_2() Demo().test_1_8() Demo().test_1_22() Demo().test_1_23() Demo().test_1_24()
三、ddt和data的源碼解析
from ddt import ddt,data import unittest test_datas=[ {'id':1,'title':'測試用例1'}, {'id':2,'title':'測試用例2'}, {'id':3,'title':'測試用例3'}, {'id':4,'title':'測試用例4'} ] def ddt(cls): '''遍歷測試數據,給類動態添加方法''' #如何通過類獲取方法? #res=cls.__dict__ #print('測試類的方法和屬性字典',res) for name,method in list(cls.__dict__.items()): #遍歷出來的屬性值(方法)是否擁有datas屬性(測試數據) if hasattr(method,'datas'): #獲取方法中保存的測試數據 datas=getattr(method,'datas') #遍歷測試數據 for index,value in enumerate(datas): print("數據:",value) #給測試類動態添加用例 method_name='{}_{}'.format(name,index+1) print('方法名',method_name) #給類動態的增加方法 def wrapper(self): method(self, value) #todo 給測試類動態添加一個測試方法 setattr(cls,method_name,wrapper) return cls def data(*args): '''將測試數據保存為測試方法的屬性''' #*args接收到的是data裝飾器傳遞進來的數據 def wrapper(func): #func接收的是data裝飾的函數 func.datas=args return func return wrapper @ddt class TestDemo(): @data(*test_datas) #test_demo1=data(*test_datas)(test_demo1) def test_demo1(self,item): print('測試用例執行',item) #print(TestDemo.test_demo1.__dict__)
這樣寫的話有bug
原因:
解決:
采用閉包進行數據鎖定,鎖定value和method
def create_test_method(method,value): def wrapper(self): method(self, value) return wrapper
from ddt import ddt,data import unittest test_datas=[ {'id':1,'title':'測試用例1'}, {'id':2,'title':'測試用例2'}, {'id':3,'title':'測試用例3'}, {'id':4,'title':'測試用例4'} ] def create_test_method(method,value): def wrapper(self): method(self, value) return wrapper def ddt(cls): #todo @ddt這個裝飾器:遍歷測試數據,每遍歷出一條數據,往測試類中添加一個test開頭的方法 #setattr(類,方法名,方法) res=list(cls.__dict__.items()) print(res) for name,method in res: print(name,method) if hasattr(method,'datas'): #如果有datas屬性,獲取方法中保存的datas datas=getattr(method,'datas') #遍歷測試數據 for index,value in enumerate(datas): print('測試數據:',value) #給測試類動態的增加測試用例 method_name='{}_{}'.format(name,index+1) print('方法:',method_name,method) #todo 給類動態的增加方法,最終希望執行def test_demo1(self,item):這個方法的 #test_method=method #但是item需要自己傳,但是unittest是不需要傳遞參數的 # def wrapper(self): # method(self,value) wrapper=create_test_method(method, value) # todo 給測試類動態添加一個測試方法 setattr(cls, method_name, wrapper) else: delattr(cls,name) return cls def data(*args): # *args為給裝飾器傳遞的參數test_datas def wrapper(func): # func為被裝飾器裝飾的函數test_demo1 #todo @data裝飾器的作用是保存測試數據,將測試數據存放到函數屬性中 func.datas = test_datas return func return wrapper @ddt class TestDemo(unittest.TestCase): @data(*test_datas) #test_demo1=data(*test_datas)(test_demo1) def test_demo1(self,item): print('測試用例執行',item)
分部解析代碼
@data(*test_datas) def test_demo1(self,item): print('測試用例執行',item)
1、上面3行代碼可以寫成如下:
@data(*test_datas)
:可以表示為test_demo1=data(*test_datas
)(test_demo1)
2、輸出屬性(方法)名稱和屬性值
for name,method in list(cls.__dict__.items())
3、將遍歷出來的屬性名(方法)判斷是否包含datas屬性,如果有datas屬性,獲取方法中保存的datas
if hasattr(method,'datas'): datas=getattr(method,'datas')
總結
到此這篇關於python自動化測試中裝飾器@ddt與@data源碼解析的文章就介紹到這瞭,更多相關python裝飾器@ddt和@data源碼解析內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- 如何利用python處理原始音頻數據
- python中unittest框架應用詳解
- python魔法方法之__setattr__()
- 微信小程序抽獎組件的使用步驟
- 基於Python的接口自動化unittest測試框架和ddt數據驅動詳解