pytest+parametrize+yaml实例

# 一、yaml格式
#
# yaml是一种数据类型，可以和json之间灵活的切换，支持注释、换行、字符串等。可以用于配置文件或编写测试用例。
#
# 数据结构：一般是键值对的方式出现。注意编写时值前面必须有空格，键：（空格）值。
#
# 如果是数组，以-（空格） 开头来表示

# 二、定义读取yaml测试用例的方法

# Configuration.yaml_util.py

# yaml文件的读、写、删除

import os.path
import yaml# 读取yaml文件
# def read_yml(ymlPath):
#     if not os.path.isfile(ymlPath):
#         raise FileNotFoundError("文件路径不存在，请输入正确的路径")
#     f = open(ymlPath, 'r', encoding="utf-8")
#     cfg = f.read()
#     d = yaml.safe_load(cfg)
#     return ddef read_yaml(path):"""读取yaml文件:param path: 要读取的yaml文件路径:return: 返回yaml内容"""with open(path, encoding="utf-8") as f:value = yaml.load(stream=f, Loader=yaml.FullLoader)# return value[key]return valuedef write_yaml(path, data):"""写入yaml文件:param path::param data::return:"""with open(path, encoding="utf-8", mode='a') as f:yaml.dump(data, stream=f, allow_unicode=True)def clear_yaml(path):"""清空yaml文件，一般在整个项目执行之前，清空:return:"""with open(path, encoding="utf-8", mode="w") as f:f.truncate()

# Configuration.config.yaml

# 用于存储环境地址

# 三、把测试的服务器地址作为配置文件
#
# 测试的服务器，正常来讲都是比较固定的，但测试环境和正式环境是不同的，因此我们将服务器地址抽出来作为配置文件，方便切换不同的测试环境。这里将配置文件放在config.yaml文件中。

Host:# 有些系统会有生产环境、小版本环境、测试环境等等，所以这里统一管理这些环境地址，后面使用的时候来这里调用就行了baseurl: "http://192.168.0.103/"

#datas.user.yaml

# 用来存储用例

# 不同用例之间用-分隔
-batteryBrandName: 12,batteryRatedCapacity: 12,batteryRatedVoltage: 12,batterySpceC: 12,batterySpceG: 12,batterySpceK: 12,batteryType: 12,batteryTypeName: 12,batteryWeight: 12,cellCount: 12
-batteryBrandName: 13,batteryRatedCapacity: 13,batteryRatedVoltage: 13,batterySpceC: 13,batterySpceG: 13,batterySpceK: 13,batteryType: 13,batteryTypeName: 13,batteryWeight: 13,cellCount: 13

#Module.AddBatterTypeInfo.py

# 四、执行测试的用例文件
# 这里结合@pytest.mark.parametrize方法实现用例数据驱动。

import pytest
import requests
from Configuration.Logins import login
from Configuration.Logins import host
# from Configuration.yaml_util import read_yaml
from Configuration.yaml_util import *@pytest.mark.parametrize("batteryBrandName, batteryRatedCapacity, batteryRatedVoltage, batterySpceC,batterySpceG, batterySpceK, batteryType, batteryTypeName, batteryWeight, cellCount",read_yaml('../datas/user.yaml'))
def test_add_batter_type_info(batteryBrandName, batteryRatedCapacity, batteryRatedVoltage, batterySpceC,batterySpceG, batterySpceK, batteryType, batteryTypeName, batteryWeight, cellCount):API_url = "be/cloud/bsm/batterytypeinfo"baseurl = read_yaml('../Configuration/config.yaml')['Host']['baseurl']FULL_URL = baseurl + API_url# FULL_URL = host() + API_urlheader = {'Authorization': login()}body = {"batteryBrandName": batteryBrandName,"batteryRatedCapacity": batteryRatedCapacity,"batteryRatedVoltage": batteryRatedVoltage,"batterySpceC": batterySpceC,"batterySpceG": batterySpceG,"batterySpceK": batterySpceK,"batteryType": batteryType,"batteryTypeName": batteryTypeName,"batteryWeight": batteryWeight,"cellCount": cellCount}r = requests.post(FULL_URL, headers=header, json=body)res = r.json()# 上面的两句也可以合成一句： res = requests.post(FULL_URL, headers=header, json=body).json()print(res)return res# 用于仅执行该文件if __name__ == '__main__':# test_add_batter_type_info()pytest.main(["-s"])

#Module.BatterTpyeInfo_Search.py

import pytest
import requests
from Configuration.Logins import login
from Configuration.Logins import host# 方法必须是test_开头，因为装饰器parametrize是属于pytest框架的
# 在方法中传入这几个参数名信息
# 格式：@pytest.mark.parametrize("参数1，参数2，参数3，参数4",[(用例1)(用例2)(用例3)])    每个用例中有四个参数
@pytest.mark.parametrize("batteryBrandName,batteryTypeName,batteryType,delFlag",[("乐圆技术", "LY6050", "", ""), (1, 3, 3, "在用")])
def test_battery_type_info_search(batteryBrandName, batteryTypeName, batteryType, delFlag):API_URL = "be/cloud/bsm/batterytypeinfo/list"# 拼接URLFULL_URL = host() + API_URL# print("实际URL的值为：",FULL_URL)header = {'Authorization': login()}# print("实际URL的值为",header)# 执行用例，不要每条用例写一遍了，直接引用参数名即可par = {"batteryBrandName": batteryBrandName,"batteryTypeName": batteryTypeName,"batteryType": batteryType,"delFlag": delFlag}# get请求下，参数需要使用【params】来设置# post请求中，是使用data或者json来传递的r = requests.get(FULL_URL, headers=header, params=par).json()print("\n", r)return rif __name__ == '__main__':test_battery_type_info_search()

# run.run_case.py

# 调用执行所以模块的所有测试用例

import pytest
from Module.AddBatterTypeInfo import test_add_batter_type_info
from Module.BatteryTypeInfo_Search import test_battery_type_info_search
from Module.battery_type_info_exportSelect import battery_type_info_export# def add_batter_type_info():
#     res1 = test_add_batter_type_info()
#
#
# def battery_type_info_search():
#     res2 = test_battery_type_info_search()#
# def battery_type_info_export():
#     res3 = battery_type_info_export()if __name__ == '__main__':pytest.main(["run_case.py", "-s"])