前言

大家在接口測試的過程中，很多時候會用到對CSV的讀取操作，本文主要說明Python3對CSV的寫入和讀取。下面話不多說了，來一起看看詳細的介紹吧。

1、需求

某API，GET方法，token,mobile,email三個參數

2、方案

針對上面的API，在做接口測試時，需要的測試用例動輒會多達10+, 這個時候采用數據驅動的方式將共性的內容寫入配置文件或許會更合適。

這里考慮把API、參數、以及預期結果預行在格式化的CSV里保存，利用csv組件從CSV里讀取URL、參數以及預期結果，Requests組件發起請求，將響應結果與預期結果進行比對，最后把比對結果寫到結果CSV。

流程如下圖 ☟

3、實現

1、在上代碼之前，先安裝好如下幾個組件：

2、data.csv(本示例選取部分用例）

3、reader_CSV函數代碼示例

import csvimport jsonimport requestsimport timeimport unittestdef 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,params=params) print(r.text) return rdef 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