基于Python的接口測試框架實例

2019-11-25 16:30:08

字體：大中小

來源：轉載

供稿：網友

背景

最近公司在做消息推送，那么自然就會產生很多接口，測試的過程中需要調用接口，我就突然覺得是不是可以自己寫一個測試框架？

說干就干，由于現有的接口測試工具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的方法就很容易做處理。

前端代碼如下：

<!DOCTYPE html><html lang="en"><head>  <meta charset="UTF-8">  <title>Title</title>  <link  rel="stylesheet">  <script src="http://VeVB.COm/jquery/2.0.0/jquery.min.js"></script>  <script src="http://VeVB.COm/bootstrap/3.0.3/js/bootstrap.min.js"></script></head><body><div class="container">  <div class="row">    <div class="page-header">      <h1>接口測試報告        <small>Design By Sven</small>      </h1>    </div>  </div>  <div class="row">    <div class="col-md-4">      <h3>測試通過 <span class="label label-success">{{ pass }}</span></h3>    </div>    <div class="col-md-4">      <h3>測試失敗 <span class="label label-danger">{{ fail }}</span></h3>    </div>    <div class="col-md-4">      <h3>無結果 <span class="label label-warning">{{ noresult }}</span></h3>    </div>  </div>  <p></p>  <div class="row">    <div class="col-md-3">      <ul class="list-group">        {% for ls in leftside %}          <li class="list-group-item"><a href="#{{ ls.code }}">{{ ls.code }} - {{ ls.title }}</a></li>        {% endfor %}      </ul>    </div>    <div class="col-md-9">      {{ x.result }}      {% for x in rst_data %}        <div class="panel-group" id="accordion">        {% if x.result == 'pass' %}          <div class="panel panel-success">        {% elif x.result == 'fail' %}          <div class="panel panel-danger">        {% elif x.result == 'no except result' %}          <div class="panel panel-warning">        {% endif %}      <div class="panel-heading">        <h4 class="panel-title">          <a data-toggle="collapse" href="#{{ x.sendData.testid }}">            {{ x.sendData.testid }} - {{ x.sendData.csbt }}          </a>        </h4>      </div>      <div id="{{ x.sendData.testid }}" class="panel-collapse collapse">        <div class="panel-body">          <b>{{ x.sendData.fssj }}</b><br>          {{ x.sendData.csbt }}<br>          {{ x.sendData.fsfs }}<br>          {{ x.sendData.alms }}<br>          {{ x.sendData.fsdz }}<br>          {{ x.sendData.fscs }}          <hr>          {% for v in x.dealData %}            {{ v }}<br>          {% endfor %}        </div>      </div>      </div>      </div>        <p></p>      {% endfor %}      </div>      </div>    </div>    <script>      $(function () {        $(window).scroll(function () {          if ($(this).scrollTop() != 0) {            $("#toTop").fadeIn();          } else {            $("#toTop").fadeOut();          }        });        $("body").append("<div id=/"toTop/" style=/"border:1px solid #444;background:#333;color:#fff;text-align:center;padding:10px 13px 7px 13px;position:fixed;bottom:10px;right:10px;cursor:pointer;display:none;font-family:verdana;font-size:22px;/">^</div>");        $("#toTop").click(function () {          $("body,html").animate({scrollTop: 0}, 800);        });      });    </script></body></html>

測試報告效果圖

測試報告