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函数代码示例

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32

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个方法)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

def get_request(self,url,params):  '''  通用的调用GET接口方法  :param url:string 接口路径  :param params:{"":"","":""} 需要传入的参数  :return: response响应体  '''  print("调用API...")  r = requests.get(url,params=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函数示例

1 2 3 4 5 6 7 8 9

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函数示例

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

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函数示例

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44

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("测试结束")

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

总结

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

​现在我也找了很多测试的朋友，做了一个分享技术的交流群，共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源，没人解答问题，坚持几天便放弃的感受
qq群号：485187702【暗号：csdn11】
可以加入我们一起交流。而且还有很多在自动化，性能，安全，测试开发等等方面有一定建树的技术大牛
分享他们的经验，还会分享很多直播讲座和技术沙龙
可以免费学习！划重点！开源的！！！
视频+文档+PDF+面试题可以关注公众号：【软件测试小dao】

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走！ 希望能帮助到你！【100%无套路免费领取】