Python接口测试之数据驱动（python驱动程序开发）

在接口的自动化测试中，客户端发送请求给服务端，在客户端发送请求的时候，包含了请求地址，请求方法，以及请求参数等数据，那么在接口的自动化测试中如何来分离这些请求地址和请求参数了，最好的方式是以数据驱动的方式分离到excel中，这样在excel中直接维护，即使后期由于某些原因修改了请求参数，在excel中修改也是很快的。在下来的案例中，一个系统，请求登录成功后，服务端返回token给客户端，客户端再次请求的时候需要带着这个token。关于HTTP的请求流程，token，session这些的处理，在前面的文章中有很详细的介绍，这里就不再介绍，下面会直接引入代码实战这部分。

首先安装处理excel的库，这里使用的Python版本是3.6，所以先安装第三方库，安装的命令为(已安装的会提示已存在)：

安装xlrd库好之后，创建excel文件，把请求地址和请求参数分离到excel中，见excel的数据：

现在来编写读取excel中的数据，主要思路为读取excel的数据后，把数据类型转为字典的数据类型，并且是按行的方式读取，实现的代码：

#!/usr/bin/env python

#-*-coding:utf-8-*-

#author:wuya

import os

import xlrd

import json

def readExcel(rowx, filePath='data.xlsx'):

''' 读取excel中数据并且返回

:parameter filePath:xlsx文件名称

:parameter rowx:在excel中的行数

'''

book = xlrd.open_workbook(filePath)

sheet = book.sheet_by_index(0)

return sheet.row_values(rowx)

来读取excel中的数据，并且查看它的数据类型是否是期望的字典类型，调用readExcel函数后，见执行的结果截图：

在截图中可以看到，数据类型是列表，并且返回了所有的数据，再次编写函数，返回XX行的请求地址和请求参数，在excel中，存在的共同点是不管数据是在那一行，第二列永远是请求地址，第三列是请求参数，编写获取请求地址和请求参数的函数，见源码：

#!/usr/bin/env python

#-*-coding:utf-8-*-

#author:wuya

import os

import xlrd

import json

def readExcel(rowx, filePath='data.xlsx'):

''' 读取excel中数据并且返回

:parameter filePath:xlsx文件名称

:parameter rowx:在excel中的行数

'''

book = xlrd.open_workbook(filePath)

sheet = book.sheet_by_index(0)

return sheet.row_values(rowx)

def getUrl(rowx):

'''

获取请求URL

:parameter rowx:在excel中的行数

'''

return readExcel(rowx)[1]

def getData(rowx):

'''

获取请求参数

:parameter rowx:在excel中的行数

'''

return json.loads(readExcel(rowx)[2])

在上面代码中，新增了获取获取请求地址和请求参数，因为请求参数数据类型是字典，所以进行了反序列化的处理。

下来编写接口用例，见数据未分离的接口用例，见实现的代码：

import unittest

import time as t

import requests

class ApiTest(unittest.TestCase):

@classmethod

def setUpClass(cls):

t.sleep(1)

@classmethod

def tearDownClass(cls):

pass

def getHeaders(self):

return {

'Parkingwang-Client-Source':'ParkingWangAPIClientWeb',

'Content-Type':'application/json;charset=UTF-8'}

def test_login_001(self):

'''登录业务:登录成功'''

r=requests.post(

url='http://180.97.80.42:9090/v5/login',

json={"username":"6666666666","password":"8144ed050cd8d053f24a1e179d7529e17c3a2ba9cfcfcd7d3bda9ec6a8156758"}, headers=self.getHeaders())

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

with open('token','w') as f:

f.write(r.json()['data']['token'])

def getToken(self):

with open('token','r') as f:

return f.read()

def test_login_002(self):

'''登录业务:查看用户信息'''

r=requests.post(

url='http://180.97.80.42:9090/v5/infoGet',

json={"token":self.getToken()},

headers=self.getHeaders())

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

if __name__ == '__main__':

unittest.main(verbosity=2)

下来使用数据驱动的方式把请求地址和请求参数分离出来，见修改后的源码：

#!/usr/bin/env python

#-*-coding:utf-8-*-

#author:wuya

import os

import xlrd

import json

def readExcel(rowx, filePath='data.xlsx'):

'''

读取excel中数据并且返回

:parameter filePath:xlsx文件名称

:parameter rowx:在excel中的行数

'''

book = xlrd.open_workbook(filePath)

sheet = book.sheet_by_index(0)

return sheet.row_values(rowx)

def getUrl(rowx):

'''

获取请求URL

:parameter rowx:在excel中的行数

'''

return readExcel(rowx)[1]

def getData(rowx):

'''

获取请求参数

:parameter rowx:在excel中的行数

'''

return json.loads(readExcel(rowx)[2])

import unittest

import time as t

import requests

class ApiTest(unittest.TestCase):

@classmethod

def setUpClass(cls):

t.sleep(1)

@classmethod

def tearDownClass(cls):

pass

def getHeaders(self):

return {

'Parkingwang-Client-Source':'ParkingWangAPIClientWeb',

'Content-Type':'application/json;charset=UTF-8'}

def test_login_001(self):

'''

登录业务:登录成功

'''

r=requests.post(

url=getUrl(1),

json=getData(1),

headers=self.getHeaders())

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

with open('token','w') as f:

f.write(r.json()['data']['token'])

def getToken(self):

with open('token','r') as f:

return f.read()

def test_login_002(self):

'''

登录业务:查看用户信息

'''

r=requests.post(

url=getUrl(2),

json=getData(2),

headers=self.getHeaders())

print(r.text)

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

if __name__ == '__main__':

unittest.main(verbosity=2)

数据分离后，维护数据统一是在excel中，第二个接口这些成功后，直接失败，见失败信息：

问题在于数据分离后，test_login_002的测试用例请求数据与登录成功后的token不一致，导致了错误，那么如何对这些动态参数进行处理了，处理的思路是：

从excel中读取数据

对如token这些动态参数再次进行赋值

返回赋值后的数据

调用赋值后的数据

依据如上的思路，对getToken方法进行修改，对token再次进行赋值，新增setToken方法，见修改后的该方法代码：

def getToken(self):

'''读取token文件里面的内容'''

with open('token','r') as f:

return f.read()

def setToken(self,rowx):

''' 对动态参数token进行赋值

:parameter rowx:在excel中的行数

''' dict1=getData(rowx)

#对tokek赋值

dict1['token']=self.getToken()

return dict1

在test_login_002的接口用例中，调用请求参数直接调用setToken方法，这样就不会出现错误了，见完整的代码：

#!/usr/bin/env python

#-*-coding:utf-8-*-

#author:wuya

import os

import xlrd

import json

import unittest

import time as t

import requests

def readExcel(rowx, filePath='data.xlsx'):

''' 读取excel中数据并且返回

:parameter filePath:xlsx文件名称

:parameter rowx:在excel中的行数

'''

book = xlrd.open_workbook(filePath)

sheet = book.sheet_by_index(0)

return sheet.row_values(rowx)

def getUrl(rowx):

''' 获取请求URL

:parameter rowx:在excel中的行数

'''

return readExcel(rowx)[1]

def getData(rowx):

'''

获取请求参数

:parameter rowx:在excel中的行数

'''

return json.loads(readExcel(rowx)[2])

class ApiTest(unittest.TestCase):

@classmethod

def setUpClass(cls):

t.sleep(1)

@classmethod

def tearDownClass(cls):

pass

def getHeaders(self):

return {

'Parkingwang-Client-Source':'ParkingWangAPIClientWeb',

'Content-Type':'application/json;charset=UTF-8'}

def test_login_001(self):

'''登录业务:登录成功'''

r=requests.post(

url=getUrl(1),

json=getData(1),

headers=self.getHeaders())

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

with open('token','w') as f:

f.write(r.json()['data']['token'])

def getToken(self):

'''读取token文件里面的内容'''

with open('token','r') as f:

return f.read()

def setToken(self,rowx):

'''

对动态参数token进行赋值

:parameter rowx:在excel中的行数

'''

dict1=getData(rowx)

#对tokek赋值

dict1['token']=self.getToken()

return dict1

def test_login_002(self):

'''登录业务:查看用户信息'''

r=requests.post(

url=getUrl(2),json=self.setToken(2),headers=self.getHeaders())

self.assertEqual(r.status_code,200)

self.assertEqual(r.json()['status'],0)

if __name__ == '__main__':

unittest.main(verbosity=2)