用Python实现数据驱动的接口自动化测试

在接口测试的实试过程中，很多时候会用到对CSV的现数读取操作，本文主要说明Python3对CSV的据驱接口写入和读取。

1. 需求

某API，自动GET方法，化测token,实试mobile,email三个参数

token为必填项 mobile,email 必填其中1项 mobile为手机号,email为email格式

2. 方案

针对上面的API，在做接口测试时，现数需要的据驱接口测试用例动辄会多达10+, 这个时候采用数据驱动的方式将共性的内容写入配置文件或许会更合适。

这里考虑把API、自动参数、化测以及预期结果预行在格式化的实试CSV里保存，利用csv组件从CSV里读取URL、现数参数以及预期结果，据驱接口Requests组件发起请求，自动将响应结果与预期结果进行比对，源码库化测***把比对结果写到结果CSV。

流程如下图

3. 实现

(1) 在上代码之前，先安装好如下几个组件：

csv 读写CSV文件 json requests 发起请求，获取响应结果 unittest 测试用例调度

(2) data.csv(本示例选取部分用例)

(3) reader_CSV函数代码示例

import csv import json import requests import time import unittest def readCSV(self,filename):  :param filename: 需要读取的数据文件  :return: [{data1},{data2}...]  datas = []  try:  #以DictReader的方式读取数据文件，方便与json互做转换  with open(filename,r) as csvfile :  #从文件里读取到的数据转换成字典列表的格式  reader = csv.DictReader(csvfile)  for row in reader:  data = {}  data[id] = row[id]  data[url] = row[url]  data[token] = str(row[token])  data[mobile] = row[mobile]  data[email] = row[email]  data[expect] = json.dumps(row[expect]) \  if isinstance(row[expect],dict) \  else row[expect] #如果expect读取出来的不是json则取其原值，否则转为json格式保存到result里  datas.append(data)  return datas  #如果文件找不到，返回空的datas  except FileNotFoundError:  print("文件不存在",filename)  return datas 

(4) request_URL函数示例(包含GET请求和POST请求2个方法)

def get_request(self,url,params):  通用的调用GET接口方法  :param url:string 接口路径  :param params:{"":"","":""} 需要传入的参数  :return: response响应体  print("调用API...")  r = requests.get(url,paramsparams=params)  print(r.text)  return r def post_request(self,url,params):  通用的调用POST接口方法  :param url: string 接口路径  :param params: {"":"","":""} 需要传入的参数  :return:response响应体  print("调用API...")  r = requests.post(url,params=json.dumps(params)) #post的方法必须用json.dumps()转化成json格式  print(r.text)  return r 

(5) assert_Result函数示例

def assertResult(self,except_value,real_value):  校验样本字符串中是否包含指定字符串  :param except_value: string 指定字符串  :param real_value: string 样本字符串  :return: Boolean 样本中包含指定字符串返回True,否则返回False  ifsuccess = except_value in str(real_value)  return ifsuccess 

(6) write_CSV函数示例

def writeCSV(self,filename,results):  写入csv文件指定内容  :param filename: string 需要写入的文件名称  :param results: [{data1},{data2},...] 写入的内容  :return: 无  print("写文件:",filename)  #以DictWriter的免费信息发布网方式写文件  with open(filename,w+) as csvfile:  headers="id,url,token,mobile,email,expect,real_value,assert_value".split(",")  writer = csv.DictWriter(csvfile,fieldnames=headers)  #写表头  writer.writeheader()  #写数据  if results.__len__() > 0 :  for result in results:  writer.writerow(result)  csvfile.close() 

(7) test_interface1函数示例

def test_interface1(self):  #指定读取的数据文件名称  data_file = "../data/data.csv"  #指定最终结果生成的数据文件名称  result_file = "../data/result_{}.csv".format(str(time.time()).split(".")[0])  #读取指定文件的数据  datas = self.readCSV(data_file)  #数据文件有内容则调用接口，否则直接测试结束  if datas.__len__() > 0:  results =[]  #获取数据文件里的每一行  for testcase in datas :  result = {}  result["id"] = testcase["id"]  result["url"] = testcase["url"]  result["token"] = testcase["token"]  result["mobile"] = testcase["mobile"]  result["email"] = testcase["email"]  result["expect"] = testcase["expect"]  #组装参数  params = {  "token":result["token"],  "mobile":result["mobile"],  "email":result["email"]  }  #调用API接口，获取响应结果  real_value = self.get_request(result["url"],params)  #调用assert方法，检查预期结果是否在响应结果中存在  assert_value = self.assertResult(result["expect"],real_value.text)  result["real_value"] = real_value.text  result["assert_value"] = assert_value  #获取每一行里的所有字段以及实际结果和验证结果  results.append(result)  #执行完所有记录后，将所有结果写入result.csv  self.writeCSV(result_file,results) #写入csv文件  print("测试结束") 

8result_1523956055.csv(本示例中的测试结果请忽略)

4. 总结

python封装了很多方法，对于测试来说开发速度相对较快，接口自动化测试如果采用CSV管理的数据驱动方式，使用csv+requests是测试开发不容错过的利器之一。

【本文是专栏机构“岂安科技”的原创文章，转载请通过微信公众号(bigsec)联系原作者】

戳这里，b2b供应网看该作者更多好文

本文地址：http://www.bzve.cn/news/301a8099618.html