前言：

文章是對(duì)上一篇接口自動(dòng)化做的部分優(yōu)化，由于之前的代碼將全部精力投放在單接口如何執(zhí)行，如何讀取，如何循環(huán)傳參上，所以導(dǎo)致使用起來(lái)仍然比較重，于是就有了本次優(yōu)化的文章了

相信大家如果看過(guò)之前文章的測(cè)工應(yīng)該知道 *parameterized *這個(gè)裝飾器的實(shí)用性，這個(gè)裝飾器可以將一個(gè)列表內(nèi)的所有集合生成等同于列表長(zhǎng)度的方法個(gè)數(shù)，來(lái)實(shí)現(xiàn)循環(huán)傳參斷言，（ps:這里講一下為什么要?jiǎng)?chuàng)建測(cè)試用例個(gè)數(shù)的方法，因?yàn)閡nittest庫(kù)中的assertEquals方法在碰到斷言不通過(guò)時(shí)，整個(gè)方法就直接return出來(lái)了，如果在一個(gè)測(cè)試方法中寫(xiě)for循環(huán)來(lái)進(jìn)行循環(huán)測(cè)試，這種做法顯然不符合要求，例如我有三條case要走，執(zhí)行到第二個(gè)拋個(gè)異常，這樣后面的方法無(wú)法被執(zhí)行到）

如果仍然不是很理解parameterized方法的同學(xué)，我就再講下吧

image

上圖是一個(gè)普通的加法判斷方法，有兩條case，case1是判斷2+3 = 5 case2是判斷2+3=6

如果判斷失敗則拋異常，并返回結(jié)果：fail

這樣該方法生成的測(cè)試方法為2個(gè)，但是按照之前的設(shè)計(jì)思路，我一個(gè)接口有多條測(cè)試用例，使用一個(gè)測(cè)試方法，則可以測(cè)試這個(gè)接口的多條用例，但是我的斷言結(jié)果都是前置處理之后，將結(jié)果拿進(jìn)測(cè)試方法中做比對(duì)的，（這句話(huà)的意思是，我的測(cè)試方法中全是相同的代碼，唯一不同的是，調(diào)用public_api文件時(shí)不一樣而已），所以我得拼命去復(fù)制相同的代碼，去命名不同的測(cè)試方法，這樣我當(dāng)時(shí)糾結(jié)了很久，這種寫(xiě)法是不是太傻了，

于是在一個(gè)晚上，我想來(lái)想去（主要是有個(gè)在這邊工作的同學(xué)要離開(kāi)上海，臨走在我這住了一晚，整完的呼嚕聲吵得我無(wú)法入睡），于是想到了一個(gè)辦法，我決定摒棄掉parameterize裝飾器，

我想到了我的第二份工作時(shí)寫(xiě)的接口測(cè)試代碼，

# -*- coding: utf-8 -*-
import unittest
import requests
import time
import json
import random
from sql_tool import sqlconnect
import csv
url = r'C:\test_document\api_unittest\test_data'
class landlordApplyCreditToAccount(unittest.TestCase):
    def setUp(self):
        self.base_url = r"http://121.40.178.164:7171/settlement-center-service/api/accountTransaction/landlordApplyCreditToAccount.json"
        self.headers = {"Content-type":"application/x-www-form-urlencoded"}
        self.timestamp = str(int(time.time()))
    
    def t_landlord(self,arg1,arg2,arg3):
        self._user = sqlconnect.model.execQuery("SELECT id FROM member_info WHERE accountOpenId = '"+arg2+"'")
        self._Amount = sqlconnect.model.execQuery("SELECT realAmount FROM tally_account_info WHERE memberInfoId = '"+self._user[0][0]+"'")
        value = json.dumps({"name":arg1})
        data = {
            "projectNo":"ZG",
            "projectOrderNo":"YJSH0000130021198038a28b50f5907bcnew4691585a3a"+str(random.randint(1000,9999))+"c"+self.timestamp,
            "description":value,
                "accountOpenId":arg2,
                "amountABS":arg3,
                
                        }
        #json_request = requests.post(url=self.base_url,headers=self.headers,data=data)
        #print json_request.json()['res']['msg']
        self.Amount = sqlconnect.model.execQuery("SELECT realAmount FROM tally_account_info WHERE memberInfoId = '"+self._user[0][0]+"'")
        if self.Amount[0][0] < data["amountABS"]:
            print "余額不足請(qǐng)，請(qǐng)?zhí)岈F(xiàn)接口無(wú)效"
        else:
            print self.Amount[0][0],self._Amount[0][0]
            self.assertEquals(self.Amount[0][0],self._Amount[0][0] + data["amountABS"],"請(qǐng)求接口結(jié)果錯(cuò)誤")
    @staticmethod
    def getTestFunc(arg1,arg2,arg3):
        def func(self):
            self.t_landlord(arg1, arg2,arg3)
        return func
            
    def tearDown(self):
        pass
        #sqlconnect.model.execDml("UPDATE tally_account_info SET realAmount = 1000 WHERE memberInfoId = '"+self._user[0][0]+"'")
def __generateTestCases():
    arglists = []
    with open(url+r'\landlordApplyCreditToAccount.csv') as f:
        reader = csv.reader(f)
        for i in reader:    
            arglists.append(i)
    for args in arglists:
        setattr(landlordApplyCreditToAccount,'test_landlord_%s'%(args[1]),landlordApplyCreditToAccount.getTestFunc(*args))
__generateTestCases()
if __name__ == "__main__":
    unittest.main()

我這里就不做敏感處理了，這些IP估計(jì)現(xiàn)在都ping不通了，上面這段代碼中大家可以關(guān)注兩個(gè)函數(shù)，
一個(gè)閉包的靜態(tài)方法，這個(gè)方法的作用執(zhí)行t_landlord并返回一個(gè)函數(shù)地址

@staticmethod
    def getTestFunc(arg1,arg2,arg3):
        def func(self):
            self.t_landlord(arg1, arg2,arg3)
        return func

一個(gè)生成測(cè)試方法的函數(shù)

def __generateTestCases():
    arglists = []
    with open(url+r'\landlordApplyCreditToAccount.csv') as f:
        reader = csv.reader(f)
        for i in reader:    
            arglists.append(i)
    for args in arglists:
        setattr(landlordApplyCreditToAccount,'test_landlord_%s'%(args[1]),landlordApplyCreditToAccount.getTestFunc(*args))

如果有過(guò)基礎(chǔ)的同學(xué)應(yīng)該沒(méi)問(wèn)題，但是我還是簡(jiǎn)單解釋一下：

image.png

上圖中Test類(lèi)中時(shí)什么變量都沒(méi)有的，在經(jīng)過(guò)setattr加工處理過(guò)后，就可以將變量name給到Test類(lèi)，于是我們將Test類(lèi)想做是我們被測(cè)類(lèi)，name想做是類(lèi)中的測(cè)試方法(其實(shí)變量和方法一樣都指向一個(gè)內(nèi)存地址)于是就有了這段代碼：
setattr(landlordApplyCreditToAccount,'test_landlord_%s'(args[1]),landlordApplyCreditToAccount.getTestFunc(*args))
第一個(gè)參數(shù)，類(lèi)名稱(chēng)，第二個(gè)參數(shù)類(lèi)中的方法，第三個(gè)方法中具體的實(shí)現(xiàn)
經(jīng)過(guò)這樣處理之后，就可以生成多個(gè)測(cè)試方法來(lái)執(zhí)行測(cè)試用例，由于上面代碼時(shí)幾年前的代碼了，所以當(dāng)然不能這樣寫(xiě)，結(jié)合我們已實(shí)現(xiàn)的測(cè)試代碼，我做了如下調(diào)整：

def generate_test_cases(case_url,class_name,debug=True,file_name=None):
    '''文件列表'''
    for root, dirs, files in os.walk(case_url):
        file_list = files
    '''將文件夾的列表放入列表內(nèi)'''
    if debug:
        file_url = os.path.join(case_url, file_name)
        test_data = getJsonData(file_url)
        for i in range(len(test_data)):
            setattr(class_name, 'test_dispatch_%s' % (file_name[:-4] + str(i)), class_name.getTestFunc(test_data[i][0]))
        print('------生成結(jié)束')
    else:
        for file in file_list:
            '''循環(huán)創(chuàng)建測(cè)試方法，test_dispatch_文件名，arg為文件路徑'''
            file_url = os.path.join(case_url,file)
            test_data = getJsonData(file_url)
            for i in range(len(test_data)):
                setattr(class_name, 'test_dispatch_%s' % (file[:-4]+str(i)), class_name.getTestFunc(test_data[i]))
            print('------生成結(jié)束')

簡(jiǎn)而言之：就是運(yùn)行時(shí)動(dòng)態(tài)生成被測(cè)方法，以上就是優(yōu)化部分，
其他細(xì)節(jié)部分，我就不再贅述，如果有需要解釋的地方可以留言，我再單獨(dú)開(kāi)篇文章繼續(xù)詳解，

代碼：https://github.com/ZhangWei55kai/api_testV3.1