python3+requests+unittest介面自動化例項講解

摘要： 前言 這個季度初的時候，我們公司管理層提出了全面提升部門技術水準的要求，而作為專案開發流程中最重要的一環——測試，我們也是有自己的技術追求的，所以為了，由原來的業務功能測試轉變成自動化測試。之前大多都是手工進行，利用postman和fiddler進行抓包和擷取資料，後來，經過大家的討論...

前言

這個季度初的時候，我們公司管理層提出了全面提升部門技術水準的要求，而作為專案開發流程中最重要的一環——測試，我們也是有自己的技術追求的，所以為了，由原來的業務功能測試轉變成自動化測試。之前大多都是手工進行，利用postman和fiddler進行抓包和擷取資料，後來，經過大家的討論，決定基於python學習，開始搞介面自動化，經過大家的努力，這套自動化框架目前已經基本完成了，於是進行一些總結，便於以後回顧溫習，有許多不完善的地方，大家可以留言一起討論研究，共同進步。

首先，我們可以捋一捋思路，想一想介面測試的流程

其次，選擇合適的框架

流程清晰之後，我們需要選擇一個合適的框架，於是選擇了Python3+requests+unittest框架

Requests模組傳送http的網路請求，請求型別主要包含了post，get， PUT，DELETE，HEAD

python+unittest單元測試框架構成，和測試報告生成（HTMLTestRunner）

框架詳解不在此贅述，requests可參考點選前往

unittest參考點選前往

基於以上，我們來一步步搭建我們的框架。在這個過程中，我們需要做到業務和資料的分離，這樣才能靈活，達到我們寫框架的目的。接下來，我們來進行結構的劃分。

我們的結構是這樣的

data：存放自動化測試所用到的資料文件

log：存放生成的日誌檔案

base：存放公共的方法

report：存放生成的自動化測試報告

testcase：存放測試指令碼

接下來，公共方法的開發

整體結構有了劃分，接下來就該一步步的填充整個框架了，我們先來看看Base檔案中的公共類或函式，這主要用於後續測試case的呼叫，所有公共的、一成不變的資料都可以放在這裡，維護也方便

配置文件如下

[DATABASE]
data_address = ./data/
report_address = ./report/

[HTTP]
base_url = http://xxx.xx

複製程式碼複製程式碼

想知道怎樣從配置文件中得到或寫入相應的資料嗎？那繼續看吧

import os
import configparser

# 獲取當前py檔案地址
proDir = os.path.split(os.path.realpath(__file__))[0]
# 組合config檔案地址
configPath = os.path.join(proDir,"config.ini")

class ReadConfig:
def __init__(self):
#獲取當前路徑下的配置檔案
self.cf = configparser.ConfigParser()
self.cf.read(configPath)

def get_config(self,field,key):
#獲取配置檔案中的key值
result = self.cf.get(field,key)
return result

def set_config(self,field,key,value):
#向配置檔案中寫入配置資訊
fb = open(configPath,'w')
self.cf.set(field,key,value)
self.cf.write(fb)

複製程式碼複製程式碼

那問題又來了，我們的測試資料放在哪裡？怎麼取值？怎麼寫入？怎麼儲存？。。。

別急，接著往下看

測試資料優先考慮放在excel或database中，此處以excel為例做個簡單介紹

這裡需要用到兩個操作表格的庫，xlrd資料驅動的讀取，作用於excel文件，但xlrd不能寫入資料，所以引入xlutils資料驅動的讀取和寫入

安裝方法可以用pip3 install xlrd和pip3 install xlutils 來安裝。

import xlrd
import xlutils.copy
from Base.readConfig import ReadConfig
import time

class ReadExcel:

def __init__(self,section,field,sheet):
# 開啟工作表，並定位到sheet
data_address = ReadConfig().get_config(section,field)
workbook = xlrd.open_workbook(data_address)
self.table = workbook.sheets()[sheet]


def get_rows(self):
# 獲取excel行數
rows = self.table.nrows
return rows

def get_cell(self,row,col):
# 獲取單元格資料
cell_data = self.table.cell(row,col).value
return cell_data

def get_col(self,col):
# 獲取整列資料
col_data = self.table.col_value(col)
return col_data

class WriteExcel:
def __init__(self,section,field,sheet):
# 開啟工作表
self.address = ReadConfig().get_config(section,field)
self.workbook = xlrd.open_workbook(self.address)
self.wf = xlutils.copy.copy(self.workbook)
self.ws = self.wf.get_sheet(sheet)

def set_cell(self,row,col,value):
#設定單元格資料
self.ws.write(row,col,value)

def save_excel(self,filename,format):
#獲取當前時間
self.time = time.strftime("%Y%m%d%H%M%S", time.localtime())
#生成檔案的檔名及格式
self.report = filename + '_' +self.time + format
#儲存檔案
self.wf.save(self.report)


複製程式碼複製程式碼

然後，測試指令碼的編輯

一切準備就緒，下面通過一個簡單的、完整的程式碼進行演示公共函式的呼叫，框架的使用及報告的生成

import unittest
import requests
from Base.readConfig import ReadConfig
from Base.readExcel import ReadExcel
from Base.readExcel import WriteExcel
#例項化
readexcel = ReadExcel('DATABASE','data_address',0)
writeexcel = WriteExcel('DATABASE','data_address',0)

class testcase(unittest.TestCase):
#初始化
def setUp(self):
#獲取url
self.base_url = ReadConfig().get_config('HTTP', 'base_url')
self.url = self.base_url + readexcel.get_cell(1,1)
#獲取請求頭
self.headers = readexcel.get_cell(1,4)

def test_case(self):
nok = 0
ner = 0
# 迴圈讀取excel中的測試資料，進行結果驗證，並生成excel形式的測試報告
for i in range(3,readexcel.get_rows()):
#傳送網路請求，得到響應值
response = requests.post(self.url, headers=self.headers,data=readexcel.get_cell(i,4).encode('utf-8'))
actualresult = response.json()
#獲取excel中的預期結果
expectresult = eval(readexcel.get_cell(i,6))
# 獲取需驗證的資料
key = eval(readexcel.get_cell(i, 5))
keylen = len(key)
j = 0
for k in range(keylen):
aresult = 'actualresult' + key[k]
eresult = 'expectresult' + key[k]
if eval(aresult) == eval(eresult):
#預期結果和實際結果一致
j = j + 1
if j == keylen:
#測試資料執行通過
nok = nok + 1
writeexcel.set_cell(i, 8, 'SUCCESS')
else:
# 測試資料執行失敗，並將實際結果寫入excel
ner = ner + 1
writeexcel.set_cell(i, 8, 'FAILURE')
writeexcel.set_cell(i, 7, str(actualresult))
print('第', i + 1, '行用例執行失敗：預期結果是', expectresult, '實際結果是', actualresult)
# 儲存測試報告
writeexcel.save_excel('testreport', '.xls')
print('測試資料中總共', nok, '條用例執行通過', ner, '條用例執行失敗')

#釋放資源
def tearDown(self):
pass

if __name__ == '__main__':
#構造測試集合
suite = unittest.TestSuite()
suite.addTest(testcase('test_case'))
#建立html檔案
filename = ReadConfig().get_config('DATABASE', 'report_address') + 'testreport.html'
fb = open(filename,"wb")
#執行測試並生成html測試報告
runner = HTMLTestRunner.HTMLTestRunner(stream = fb,description = '針對介面的描述資訊',title = '某某的自動化測試報告')
runner.run(suite)
#關閉檔案
fb.close()

複製程式碼複製程式碼

最後，生成的html報告

以上是最近總結的部分成果，希望對大家有幫助，後續會陸續更新，也請大家多多留言，互相交流互相成長