python+jinja2實現(xiàn)接口數(shù)據(jù)批量生成工具

在做接口測試的時候,我們經(jīng)常會遇到一種情況就是要對接口的參數(shù)進行各種可能的校驗,手動修改很麻煩,尤其是那些接口參數(shù)有幾十個甚至更多的,有沒有一種方法可以批量的對指定參數(shù)做生成處理呢。

答案是肯定的!

python的jinja2模板庫可以很好的滿足我們的需求,通過維護一個原始數(shù)據(jù)模板,將我們想要動態(tài)生成的變量模板化,就可以實現(xiàn)需求。


現(xiàn)在我們有這樣的一個請求數(shù)據(jù)

{
    "abc":"123",
    "p2p":"123",
    "smid":"20180807220733939b66d80092eea34ce9e77f30bedff12345b7d5a3faa11b",
    "test":{
        "test1":"1",
        "test2":"2"
    },
    "test3":"3"
}

如果想對其中的smid字段進行批量修改并生成新的請求數(shù)據(jù)就可以進行如下操作:

1、首先新建一個名為fp_template.txt的文本文件;

2、將上面的接口請求數(shù)據(jù)復制粘貼進這個txt文件中,我們以此作為“模板文件”;

3、新建一個predata文件夾用來存放生成后的數(shù)據(jù)文件;

4、對smid字段進行模板化(模板格式可以參考jinja2的語法,這里不做贅述),于是上面的請求參數(shù)就變成了下面這樣:

{
    "abc":"123",
    "p2p":"123",
    "smid":"{{ smid }}",
    "test":{
        "test1":"1",
        "test2":"2"
    },
    "test3":"3"
}

實現(xiàn)代碼代碼如下:

# -*- coding: UTF-8 -*-
from jinja2 import Environment,FileSystemLoader
import os

class DataTemplateFaker:
    def __init__(self):
        self.aesPath = os.getcwd()#獲取啟動路徑
        self.resultPath = self.aesPath + "/predata/"#指定用來保存生成數(shù)據(jù)的路徑
        self.templateFile = "fp_template.txt"

    #修改我們要批量生成smid的格式
    def init_smid(self,start,end):
        smidArg = [x for x in range(start, end)]
        re = []
        for n in smidArg:
            re.append("20180807220733939b66d80092eea34ce9e77f30bedff" + str(n) + "b7d5a3faa11b")
        return re

    #操作模板文件
    def preContent(self,arg):
        env = Environment(loader=FileSystemLoader('./'))
        tpl = env.get_template(self.templateFile)
        renderContent = tpl.render(smid=arg)
        return renderContent

    #通過修改的smid列表批量替換模板文件并寫入指定文件中
    def makeContent(self,preList):
        x = 0
        for i in preList:
            x = x + 1
            filename = str(self.resultPath) + 'data_' + str(x) + '.txt' #用以區(qū)分存放新生成的請求數(shù)據(jù)(也可以寫到一個文件中)
            renderContent = self.preContent(i)
            with open(filename, 'w') as f:
                f.writelines(renderContent)
                f.close()

if __name__ == "__main__":
    AT = DataTemplateFaker()
    reList = AT.init_smid(1,10)#控制生成數(shù)據(jù)的范圍
    AT.makeContent(reList)

運行程序,就能得到新生成的數(shù)據(jù)

當然,我們也可以對其他的參數(shù)進行指定修改,如修改p2p,只需要修改模板文件:

{
    "abc":"123",
    "p2p":"{{ p2p }}",
    "smid":"20180807220733939b66d80092eea34ce9e77f30bedff12345b7d5a3faa11b",
    "test":{
        "test1":"1",
        "test2":"2"
    },
    "test3":"3"
}

然后在代碼中加入一個方法init_p2p()

# -*- coding: UTF-8 -*-
from jinja2 import Environment,FileSystemLoader
import os

class DataTemplateFaker:
    def __init__(self):
        self.aesPath = os.getcwd()#獲取啟動路徑
        self.resultPath = self.aesPath + "/predata/"#指定用來保存生成數(shù)據(jù)的路徑
        self.templateFile = "fp_template.txt"

    #修改我們要批量生成smid的格式
    def init_smid(self,start,end):
        smidArg = [x for x in range(start, end)]
        re = []
        for n in smidArg:
            re.append("20180807220733939b66d80092eea34ce9e77f30bedff" + str(n) + "b7d5a3faa11b")
        return re
    #修改我們要批量生成p2p的格式   
    def init_p2p(self,start,end):
        p2pArg = [x for x in range(start, end)]
        return p2pArg

    #操作模板文件
    def preContent(self,arg):
        env = Environment(loader=FileSystemLoader('./'))
        tpl = env.get_template(self.templateFile)
        renderContent = tpl.render(smid=arg)
        return renderContent

    #通過修改的smid列表批量替換模板文件并寫入指定文件中
    def makeContent(self,preList):
        x = 0
        for i in preList:
            x = x + 1
            filename = str(self.resultPath) + 'data_' + str(x) + '.txt' #用以區(qū)分存放新生成的請求數(shù)據(jù)(也可以寫到一個文件中)
            renderContent = self.preContent(i)
            with open(filename, 'w') as f:
                f.writelines(renderContent)
                f.close()

if __name__ == "__main__":
    AT = DataTemplateFaker()
    reList = AT.init_p2p(1,10)#控制生成數(shù)據(jù)的范圍
    AT.makeContent(reList)

這只是一個很簡單的demo,當然還有很多可優(yōu)化的地方,比如多字段同時修改、引入faker庫進行關(guān)聯(lián)生成偽造數(shù)據(jù)等,越是復雜且參數(shù)繁多的接口越適用,其他的方法就可以天馬行空,任君發(fā)揮了。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

推薦閱讀更多精彩內(nèi)容