前言
大家在接口测试的过程中，很多时候会用到对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是测试开发不容错过的利器之一。
总结：
感谢每一个认真阅读我文章的人！！！
作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。
软件测试面试文档
我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。
视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。
![[NOIP1998 提高组] 拼数](https://img-blog.csdnimg.cn/direct/667a8d08e2df436982d3986eec37d18c.png#pic_center)
