Python unittest+ddt+openpyxl+configparser

1.技術介紹

框架：unittest

請求處理：requests

excel資料處理：openpyxl

引數化：ddt

配置決議器：configparser

報告模板：HTMLTestRunnerNew.py(下載地址：https://pan.baidu.com/s/1w9AZU9AkIpxCYuzTto0EQA?pwd=1234)

testdemo.xlsx：測驗資料，注意：implement=y為執行資料，否則不執行

2.代碼結構

----->case.config　　組態檔
----->DoCase.py　　測驗用例類
----->DoConfig　　組態檔處理
----->DoExcel.py　　Excel檔案處理類，篩選有效資料回傳 
----->DoHttp.py　　 http請求處理類 
----->HttpRunner　　unittest加載用例執行生成報告 
----->report.html　　 執行后生成的報告 
----->HTMLTestRunnerNew.py　　測驗報告模板

3.目錄介紹

4.excel資料

5.case.config

[CASE]
caseid=[1,3]

6.Docase.py

#測驗用例類
import unittest
from test02.DoHttp import Request_Http#引入請求類
from ddt import ddt,data,unpack#引入ddt做引數化
from test02.DoExcel import GetExcelData
data_list = GetExcelData("C:\\Users\\Administrator\\Desktop\\testdemo.xlsx","s1").get_data()
#用例類
@ddt#裝飾器裝飾類
class HttpCase(unittest.TestCase):#繼承unittest測驗用例類
    @data(*data_list)#脫外套
    def test_login_yes(self,item):
        res = Request_Http(item["method"], item["url"], item["data"]).request_http()
        try:
            self.assertEqual(item["expect"],res.json()["code"])#斷言：預期與實際是否相等
        except Exception as e:
            print("斷言例外：{0}".format(e))
            raise e#拋出例外

7.DoConfig.py

import configparser

class GetConfig():
    def get_config_data(self,file,section,option):
        cf = configparser.ConfigParser()
        cf.read(file, encoding="utf8")  # 讀取config，有中文注意編碼
        # 回傳value
        return cf[section][option]

8.DoExcel.py

from openpyxl import load_workbook
from test02.DoConfig import GetConfig
class GetExcelData():
    def __init__(self,file,sheet):
        self.file=file
        self.sheet=sheet

    def get_data(self):
        caseid = eval(GetConfig().get_config_data("case.config","CASE","caseid"))#處理組態檔獲取要執行的caseid
        wb = load_workbook(self.file)#打開excel
        sheet = wb[self.sheet]#定位作業簿
        data_list = []
        for n in range(2,sheet.max_row+1):
            data_dic = {}
            for m in range(2,sheet.max_column+1):
                data_dic["case_id"] = sheet.cell(n, 1).value
                data_dic["module"] = sheet.cell(n, 2).value
                data_dic["title"] = sheet.cell(n, 3).value
                data_dic["method"]=sheet.cell(n,4).value
                data_dic["url"]=sheet.cell(n,5).value
                data_dic["data"]=eval(sheet.cell(n, 6).value)#還原資料型別
                data_dic["expect"]=sheet.cell(n, 7).value
            if data_dic["case_id"] in caseid:#判斷case_id是否包含在組態檔中
                data_list.append(data_dic)
        return data_list

5.DoHttp.py

#http請求處理類
import requests
class Request_Http():
    def __init__(self,method,url,data,expected=None,headers=None,cookie=None):
        self.method=method.lower()#請求方式轉小寫
        self.url=url
        self.data=data
        self.excepted=expected
        self.headers=headers
        self.cookie=cookie

    def request_http(self):
        if self.method=="get":
            try:
                return requests.get(self.url, params=self.data, headers=self.headers, cookies=self.cookie)
            except Exception as e:
                print("例外請求：{0}".format(e))
                raise e  # 拋出例外
        else:
            try:
                return requests.post(self.url,params=self.data,headers=self.headers,cookies=self.cookie)
            except Exception as e:
                print("例外請求：{0}".format(e))
                raise e#拋出例外

6.HttpRunner.py

#unittest加載用例執行生成報告
import unittest
from test02 import DoCase
import HTMLTestRunnerNew
suite = unittest.TestSuite()#存放測驗用例

loader = unittest.TestLoader()#加載器
suite.addTest(loader.loadTestsFromModule(DoCase))#加載測驗模塊

#背景關系管理器
with open("report.html","wb") as file:
    runner = HTMLTestRunnerNew.HTMLTestRunner(stream=file, verbosity=2,title="標題",description="備注",tester="姓名")
    runner.run(suite)

7.report.html

8.報告模板下載

HTMLTestRunnerNew.py(下載地址：https://pan.baidu.com/s/1w9AZU9AkIpxCYuzTto0EQA?pwd=1234)

標籤：Python

上一篇：pandas替換，加載，透視表

下一篇：matplotlib畫圖基礎知識