1、定义csv文件
name,username,password 输入正确的用户名和密码,USERNAME,PASSWORD 输入错误的用户名和密码,USERNAME1,PASSWORD1
2、在Yaml文件中以$csv{}格式写
注意：name-username-password: username.csv（csv文件路径）
- parameters:name-username-password: username.csvname: $csv{name}request:json:username: $csv{username}password: $csv{password}
3、定义读取csv, 将$csv{}标记出的值，替换成csv表格中的值，最后读取yaml文件时返回
import json
from typing import Text, Dict, Union, List
import yaml
import jsonpath
import csv# 读取csv文件
def read_csv_file(csv_path: Text) -> List:""":param csv_path: csv路径:return:"""csv_data_list = []with open(csv_path, mode="r", encoding="utf-8-sig") as file:csv_data = csv.reader(file)for row in csv_data:csv_data_list.append(row)return csv_data_list# 读取测试用例使用该方法，以前的读取方法在这个方法中调用和解析
def read_testcase_yaml(yaml_path: Text) -> Union[Dict, List]:""":param yaml_path: 测试用例文件:return:"""with open(yaml_path, mode="r", encoding="utf-8") as file:args = yaml.load(file, Loader=yaml.FullLoader)# 当yaml文件大于等于两个字典的时候，不进行csv数据驱动if len(args) >= 2:return argselse:# 判断用例中是否有csv关键字"parameters"if jsonpath.jsonpath(*args, '$.parameters'):new_args = analysis_parameters(*args)return new_argselse:return args# 分析参数化
def analysis_parameters(args: Union[List, Dict]) -> Union[List, Dict]:args_keys = dict(args).keys()if "parameters" in args_keys:args_str = str(eval(json.dumps(args).replace('null', '" "')))for key, value in dict(args['parameters']).items():key_list = str(key).split("-")# 规范csv的数据的写法length_flag = Truecsv_data_list = read_csv_file(value)one_row_csv_data = csv_data_list[0]for csv_data in csv_data_list:if len(csv_data) != len(one_row_csv_data):length_flag = Falsebreak# 解析new_args = []if length_flag is True:# row代表行,column代表列for row in range(1, len(csv_data_list)):temp_args = args_str # 保持args_str数据不变for column in range(0, len(csv_data_list[row])):if csv_data_list[0][column] in key_list:temp_args = temp_args.replace("$csv{" + csv_data_list[0][column] + "}",csv_data_list[row][column])new_args.append(eval(temp_args)) # 使用eval将temp_args字符串转换回字典格式return new_argselse:return args
4、main函数演示读取结果
from userUtil import read_testcase_yamlif __name__ == '__main__':args = read_testcase_yaml("user.yaml")print(args)
5、执行结果
[{'parameters': {'name-username-password': 'username.csv'}, 'name': '输入正确的用户名和密码', 'request': {'json': {'username': 'USERNAME', 'password': 'PASSWORD'}}}, {'parameters': {'name-username-password': 'username.csv'}, 'name': '输入错误的用户名和密码', 'request': {'json': {'username': 'USERNAME1', 'password': 'PASSWORD1'}}}]