詳解資料驅動

傻姑娘家的李先生發表於2022-05-04

本章主要介紹pytest幾種資料驅動的方法，也是我們做介面自動化中經常要使用到的，大致分為以下四種。

 一.yaml

 二.json

 三.csv

 四.excel

 

 

一.yaml

1.安裝外掛：pip install pyyaml

2.yaml的兩種讀寫方法load()和dump()，話不多說，直接上程式碼

  load()為讀取json流，讀取是加上Loader=yaml.FullLoader，防止亂碼，self.data為讀取的yaml檔案

    def param_yaml_ready(self):
        with open(self.data, mode="r", encoding="utf-8") as f1:
            data = yaml.load(f1, Loader=yaml.FullLoader)
        return data

 dump()為寫入dict型別的資料，allow_unicode=True防止寫入時亂碼，

    def param_yaml_write(寫入dict型別的資料):
        with open("寫入的檔名.yaml", mode="w",
                  encoding="utf-8") as f1:
            data = yaml.dump(寫入dict型別的資料, f1, allow_unicode=True)
        return data

3.yaml檔案的寫法。

  yaml檔案中-程式碼中括號，list的意思，每個層級代表一個{}，dict的意思，下面內容讀取出來後展示如下

- class:
      name: 李明
      age: 12
讀取的內容：[{'class': {'name': '李明', 'age': 12}}]

4.例項。下面我們以一個具體的例項來說下yaml的用法

 （1）首先我們先建立一個tokenApproval.yaml檔案，具體寫法如下。這個yaml檔案中只有一條case

-
  #獲取稽核token
  tokenApproval:
    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken   #request-url
        param:     #request-body
          corpid: ww29957e72e534d2
          corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz
        methon: get    #request-methon
      response:
        eq:
          errcode: 0   #斷言判斷

  （2）然後建立params.py封裝yaml讀寫方法，./params/globalParametersToken.yaml是我寫入json資料的yaml檔案

    def param_yaml_ready(self):
        with open(self.data, mode="r", encoding="utf-8") as f1:
            data = yaml.load(f1, Loader=yaml.FullLoader)
        return data

    @staticmethod
    def param_yaml_write(paramYaml):
        with open("./params/globalParametersToken.yaml", mode="w",
                  encoding="utf-8") as f1:
            data = yaml.dump(paramYaml, f1, allow_unicode=True)
        return data

  （3） 結合pytest.mark.parametrize實現資料驅動，pytest.mark.parametrize具體寫法格式可查考：

            https://www.cnblogs.com/lihongtaoya/p/15840026.html

import pytest
from params.params import param
import requests


class Testtoken:
    @pytest.mark.parametrize("parama", param("./params/tokenApproval.yaml").param_yaml_ready())
    def test_tokenApproval(self, parama):
        url = parama["tokenApproval"][0]["request"]["url"]
        params = parama["tokenApproval"][0]["request"]["param"]
        data = requests.get(url, params)
        assert data.json()["errcode"] == parama["tokenApproval"][0]["response"]["eq"]["errcode"]

    若是yaml檔案中寫入的單介面資料不止一條case，如下程式碼，那麼parama["tokenApproval"][0]這樣的

    寫法就不適用了，這裡需要新增一個for迴圈即可for i in range(0, len(parama["tokenApproval"])):

-
  #獲取稽核token
  tokenApproval:
    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
        param:
          corpid: ww29957e72e534d2
          corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz
        methon: get
      response:
        eq:
          errcode: 0

    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
        param:
          corpid: ww29957e72e534d2
          corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjk
        methon: get
      response:
        eq:
          errcode: 0

   若yaml檔案中寫入的是多介面資料，如下程式碼寫法，這裡每個介面我們只要有一個關鍵字去區分即可：

    tokenApproval  ||  tokenContacts，這樣我們在獲取的時候parama["tokenApproval"]，tokenApproval

    這個值隨機變換即可

 -
  #獲取稽核token
  tokenApproval:
    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
        param:
          corpid: ww29957e72e534d27e
          corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc
        methon: get
      response:
        eq:
          errcode: 0

    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
        param:
          corpid: ww29957e72e534d27e
          corpsecret: rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknz
        methon: get
      response:
        eq:
          errcode: 0
  #獲取通訊錄token
  tokenContacts:
    - request:
        url: https://qyapi.weixin.qq.com/cgi-bin/gettoken
        param:
          corpid: ww29957e72e534d27e
          corpsecret: hUQCww5PR0o_QHF4HBt7lLQlVf3M5Amssx6OQZxUkPY
        methon: get
      response:
        eq:
          errcode: 0




   （4）dump()用法，先獲取相應的結果，已鍵值對的格式新增到list中，最後呼叫封裝好的寫入方法即可。




 datalist = [{"tokenApproval": data.json()["token"]}]
            param.param_write_csv(datalist)




 


 


二.json


1.json讀寫的兩種方法和yaml一致load()和dump()




        with open("讀取的json檔案", mode="r", encoding="utf-8") as p:
            data = json.load(p)
     
        with open("寫入資料的json檔案", mode="w", encoding="utf-8")as p:
            json.dump(寫入的資料data, p)





2.剩餘部分json的操作方法和yaml的大致相同，唯一不同的區別就是json檔案的寫法。在實際操作中yaml中寫


   case會比json方便一點，所以我們可以在yaml寫完後，通過第三方工具轉換json格式，這樣方便的很多：


   https://json.im/json2/yaml2json.html。轉換過後獲取資料和yaml方法相同


 


 


三.csv


1.csv兩種讀寫的方法reader()和write()


   下圖程式碼，csv.reader(file)返回的是一個可迭代的結果集，最後通過for迴圈遍歷出來新增到list中。writerow()


    為寫入一個列表，也就是一行資料，如果要寫入多條資料的話可使用writerows()方法，寫法writes.writerows([list1,list2,list3........])




        with open("讀取的csv檔案", mode="r", encoding="utf-8")as f:
            data = csv.reader(f)
            dataparam=[]
            for lis in data:
                dataparam.append(lis)

        with open("寫入的csv檔案", mode="w", encoding="utf-8")as f:
            writes = csv.writer(f)
            writes.writerow("寫入的list")





2.csv檔案寫法


   首先我們在建立csv檔案後開啟會在右上角看見一個鍵盤似地圖示，點選後選擇csv後點選確定，這個時候在改


   檔案左下角有兩個tab，text and data


   


  text寫法類似文字，每個元素中間用逗號隔開，而data則是表格的寫法。這裡需要注意的是如果你寫入的是鍵值


  對格式，需要嚴格按照格式來寫，如下02例子




01
name,age,class
李明,12,三年級二班

02

class,name


三年級二班,"{""name"":""小明"",""age"":""12""}"






三.例項。下面我們以一個具體的例項來說下csv的用法


   （1）首先我們建立一個tokencsv.csv檔案，具體寫法如下




tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d2"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0





　（2）來封裝我們所需要的讀寫方法




    def param_reader_csv(self):
        with open(self.data, mode="r", encoding="utf-8")as f:
            data = csv.reader(f)
            dataparam=[]
            for lis in data:
                dataparam.append(lis)
            return dataparam

    @staticmethod
    def param_write_csv(data):
        with open("./params/globalParametersToken.csv", mode="w", encoding="utf-8")as f:
            writes = csv.writer(f)
            writes.writerow(data)





　（3）結合pytest.mark.parametrize實現資料驅動，parametrize中添寫的變數數量一定要和csv檔案中的數


            量對齊，否則會報錯。另外這裡讀取csv檔案中的內容都是以str 型別輸出的，所以當涉及到獲取鍵值


            對資料時，即下方parama入參值，需使用json.loads()方法將其轉成dict，




 @pytest.mark.parametrize("typetoken,methon,url,parama,errcode", param("./params/tokenCsv.csv").param_reader_csv())
    def test_tokenApproval(self, typetoken, methon, url, parama, errcode):
            data = method(json.loads(parama), url).request_get()
            if data.json()["errcode"] == int(errcode):
                tokenList = [data.json()["access_token"]]
                param.param_write_csv(tokenList)
            else:
                with open("./log/error.log", mode="a+", encoding="utf-8")as f:
                    f.write(
                        f"[{datetime.datetime.now()}]:獲取稽核token失敗，response：{data.json()},request-body:{data.request.body},request-url:{data.request.url}\n")





　　當csv檔案中寫入單介面多條case時，無需過多處理，按照以上程式碼即可。當csv檔案中寫入多介面多條


        case時，我們可以在csv檔案中寫入一個變數來區分這條case是哪條介面的，即上面程式碼typetoken引數。


        cvs檔案中首個資料tokenApproval和tokenContacts可代表不同的介面，我們在處理資料時加個if判斷即可


        (這種方法不太能接受，所以csv做資料驅動時儘量一個介面寫 一個csv檔案)




tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27e"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0
tokenApproval,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27"",""corpsecret"":""rVP15c62Ybjl9xTJIbCCXwoDq8eVuTQ-9FOmqjknzBc""}",0
tokenContacts,get,https://qyapi.weixin.qq.com/cgi-bin/gettoken,"{""corpid"":""ww29957e72e534d27e"",""corpsecret"":""hUQCww5PR0o_QHF4HBt7lLQlVf3M5Amssx6OQZxUkPY""}",0




 


 


四.excel


表格的操作太過複雜，能不用就不用吧，這裡就簡單展示的我封裝讀取excel的方法吧




    def param_excel(self):
        file = openpyxl.load_workbook(self.data)
        file_name = file.active
        data = []
        for x in range(1, file_name.max_row + 1):
            data1 = []
            for y in range(1, file_name.max_column + 1):
                data1.append(file_name.cell(x, y).value)
            data2 = tuple(data1)
        data.append(data2)
        return data





　　


 


 


 


 


 


 


 


 


 


 完.....................






相關文章