floraachy
123 lines
4.9 KiB
Python
123 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Time : 2023/6/7 16:37
|
|
# @Author : chenyinhua
|
|
# @File : case_data_analysis.py
|
|
# @Software: PyCharm
|
|
# @Desc: 分析用例数据是否符合规范
|
|
|
|
# 标准库导入
|
|
from typing import Text
|
|
# 第三方库导入
|
|
from config.models import TestCaseEnum, Method, RequestType, Severity
|
|
|
|
|
|
class CaseDataCheck:
|
|
"""
|
|
用例数据解析, 判断数据填写是否符合规范
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.case_data = None
|
|
self.case_id = None
|
|
|
|
@property
|
|
def get_method(self) -> Text:
|
|
"""
|
|
处理method参数：校验method参数是否是指定枚举值中的一个
|
|
"""
|
|
return self.check_params_right(
|
|
Method,
|
|
self.case_data.get(TestCaseEnum.METHOD.value[0])
|
|
)
|
|
|
|
@property
|
|
def get_severity(self) -> Text:
|
|
"""
|
|
处理severity参数：如果Severity为空或者不传或者传错，视为NORMAL
|
|
"""
|
|
# 如果Severity为空或者不传或者传错，视为NORMAL
|
|
attr = self.case_data.get(TestCaseEnum.SEVERITY.value[0])
|
|
if attr is None or attr.upper() not in Severity._member_names_:
|
|
return "NORMAL"
|
|
else:
|
|
return attr.upper()
|
|
|
|
@property
|
|
def get_request_type(self):
|
|
"""
|
|
处理request_type参数：校验request_type参数是否是指定枚举值中的一个
|
|
"""
|
|
return self.check_params_right(
|
|
RequestType,
|
|
self.case_data.get(TestCaseEnum.REQUEST_TYPE.value[0])
|
|
)
|
|
|
|
def check_case_data_attr(self, attr: Text):
|
|
"""
|
|
检查用例中是否缺失某个参数
|
|
"""
|
|
assert attr in self.case_data.keys(), (
|
|
f"用例ID: {self.case_id} --> 缺少 {attr} 参数，请确认用例内容是否编写规范."
|
|
)
|
|
|
|
def check_params_exit(self):
|
|
"""
|
|
遍历一个枚举类中所有成员，并检查与每个成员对应的实例属性是否存在。
|
|
如果属性存在，则什么也不做，如果不存在，则抛出异常或执行其他操作
|
|
"""
|
|
for enum in list(TestCaseEnum._value2member_map_):
|
|
if enum[1]:
|
|
self.check_case_data_attr(enum[0])
|
|
|
|
def check_required_fields(self):
|
|
for field in TestCaseEnum:
|
|
if field.value[1]: # 判断是否为必填参数
|
|
field_name = field.value[0]
|
|
value = self.case_data[field_name]
|
|
if value is None or (isinstance(value, str) and len(value.strip()) == 0):
|
|
raise ValueError(f"用例ID: {self.case_id} --> {field_name}字段的值是：{value}， 该字段值是必填，请检查是否填写正确.")
|
|
|
|
def check_params_right(self, enum_name, attr):
|
|
"""
|
|
检查参数值是否正确，符合要求规范
|
|
"""
|
|
_member_names_ = enum_name._member_names_
|
|
assert attr.upper() in _member_names_, (
|
|
f"用例ID: {self.case_id} --> {enum_name}: {attr} 填写不正确，"
|
|
f"当前框架中只支持 {_member_names_} 类型."
|
|
f"如有疑问，请联系管理员."
|
|
)
|
|
return attr.upper()
|
|
|
|
def case_process(self, cases):
|
|
case_list = []
|
|
for key, values in cases.items():
|
|
# 公共配置中的数据，与用例数据不同，需要单独处理
|
|
if key == 'case_info':
|
|
for value in values:
|
|
self.case_data = value
|
|
self.case_id = value.get("id")
|
|
# 检查用例参数，需要必填的是否有值
|
|
self.check_required_fields()
|
|
# 检查用例参数值，是否都填写正确
|
|
self.check_params_exit()
|
|
case_data = {
|
|
'id': self.case_data.get(TestCaseEnum.ID.value[0]),
|
|
'title': self.case_data.get(TestCaseEnum.TITLE.value[0]),
|
|
'severity': self.get_severity,
|
|
'url': self.case_data.get(TestCaseEnum.URL.value[0]),
|
|
'run': self.case_data.get(TestCaseEnum.RUN.value[0]),
|
|
'method': self.get_method,
|
|
'headers': self.case_data.get(TestCaseEnum.HEADERS.value[0]),
|
|
'cookies': self.case_data.get(TestCaseEnum.COOKIES.value[0]),
|
|
'request_type': self.get_request_type,
|
|
'payload': self.case_data.get(TestCaseEnum.PAYLOAD.value[0]),
|
|
'files': self.case_data.get(TestCaseEnum.FILES.value[0]),
|
|
'extract': self.case_data.get(TestCaseEnum.EXTRACT.value[0]),
|
|
"assert_response": self.case_data.get(TestCaseEnum.ASSERT_RESPONSE.value[0]),
|
|
"assert_sql": self.case_data.get(TestCaseEnum.ASSERT_SQL.value[0]),
|
|
}
|
|
case_list.append(case_data)
|
|
|
|
return case_list
|