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

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

创新互联是一家专注于网站建设、成都网站建设与策划设计,图们网站建设哪家好?创新互联做网站,专注于网站建设10多年,网设计领域的专业建站公司;建站业务涵盖:图们等地区。图们做网站价格咨询:13518219792

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. import csv 
    2.   
  
  
  
  2. import json 
    3.   
  
  
  
  3. import requests 
    4.   
  
  
  
  4. import time 
    5.   
  
  
  
  5. import unittest 
    6.   
  
  
  
  6. def readCSV(self,filename): 
    7.   
  
  
  
  7.  ''' 
    8.   
  
  
  
  8.  :param filename: 需要读取的数据文件 
    9.   
  
  
  
  9.  :return: [{data1},{data2}...] 
    10.   
  
  
  
  10.  ''' 
    11.   
  
  
  
  11.  datas = [] 
    12.   
  
  
  
  12.  try: 
    13.   
  
  
  
  13.  #以DictReader的方式读取数据文件，方便与json互做转换 
    14.   
  
  
  
  14.  with open(filename,'r') as csvfile : 
    15.   
  
  
  
  15.  #从文件里读取到的数据转换成字典列表的格式 
    16.   
  
  
  
  16.  reader = csv.DictReader(csvfile) 
    17.   
  
  
  
  17.  for row in reader: 
    18.   
  
  
  
  18.  data = {} 
    19.   
  
  
  
  19.  data['id'] = row['id'] 
    20.   
  
  
  
  20.  data['url'] = row['url'] 
    21.   
  
  
  
  21.  data['token'] = str(row['token']) 
    22.   
  
  
  
  22.  data['mobile'] = row['mobile'] 
    23.   
  
  
  
  23.  data['email'] = row['email'] 
    24.   
  
  
  
  24.  data['expect'] = json.dumps(row['expect']) \ 
    25.   
  
  
  
  25.  if isinstance(row['expect'],dict) \ 
    26.   
  
  
  
  26.  else row['expect'] #如果expect读取出来的不是json则取其原值，否则转为json格式保存到result里 
    27.   
  
  
  
  27.  datas.append(data) 
    28.   
  
  
  
  28.  return datas 
    29.   
  
  
  
  29.  #如果文件找不到，返回空的datas 
    30.   
  
  
  
  30.  except FileNotFoundError: 
    31.   
  
  
  
  31.  print("文件不存在",filename) 
    32.   
  
  
  
  32.  return datas 
    33.

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

 
 
 
 
      
  
  
  
  1. def get_request(self,url,params): 
    2.   
  
  
  
  2.  ''' 
    3.   
  
  
  
  3.  通用的调用GET接口方法 
    4.   
  
  
  
  4.  :param url:string 接口路径 
    5.   
  
  
  
  5.  :param params:{"":"","":""} 需要传入的参数 
    6.   
  
  
  
  6.  :return: response响应体 
    7.   
  
  
  
  7.  ''' 
    8.   
  
  
  
  8.  print("调用API...") 
    9.   
  
  
  
  9.  r = requests.get(url,paramsparams=params) 
    10.   
  
  
  
  10.  print(r.text) 
    11.   
  
  
  
  11.  return r 
    12.   
  
  
  
  12. def post_request(self,url,params): 
    13.   
  
  
  
  13.  ''' 
    14.   
  
  
  
  14.  通用的调用POST接口方法 
    15.   
  
  
  
  15.  :param url: string 接口路径 
    16.   
  
  
  
  16.  :param params: {"":"","":""} 需要传入的参数 
    17.   
  
  
  
  17.  :return:response响应体 
    18.   
  
  
  
  18.  ''' 
    19.   
  
  
  
  19.  print("调用API...") 
    20.   
  
  
  
  20.  r = requests.post(url,params=json.dumps(params)) #post的方法必须用json.dumps()转化成json格式 
    21.   
  
  
  
  21.  print(r.text) 
    22.   
  
  
  
  22.  return r 
    23.

(5) assert_Result函数示例

 
 
 
 
      
  
  
  
  1. def assertResult(self,except_value,real_value): 
    2.   
  
  
  
  2.  ''' 
    3.   
  
  
  
  3.  校验样本字符串中是否包含指定字符串 
    4.   
  
  
  
  4.  :param except_value: string 指定字符串 
    5.   
  
  
  
  5.  :param real_value: string 样本字符串 
    6.   
  
  
  
  6.  :return: Boolean 样本中包含指定字符串返回True,否则返回False 
    7.   
  
  
  
  7.  ''' 
    8.   
  
  
  
  8.  ifsuccess = except_value in str(real_value) 
    9.   
  
  
  
  9.  return ifsuccess 
    10.

(6) write_CSV函数示例

 
 
 
 
      
  
  
  
  1. def writeCSV(self,filename,results): 
    2.   
  
  
  
  2.  ''' 
    3.   
  
  
  
  3.  写入csv文件指定内容 
    4.   
  
  
  
  4.  :param filename: string 需要写入的文件名称 
    5.   
  
  
  
  5.  :param results: [{data1},{data2},...] 写入的内容 
    6.   
  
  
  
  6.  :return: 无 
    7.   
  
  
  
  7.  ''' 
    8.   
  
  
  
  8.  print("写文件:",filename) 
    9.   
  
  
  
  9.  #以DictWriter的方式写文件 
    10.   
  
  
  
  10.  with open(filename,'w+') as csvfile: 
    11.   
  
  
  
  11.  headers="id,url,token,mobile,email,expect,real_value,assert_value".split(",") 
    12.   
  
  
  
  12.  writer = csv.DictWriter(csvfile,fieldnames=headers) 
    13.   
  
  
  
  13.  #写表头 
    14.   
  
  
  
  14.  writer.writeheader() 
    15.   
  
  
  
  15.  #写数据 
    16.   
  
  
  
  16.  if results.__len__() > 0 : 
    17.   
  
  
  
  17.  for result in results: 
    18.   
  
  
  
  18.  writer.writerow(result) 
    19.   
  
  
  
  19.  csvfile.close() 
    20.

(7) test_interface1函数示例

 
 
 
 
      
  
  
  
  1. def test_interface1(self): 
    2.   
  
  
  
  2.  
    3.   
  
  
  
  3.  #指定读取的数据文件名称 
    4.   
  
  
  
  4.  data_file = "../data/data.csv" 
    5.   
  
  
  
  5.  
    6.   
  
  
  
  6.  #指定最终结果生成的数据文件名称 
    7.   
  
  
  
  7.  result_file = "../data/result_{}.csv".format(str(time.time()).split(".")[0]) 
    8.   
  
  
  
  8.  
    9.   
  
  
  
  9.  #读取指定文件的数据 
    10.   
  
  
  
  10.  datas = self.readCSV(data_file) 
    11.   
  
  
  
  11.  
    12.   
  
  
  
  12.  #数据文件有内容则调用接口，否则直接测试结束 
    13.   
  
  
  
  13.  if datas.__len__() > 0: 
    14.   
  
  
  
  14.  results =[] 
    15.   
  
  
  
  15.  
    16.   
  
  
  
  16.  #获取数据文件里的每一行 
    17.   
  
  
  
  17.  for testcase in datas : 
    18.   
  
  
  
  18.  result = {} 
    19.   
  
  
  
  19.  result["id"] = testcase["id"] 
    20.   
  
  
  
  20.  result["url"] = testcase["url"] 
    21.   
  
  
  
  21.  result["token"] = testcase["token"] 
    22.   
  
  
  
  22.  result["mobile"] = testcase["mobile"] 
    23.   
  
  
  
  23.  result["email"] = testcase["email"] 
    24.   
  
  
  
  24.  result["expect"] = testcase["expect"] 
    25.   
  
  
  
  25.  
    26.   
  
  
  
  26.  #组装参数 
    27.   
  
  
  
  27.  params = { 
    28.   
  
  
  
  28.  "token":result["token"], 
    29.   
  
  
  
  29.  "mobile":result["mobile"], 
    30.   
  
  
  
  30.  "email":result["email"] 
    31.   
  
  
  
  31.  } 
    32.   
  
  
  
  32.  
    33.   
  
  
  
  33.  #调用API接口，获取响应结果 
    34.   
  
  
  
  34.  real_value = self.get_request(result["url"],params) 
    35.   
  
  
  
  35.  
    36.   
  
  
  
  36.  #调用assert方法，检查预期结果是否在响应结果中存在 
    37.   
  
  
  
  37.  assert_value = self.assertResult(result["expect"],real_value.text) 
    38.   
  
  
  
  38.  result["real_value"] = real_value.text 
    39.   
  
  
  
  39.  result["assert_value"] = assert_value 
    40.   
  
  
  
  40.  #获取每一行里的所有字段以及实际结果和验证结果 
    41.   
  
  
  
  41.  results.append(result) 
    42.   
  
  
  
  42.  #执行完所有记录后，将所有结果写入result.csv 
    43.   
  
  
  
  43.  self.writeCSV(result_file,results) #写入csv文件 
    44.   
  
  
  
  44.  print("测试结束") 
    45.

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

4. 总结

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

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

戳这里，看该作者更多好文

新闻标题：用Python实现数据驱动的接口自动化测试
网站链接：http://www.shufengxianlan.com/qtweb/news28/268328.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联

猜你还喜欢下面的内容

电子商务知识

分类信息网

品牌设计　　 　成都保温材料　　 　重庆机房托管　　 　广西弘博建站　　 　成都微信小程序　　 　瑞杰企业管理　　 　成都做网站　　 　重庆建站　　 　平昌保洁　　 　鳄鱼养殖　　 　成都企业网站建设　　 　成都移动服务器托管　　 　成都法律咨询　　 　wap网站制作　　 　鑫友邦火锅桌　　 　成都申请ssl证书　　 　微信小程序开发　　 　四川盐亭做网站　　 　成都网站推广　　 　成都网站建设