Python+Excel+Unittest+HTMLTestRunner实现数据驱动接口自动化测试（二）

因为小白，这2天研究了好久才算是搞好。先附上一个测试完成后邮件的截图：

上一篇有提到：

unittest中实际运行了一个接口的很多条用例，而报告中只会有一条记录。这是因为unittest test case中我只写了一个test方法，循环测试数据。这样写用例更简单，目前在想是从报告中处理，还是测试用例读取部分进行处理。

而且这个问题会导致后续写test case出错几率大，而且耗时较多。经过几天的思考，写了个简易的解决办法。思路：轻微修改excel读取通过excel名/sheet名/用例id名来返回数据，那我们就可以在一个unittest case中固化test_开头的测试方法，如10条（我们约定excel中每个sheet最多写10条用例）。 这里我们在excel中添加一个“是否执行用例”字段，unittest中启动skipIf方法来确认是否跳过该用例，即可完美解决上一篇的问题。

excel用例数据截图：（增加runselect字段 YES代表执行用例，No代表不执行）

那我们的OperateExcel方法也做轻微修改：

def Readexcelrow_of_sheet(filename,sheet_name,row):
    '''返回某个sheet中某一行数据的列表'''
    folder = "C:\\Users\\LENOVO\\PycharmProjects\\Apiautotest-master\\testdata\\"
    fullfilename = folder + filename
    rb = xlrd.open_workbook(fullfilename)
    sheet = rb.sheet_by_name(sheet_name)
    datalist = []
    for i in range(10):
        if sheet.cell_value(row,9) =="YES":
            try:
                data = sheet.cell_value(row,i)
                datalist.append(data)
            except TypeError as f:
                break
        #print (datalist)
    return datalist

def Readexcelalldata_of_sheet(filename,sheet_name):
    '''返回整个sheet的内容的列表'''
    datalistlist =[]
    for i in range(1,100):
        try:
            data = Readexcelrow_of_sheet(filename,sheet_name,i)
            datalistlist.append(data)
        except IndexError as e:
            break
    #print(datalistlist)
    return  datalistlist

这里我们增加一个数据处理DataProcess方法，把excel读取的整个sheet数据（是一个列表，它的元素也是列表），我们用这个方法，根据行号返回一个json数据，也就是我们每条用例所要测试请求的数据。

import json

def DataProcess(datalist,myindex):
    '''datalist是一个列表，列表的元素也是列表，是从excel读到的指定sheet全部数据，根据index返回一个标准化的dict'''
    data = datalist[myindex]
    caseid =       data[0]
    casename =     data[1]
    method =       data[2]
    url =          data[3]
    payload =      data[4].encode('utf-8')
    expectedcode = data[5]
    expectedmsg =  data[6]
    expecteddata = data[7]
    headers =      json.loads(data[8])
    runselect =    data[9]

    newdict = {"runselect":runselect,
               "method":method,
               "caseid":caseid,
               "url":url,
               "data":payload,
               "headers":headers,
               "expectedcode":expectedcode,
               "expectedmsg":expectedmsg,
               "expecteddata":expecteddata}

    return newdict

　　

我们的test case样子如下：每个test方法除了一个数据处理的入参不一样，其他完全一致，无需更改。 而且我们可以做到所有的测试用例都按这个模板来撰写。

class LoginTest(unittest.TestCase):
    ''' '''

    def setUp(self):
        self.logname = logname
        self.mylog = AddLog.AddLog(self.logname).getlog()
        self.testcaseinfo = OperateExcel.Readexcelalldata_of_sheet(Excelname,Sheetname)
        self.result = {}
        global caseid
        caseid += 1

    def tearDown(self):
        self.mylog.info("case:%d completed" % caseid)

    @unittest.skipIf((max_testcaseid < 1) or (0 in indexlist),"用例最大id<1或用例标记为'不执行'")
    def test_caseid_1(self):
        testdata = DataProcess.DataProcess(self.testcaseinfo,0)
        headers= {"Content-Type": "application/json"}
        self.result = RunRequest.RunRequests(testdata["method"],
                                             testdata["url"],
                                             testdata["data"],
                                             testdata['headers']).run()
        self.myassert = AssertResult.AssertResult(self.logname, testdata, self.result)
        self.myassert.AssertEqualResult()

    @unittest.skipIf((max_testcaseid < 2) or (1 in indexlist),"用例最大id<2或用例标记为'不执行'")
    def test_caseid_2(self):
        testdata = DataProcess.DataProcess(self.testcaseinfo, 1)
        self.result = RunRequest.RunRequests(testdata["method"],
                                             testdata["url"],
                                             testdata["data"],
                                             testdata['headers']).run()
        self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
        self.myassert.AssertEqualResult()

    @unittest.skipIf((max_testcaseid < 3) or (2 in indexlist),"用例最大id<3或用例标记为'不执行'")
    def test_caseid_3(self):
        testdata = DataProcess.DataProcess(self.testcaseinfo, 2)
        self.result = RunRequest.RunRequests(testdata["method"],
                                             testdata["url"],
                                             testdata["data"],
                                             testdata['headers']).run()
        self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
        self.myassert.AssertEqualResult()

    @unittest.skipIf((max_testcaseid < 4) or (3 in indexlist), "用例最大id<4或用例标记为'不执行'")
    def test_caseid_4(self):
        testdata = DataProcess.DataProcess(self.testcaseinfo, 3)
        self.result = RunRequest.RunRequests(testdata["method"],
                                             testdata["url"],
                                             testdata["data"],
                                             testdata['headers']).run()
        self.myassert = AssertResult.AssertResult(self.logname,testdata,self.result)
        self.myassert.AssertEqualResult()

　　因为要用到不同请求方式，我们把他封装在一起了。

#coding = utf-8
import requests
import json

class RunRequests(object):

    def __init__(self,method,url,data,headers):

        self.url = url

        self.method = method

        self.data = data

        self.headers = headers

        self.result = ""

    def run(self):

        if self.method == "POST":

            if self.headers != "No":

                self.result = requests.post(self.url, data = self.data, headers = self.headers)

            else:
                self.result = requests.post(self.url, data = self.data)

        elif self.method == "GET":

            if self.headers != "No":

                self.result = requests.get(self.url, data=self.data)

                print("暂未处理headers")

            else:
                self.result = requests.get(self.url,data = self.data)

        elif self.method == "PUT":

            self.result = 1

        elif self.method == "DELETE":

            self.result = 1

        elif self.method == "HEAD":

            self.result = 1

        else:

            self.result = 1

        return  self.result.json()

　　

这些都完成后，想到测试完成后需要日志和邮件功能。也添加进来。

写完后整个框架就如下这个样子：

如果有朋友看到，觉得可以学习练练手，有需要源码可以留言，看到会及时回复。