141 lines
5.8 KiB
Python
141 lines
5.8 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 custom_utils.models import TestCaseEnum, Method, RequestType, Severity
|
||
from loguru import logger
|
||
|
||
class CaseCheckException(Exception):
|
||
"""用例检查异常类"""
|
||
def __init__(self, case_id: str, message: str):
|
||
self.case_id = case_id
|
||
self.message = message
|
||
super().__init__(self.message)
|
||
|
||
def __str__(self):
|
||
return f"用例ID: {self.case_id} --> {self.message}"
|
||
|
||
|
||
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):
|
||
"""
|
||
检查用例中是否缺失某个参数
|
||
"""
|
||
if attr not in self.case_data.keys():
|
||
error_msg = f"缺少 {attr} 参数，请确认用例内容是否编写规范"
|
||
raise CaseCheckException(self.case_id, error_msg)
|
||
|
||
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:
|
||
try:
|
||
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):
|
||
error_msg = f"{field_name}字段的值是：{value}，该字段值是必填，请检查是否填写正确"
|
||
logger.error(error_msg)
|
||
raise CaseCheckException(self.case_id, error_msg)
|
||
except Exception as e:
|
||
error_msg = f"在处理 {field} 时发生异常，需要检查的field_name = {field.value[0]}，所有的case_data = {self.case_data}，请检查拼写。"
|
||
raise CaseCheckException(self.case_id, error_msg)
|
||
|
||
def check_params_right(self, enum_name, attr):
|
||
"""
|
||
检查参数值是否正确，符合要求规范
|
||
"""
|
||
_member_names_ = enum_name._member_names_
|
||
if attr.upper() not in _member_names_:
|
||
error_msg = (f"{enum_name}: {attr} 填写不正确，"
|
||
f"当前框架中只支持 {_member_names_} 类型。"
|
||
f"如有疑问，请联系管理员。")
|
||
raise CaseCheckException(self.case_id, error_msg)
|
||
return attr.upper()
|
||
|
||
def case_process(self, cases):
|
||
case_list = []
|
||
for key, values in cases.items():
|
||
# 公共配置中的数据，与用例数据不同，需要单独处理
|
||
if key == 'teststeps':
|
||
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]),
|
||
"think_time": self.case_data.get(TestCaseEnum.THINK_TIME.value[0]),
|
||
"validate": self.case_data.get(TestCaseEnum.VALIDATE.value[0]),
|
||
'extract': self.case_data.get(TestCaseEnum.EXTRACT.value[0]),
|
||
"case_dependence": self.case_data.get(TestCaseEnum.CASE_DEPENDENCE.value[0])
|
||
}
|
||
case_list.append(case_data)
|
||
|
||
return case_list
|