背景
最近公司在做消息推送，那么自然就會產生很多接口，測試的過程中需要調用接口，我就突然覺得是不是可以自己寫一個測試框架？
說干就干，由于現有的接口測試工具Jmeter、SoupUI等學習周期有點長，干脆自己寫一個吧，不求人，所有功能自己都能一清二楚。
當然，寫工具造輪子只是學習的一種方式，現成成熟的工具肯定比我們自己的寫的好用。
開發環境
-------------------------------------------------------------
操作系統：Mac OS X EI Caption
Python版本：2.7
IDE：Pycharm
-------------------------------------------------------------
分析
接口是基于HTTP協議的，那么說白了，就是發起HTTP請求就行了，對于Python來說簡直就是小菜一碟。直接使用requests就可以很輕松的完成任務。
架構
整個框架是比較小的，涉及的東西也比較少，只要分清楚幾個模塊的功能就行了。
上面是一個接口測試的完整流程。只要一步一步的走下來就行了，并不是很難。
數據源
數據源我使用的是JSON來保存，當然，比較廣泛的是使用Excel來保存，用JSON來保存是因為JSON用起來比較方便，懶得去讀取Excel了，Python對JSON的支持是非常友好的。當然這個就看個人喜好了。
{
"TestId": "testcase004",
"Method": "post",
"Title": "單獨推送消息",
"Desc": "單獨推送消息",
"Url": "http://xxx.xxx.xxx.xx",
"InputArg": {
"action": "44803",
"account": "1865998xxxx",
"uniqueid": "00D7C889-06A0-426E-BAB1-5741A1192038",
"title": "測試測試",
"summary": "豆豆豆",
"message": "12345",
"msgtype": "25",
"menuid": "203"
},
"Result": {
"errorno": "0"
}
}
示例如上面代碼所示，可以根據個人的業務需要進行調整。
發送請求
發送請求就很簡單了，用requests模塊，然后從JSON中讀取發送的參數，post、get或者其他。由于要生成測試報告，那么發送的數據需要做一下記錄，我選擇用txt文本來作為記錄的容器。
f = file("case.json")
testData = json.load(f)
f.close()
def sendData(testData, num):
payload = {}
# 從json中獲取發送參數
for x in testData[num]['InputArg'].items():
payload[x[0]] = x[1]
with open('leftside.txt', 'a+') as f:
f.write(testData[num]['TestId'])
f.write('-')
f.write(testData[num]['Title'])
f.write('\n')
# 發送請求
data = requests.get(testData[num]['Url'], params=payload)
r = data.json()
接受返回
由于我們是需要生成測試報告的，那么返回的數據我們先需要進行一次存儲，可以選擇用數據庫存儲，但是我覺得數據庫存儲太麻煩了，只要用txt文本作為存儲容器即可。
with open('rightside.txt', 'a+') as rs:
rs.write('發送數據')
rs.write('|')
rs.write('標題:'+testData[num]['Title'])
rs.write('|')
rs.write('發送方式:'+testData[num]['Method'])
rs.write('|')
rs.write('案例描述:'+testData[num]['Desc'])
rs.write('|')
rs.write('發送地址:'+testData[num]['Url'])
rs.write('|')
rs.write('發送參數:'+str(payload).decode("unicode-escape").encode("utf-8").replace("u\'","\'"))
rs.write('|')
rs.write(testData[num]['TestId'])
rs.write('\n')
結果判定
結果判定我使用的是全等于判定。因為我們的接口只需要這樣處理就行了，如果有需要，可以寫成正則判定。
with open('result.txt', 'a+') as rst:
rst.write('返回數據')
rst.write('|')
for x, y in r.items():
rst.write(' : '.join([x, y]))
rst.write('|')
# 寫測試結果
try:
if cmp(r, testData[num]['Result']) == 0:
rst.write('pass')
else:
rst.write('fail')
except Exception:
rst.write('no except result')
rst.write('\n')
我這里結果有3種，成功、失敗或者沒結果。結果的設置就看自己的定義了。
生成測試報告
測試報告是一個重頭戲，由于我發送數據、返回數據和結果都是用txt文本存儲，那么每次使用a+模式新增，會讓結果越來越多，而且檢查起來非常蛋疼。
我的處理方式是每次測試完畢之后，用Python讀取txt文本中的數據，然后使用Django動態生成一個結果，然后再使用requests抓取這個網頁，保存在Report文件夾中。
網頁報告
Django的方法我就不多說了，博客中已經有一整個系列文章了。我們需要在views文件中打開之前記錄的3個txt文件，然后做一些數據處理，返回給前端，前端用Bootstrap來渲染，就能生成一個比較漂亮的測試報告。
def index(request):
rightside = []
result = []
rst_data = []
leftside = []
passed = 0
fail = 0
noresult = 0
with open(os.getcwd() + '/PortTest/leftside.txt') as ls:
for x in ls.readlines():
lf_data = {
'code': x.strip().split('-')[0],
'title': x.strip().split('-')[1]
}
leftside.append(lf_data)
with open(os.getcwd() + '/PortTest/rightside.txt') as rs:
for x in rs.readlines():
row = x.strip().split('|')
rs_data = {
"fssj": row[0],
"csbt": row[1],
"fsfs": row[2],
"alms": row[3],
"fsdz": row[4],
"fscs": row[5],
'testid': row[6]
}
rightside.append(rs_data)
with open(os.getcwd() + '/PortTest/result.txt') as rst:
for x in rst.readlines():
row = x.strip().split('|')
if row[len(row)-1] == 'fail':
fail += 1
elif row[len(row)-1] == 'pass':
passed += 1
elif row[len(row)-1] == 'no except result':
noresult += 1
rs_data = []
for y in row:
rs_data.append(y)
result.append(rs_data)
for a, b in zip(rightside, result):
data = {
"sendData": a,
"dealData": b,
"result": b[len(b)-1]
}
rst_data.append(data)
return render(request, 'PortTest/index.html', {"leftside": leftside,
"rst_data": rst_data,
"pass": passed,
"fail": fail,
"noresult": noresult})
基本上都是一些很基礎的知識，字符串分割等等。這里的數據處理為了方便，在獲取數據存儲的時候就要按照一定的格式來存儲，views的方法就很容易做處理。
前端代碼如下：
Title
接口測試報告
Design By Sven
測試通過
{{ pass }}
測試失敗
{{ fail }}
無結果
{{ noresult }}
{% for ls in leftside %}
-
{{ ls.code }} - {{ ls.title }}
{% endfor %}
{{ x.result }}
{% for x in rst_data %}
{% if x.result == 'pass' %}
{% elif x.result == 'fail' %}
{% elif x.result == 'no except result' %}
{% endif %}
{{ x.sendData.fssj }}
{{ x.sendData.csbt }}
{{ x.sendData.fsfs }}
{{ x.sendData.alms }}
{{ x.sendData.fsdz }}
{{ x.sendData.fscs }}
{% for v in x.dealData %}
{{ v }}
{% endfor %}
{% endfor %}
測試報告效果圖
最后
用Python寫一個工具很容易，主要還是要能更方便地滿足實際工作中的使用需要為目的。如果要做完整的接口測試，還是盡量使用已經成熟的工具。
PS：簡單的造輪子也是學習原理的一個絕佳的方法。
以上這篇基于Python的接口測試框架實例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
