因為小白，這2天研究了好久才算是搞好。先附上一個測試完成后郵件的截圖：
上一篇有提到：
unittest中實際運行了一個接口的很多條用例，而報告中只會有一條記錄。這是因為unittest test case中我只寫了一個test方法，循環測試數據。這樣寫用例更簡單，目前在想是從報告中處理，還是測試用例讀取部分進行處理。
而且這個問題會導致后續寫test case出錯幾率大，而且耗時較多。經過幾天的思考，寫了個簡易的解決辦法。思路：輕微修改excel讀取通過excel名/sheet名/用例id名來返回數據，那我們就可以在一個unittest case中固化test_開頭的測試方法，如10條（我們約定excel中每個sheet最多寫10條用例）。 這里我們在excel中添加一個“是否執行用例”字段，unittest中啟動skipIf方法來確認是否跳過該用例，即可完美解決上一篇的問題。
excel用例數據截圖：（增加runselect字段 YES代表執行用例，No代表不執行）
那我們的OperateExcel方法也做輕微修改：
def Readexcelrow_of_sheet(filename,sheet_name,row):
'''返回某個sheet中某一行數據的列表'''
folder = "C:\\Users\\LENOVO\\PycharmProjects\\Apiautotest-master\\testdata\\"
fullfilename = folder + filename
rb = xlrd.open_workbook(fullfilename)
sheet = rb.sheet_by_name(sheet_name)
datalist = []
for i in range(10):
if sheet.cell_value(row,9) =="YES":
try:
data = sheet.cell_value(row,i)
datalist.append(data)
except TypeError as f:
break
#print (datalist)
return datalist
def Readexcelalldata_of_sheet(filename,sheet_name):
'''返回整個sheet的內容的列表'''
datalistlist =[]
for i in range(1,100):
try:
data = Readexcelrow_of_sheet(filename,sheet_name,i)
datalistlist.append(data)
except IndexError as e:
break
#print(datalistlist)
return datalistlist
這里我們增加一個數據處理DataProcess方法，把excel讀取的整個sheet數據（是一個列表，它的元素也是列表），我們用這個方法，根據行號返回一個json數據，也就是我們每條用例所要測試請求的數據。
import json
def DataProcess(datalist,myindex):
'''datalist是一個列表，列表的元素也是列表，是從excel讀到的指定sheet全部數據，根據index返回一個標准化的dict'''
data = datalist[myindex]
caseid = data[0]
casename = data[1]
method = data[2]
url = data[3]
payload = data[4].encode('utf-8')
expectedcode = data[5]
expectedmsg = data[6]
expecteddata = data[7]
headers = json.loads(data[8])
runselect = data[9]
newdict = {"runselect":runselect,
"method":method,
"caseid":caseid,
"url":url,
"data":payload,
"headers":headers,
"expectedcode":expectedcode,
"expectedmsg":expectedmsg,
"expecteddata":expecteddata}
return newdict
我們的test case樣子如下：每個test方法除了一個數據處理的入參不一樣，其他完全一致，無需更改。 而且我們可以做到所有的測試用例都按這個模板來撰寫。
class LoginTest(unittest.TestCase):
''' '''
def setUp(self):
self.logname = logname
self.mylog = AddLog.AddLog(self.logname).getlog()
self.testcaseinfo = OperateExcel.Readexcelalldata_of_sheet(Excelname,Sheetname)
self.result = {}
global caseid
caseid += 1
def tearDown(self):
self.mylog.info("case:%d completed" % caseid)
@unittest.skipIf((max_testcaseid < 1) or (0 in indexlist),"用例最大id<1或用例標記為'不執行'")
def test_caseid_1(self):
testdata = DataProcess.DataProcess(self.testcaseinfo,0)
headers= {"Content-Type": "application/json"}
self.result = RunRequest.RunRequests(testdata["method"],
testdata["url"],
testdata["data"],
testdata['headers']).run()
self.myassert = AssertResult.AssertResult(self.logname, testdata, self.result)
self.myassert.AssertEqualResult()
@unittest.skipIf((max_testcaseid < 2) or (1 in indexlist),"用例最大id<2或用例標記為'不執行'")
def test_caseid_2(self):
testdata = DataProcess.DataProcess(self.testcaseinfo, 1)
self.result = RunRequest.RunRequests(testdata["method"],
testdata["url"],
testdata["data"],
testdata['headers']).run()
self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
self.myassert.AssertEqualResult()
@unittest.skipIf((max_testcaseid < 3) or (2 in indexlist),"用例最大id<3或用例標記為'不執行'")
def test_caseid_3(self):
testdata = DataProcess.DataProcess(self.testcaseinfo, 2)
self.result = RunRequest.RunRequests(testdata["method"],
testdata["url"],
testdata["data"],
testdata['headers']).run()
self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
self.myassert.AssertEqualResult()
@unittest.skipIf((max_testcaseid < 4) or (3 in indexlist), "用例最大id<4或用例標記為'不執行'")
def test_caseid_4(self):
testdata = DataProcess.DataProcess(self.testcaseinfo, 3)
self.result = RunRequest.RunRequests(testdata["method"],
testdata["url"],
testdata["data"],
testdata['headers']).run()
self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
self.myassert.AssertEqualResult()
因為要用到不同請求方式，我們把他封裝在一起了。
#coding = utf-8
import requests
import json
class RunRequests(object):
def __init__(self,method,url,data,headers):
self.url = url
self.method = method
self.data = data
self.headers = headers
self.result = ""
def run(self):
if self.method == "POST":
if self.headers != "No":
self.result = requests.post(self.url, data = self.data, headers = self.headers)
else:
self.result = requests.post(self.url, data = self.data)
elif self.method == "GET":
if self.headers != "No":
self.result = requests.get(self.url, data=self.data)
print("暫未處理headers")
else:
self.result = requests.get(self.url,data = self.data)
elif self.method == "PUT":
self.result = 1
elif self.method == "DELETE":
self.result = 1
elif self.method == "HEAD":
self.result = 1
else:
self.result = 1
return self.result.json()
這些都完成后，想到測試完成后需要日志和郵件功能。也添加進來。
寫完后整個框架就如下這個樣子：
如果有朋友看到，覺得可以學習練練手，有需要源碼可以留言，看到會及時回復。