本章主要介紹pytest幾種數據驅動的方法,也是我們做介面自動化中經常要使用到的,大致分為以下四種。 一.yaml 二.json 三.csv 四.excel 一.yaml 1.安裝插件:pip install pyyaml 2.yaml的兩種讀寫方法load()和dump(),話不多說,直接上代碼 ...
本章主要介紹pytest幾種數據驅動的方法,也是我們做介面自動化中經常要使用到的,大致分為以下四種。
一.yaml
二.json
三.csv
四.excel
一.yaml
1.安裝插件:pip install pyyaml
2.yaml的兩種讀寫方法load()和dump(),話不多說,直接上代碼
load()為讀取json流,讀取是加上Loader=yaml.FullLoader,防止亂碼,self.data為讀取的yaml文件
def param_yaml_ready(self): with open(self.data, mode="r", encoding="utf-8") as f1: data = yaml.load(f1, Loader=yaml.FullLoader) return data
dump()為寫入dict類型的數據,allow_unicode=True防止寫入時亂碼,
def param_yaml_write(寫入dict類型的數據): with open("寫入的文件名.yaml", mode="w", encoding="utf-8") as f1: data = yaml.dump(寫入dict類型的數據, f1, allow_unicode=True) return data
3.yaml文件的寫法。
yaml文件中-代碼中括弧,list的意思,每個層級代表一個{},dict的意思,下麵內容讀取出來後展示如下
- class: name: 李明 age: 12
讀取的內容:[{'class': {'name': '李明', 'age': 12}}]
4.實例。下麵我們以一個具體的實例來說下yaml的用法
(1)首先我們先創建一個tokenApproval.yaml文件,具體寫法如下。這個yaml文件中只有一條case
- #獲取審核token tokenApproval: - request: url: https://qyapi.weixin.qq.com/cgi-bin/gettoken #request-url param: #request-body corpid: ww29957e72e534d2 corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz methon: get #request-methon response: eq: errcode: 0 #斷言判斷
(2)然後創建params.py封裝yaml讀寫方法,./params/globalParametersToken.yaml是我寫入json數據的yaml文件
def param_yaml_ready(self): with open(self.data, mode="r", encoding="utf-8") as f1: data = yaml.load(f1, Loader=yaml.FullLoader) return data @staticmethod def param_yaml_write(paramYaml): with open("./params/globalParametersToken.yaml", mode="w", encoding="utf-8") as f1: data = yaml.dump(paramYaml, f1, allow_unicode=True) return data
(3) 結合pytest.mark.parametrize實現數據驅動,pytest.mark.parametrize具體寫法格式可查考:
https://www.cnblogs.com/lihongtaoya/p/15840026.html
import pytest from params.params import param import requests class Testtoken: @pytest.mark.parametrize("parama", param("./params/tokenApproval.yaml").param_yaml_ready()) def test_tokenApproval(self, parama): url = parama["tokenApproval"][0]["request"]["url"] params = parama["tokenApproval"][0]["request"]["param"] data = requests.get(url, params) assert data.json()["errcode"] == parama["tokenApproval"][0]["response"]["eq"]["errcode"]
若是yaml文件中寫入的單介面數據不止一條case,如下代碼,那麼parama["tokenApproval"][0]這樣的
寫法就不適用了,這裡需要添加一個for迴圈即可for i in range(0, len(parama["tokenApproval"])):
- #獲取審核token tokenApproval: - request: url: https://qyapi.weixin.qq.com/cgi-bin/gettoken param: corpid: ww29957e72e534d2 corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz methon: get response: eq: errcode: 0 - request: url: https://qyapi.weixin.qq.com/cgi-bin/gettoken param: corpid: ww29957e72e534d2 corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjk methon: get response: eq: errcode: 0
若yaml文件中寫入的是多介面數據,如下代碼寫法,這裡每個介面我們只要有一個關鍵字去區分即可:
tokenApproval || tokenContacts,這樣我們在獲取的時候parama["tokenApproval"],tokenApproval
這個值隨機變換即可
-
#獲取審核token
tokenApproval:
- request:
url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
param:
corpid: ww29957e72e534d27e
corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc
methon: get
response:
eq:
errcode: 0
- request:
url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
param:
corpid: ww29957e72e534d27e
corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz
methon: get
response:
eq:
errcode: 0
#獲取通訊錄token
tokenContacts:
- request:
url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
param:
corpid: ww29957e72e534d27e
corpsecret: hUQCww5PR0o_QHF4HBt7lLQlVf3M5Amssx6OQZxUkPY
methon: get
response:
eq:
errcode: 0
(4)dump()用法,先獲取相應的結果,已鍵值對的格式添加到list中,最後調用封裝好的寫入方法即可。
datalist = [{"tokenApproval": data.json()["token"]}] param.param_write_csv(datalist)
二.json
1.json讀寫的兩種方法和yaml一致load()和dump()
with open("讀取的json文件", mode="r", encoding="utf-8") as p: data = json.load(p) with open("寫入數據的json文件", mode="w", encoding="utf-8")as p: json.dump(寫入的數據data, p)
2.剩餘部分json的操作方法和yaml的大致相同,唯一不同的區別就是json文件的寫法。在實際操作中yaml中寫
case會比json方便一點,所以我們可以在yaml寫完後,通過第三方工具轉換json格式,這樣方便的很多:
https://json.im/json2/yaml2json.html。轉換過後獲取數據和yaml方法相同
三.csv
1.csv兩種讀寫的方法reader()和write()
下圖代碼,csv.reader(file)返回的是一個可迭代的結果集,最後通過for迴圈遍歷出來添加到list中。writerow()
為寫入一個列表,也就是一行數據,如果要寫入多條數據的話可使用writerows()方法,寫法writes.writerows([list1,list2,list3........])
with open("讀取的csv文件", mode="r", encoding="utf-8")as f: data = csv.reader(f) dataparam=[] for lis in data: dataparam.append(lis) with open("寫入的csv文件", mode="w", encoding="utf-8")as f: writes = csv.writer(f) writes.writerow("寫入的list")
2.csv文件寫法
首先我們在創建csv文件後打開會在右上角看見一個鍵盤似地圖標,點擊後選擇csv後點擊確定,這個時候在改
文件左下角有兩個tab,text and data
text寫法類似文本,每個元素中間用逗號隔開,而data則是表格的寫法。這裡需要註意的是如果你寫入的是鍵值
對格式,需要嚴格按照格式來寫,如下02例子
01
name,age,class 李明,12,三年級二班
02
class,name
三年級二班,"{""name"":""小明"",""age"":""12""}"
三.實例。下麵我們以一個具體的實例來說下csv的用法
(1)首先我們創建一個tokencsv.csv文件,具體寫法如下
tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d2"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0
(2)來封裝我們所需要的讀寫方法
def param_reader_csv(self): with open(self.data, mode="r", encoding="utf-8")as f: data = csv.reader(f) dataparam=[] for lis in data: dataparam.append(lis) return dataparam @staticmethod def param_write_csv(data): with open("./params/globalParametersToken.csv", mode="w", encoding="utf-8")as f: writes = csv.writer(f) writes.writerow(data)
(3)結合pytest.mark.parametrize實現數據驅動,parametrize中添寫的變數數量一定要和csv文件中的數
量對齊,否則會報錯。另外這裡讀取csv文件中的內容都是以str 類型輸出的,所以當涉及到獲取鍵值
對數據時,即下方parama入參值,需使用json.loads()方法將其轉成dict,
@pytest.mark.parametrize("typetoken,methon,url,parama,errcode", param("./params/tokenCsv.csv").param_reader_csv()) def test_tokenApproval(self, typetoken, methon, url, parama, errcode): data = method(json.loads(parama), url).request_get() if data.json()["errcode"] == int(errcode): tokenList = [data.json()["access_token"]] param.param_write_csv(tokenList) else: with open("./log/error.log", mode="a+", encoding="utf-8")as f: f.write( f"[{datetime.datetime.now()}]:獲取審核token失敗,response:{data.json()},request-body:{data.request.body},request-url:{data.request.url}\n")
當csv文件中寫入單介面多條case時,無需過多處理,按照以上代碼即可。當csv文件中寫入多介面多條
case時,我們可以在csv文件中寫入一個變數來區分這條case是哪條介面的,即上面代碼typetoken參數。
cvs文件中首個數據tokenApproval和tokenContacts可代表不同的介面,我們在處理數據時加個if判斷即可
(這種方法不太能接受,所以csv做數據驅動時儘量一個介面寫 一個csv文件)
tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27e"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0 tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0 tokenContacts,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27e"",""corpsecret"":""hUQCww5PR0o_QHF4HBt7lLQlVf3M5Amssx6OQZxUkPY""}",0
四.excel
表格的操作太過複雜,能不用就不用吧,這裡就簡單展示的我封裝讀取excel的方法吧
def param_excel(self): file = openpyxl.load_workbook(self.data) file_name = file.active data = [] for x in range(1, file_name.max_row + 1): data1 = [] for y in range(1, file_name.max_column + 1): data1.append(file_name.cell(x, y).value) data2 = tuple(data1) data.append(data2) return data
完.....................