102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
|
|
"""
|
|
@author: CNWei
|
|
@Software: PyCharm
|
|
@contact: t6i888@163.com
|
|
@file: data_driver
|
|
@date: 2025/3/3 10:56
|
|
@desc:
|
|
"""
|
|
from pathlib import Path
|
|
|
|
from commons.models import CaseInfo
|
|
from commons.templates import Template
|
|
from commons.file_processors.yaml_processor import StringOrDict
|
|
|
|
|
|
class DataDriver:
|
|
@staticmethod
|
|
def generate_cases(file_name, case_info) -> list:
|
|
|
|
if not case_info.get("parametrize"):
|
|
return {file_name + "[--]": case_info}
|
|
|
|
# cases = []
|
|
args_names = case_info.get("parametrize")[0]
|
|
for i, args_values in enumerate(case_info.get("parametrize")[1:]):
|
|
# print(args_values)
|
|
context = dict(zip(args_names, args_values))
|
|
# print(context)
|
|
# rendered = Template(CaseInfo(**case_info).to_yaml()).render(context)
|
|
rendered = Template(StringOrDict.to_string(case_info)).render(context)
|
|
# cases.append({file_name + "[" + str(i) + "]": StringOrDict.to_dict(rendered)})
|
|
yield {file_name + "[" + str(i) + "]": StringOrDict.to_dict(rendered)}
|
|
# return cases
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from commons.file_processors.yaml_processor import YamlFile
|
|
|
|
file_path = Path(r"E:\PyP\InterfaceAutoTest\TestCases\test_1_user.yaml")
|
|
|
|
file_obj = YamlFile(file_path)
|
|
# print(file_path.stem)
|
|
file_name = file_path.stem
|
|
mock_case_info = {
|
|
"case_info0": {
|
|
"feature": "页面状态",
|
|
"story": "状态",
|
|
"title": "查询状态信息",
|
|
"request": "",
|
|
"extract": "",
|
|
"validate": "",
|
|
"parametrize": [["title", "username", "password", "msg"], ["测试1", "user1", "pass1", "200"],
|
|
["测试2", "user2", "pass2", "300"]]
|
|
},
|
|
"case_info1": {
|
|
"feature": "页面状态",
|
|
"story": "状态",
|
|
"title": "查询状态信息",
|
|
"request": "",
|
|
"extract": "",
|
|
"validate": "",
|
|
"parametrize": [1, 2, 3]
|
|
},
|
|
"case_info2": {
|
|
"feature": "页面状态",
|
|
"story": "状态",
|
|
"title": "查询状态信息",
|
|
"request": "",
|
|
"extract": "",
|
|
"validate": "",
|
|
"parametrize": [1, 2, 3]
|
|
}
|
|
|
|
}
|
|
|
|
dd = DataDriver()
|
|
# cases = dd.generate_cases(mock_case_info.get("case_info0"))
|
|
cases = dd.generate_cases(file_name, file_obj)
|
|
# print(cases)
|
|
# print(len(cases))
|
|
keys_list = []
|
|
titles = []
|
|
for item in cases:
|
|
print(item)
|
|
# 遍历列表中的每个字典
|
|
for key, value in item.items():
|
|
print(f"key:{key}")
|
|
keys_list.append(key)
|
|
print(f"value:{value}")
|
|
# # 遍历内层字典（这里内层字典其实只有一个键值对）
|
|
titles.append(value['title'])
|
|
print(item)
|
|
|
|
print(keys_list)
|
|
print(titles)
|
|
|
|
# ddt_title = [data.title for data in ddt_data]
|
|
# logger.info(f"{ddt_title=}")
|