floraachy
157 lines
3.6 KiB
Python
157 lines
3.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Version: Python 3.9
|
|
# @Time : 2023/1/31 14:31
|
|
# @Author : chenyinhua
|
|
# @File : models.py
|
|
# @Software: PyCharm
|
|
# @Desc: 全局变量
|
|
|
|
# 标准库导入
|
|
from enum import Enum, unique # python 3.x版本才能使用
|
|
from typing import Text, Dict, Union, Any
|
|
# 第三方库导入
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class NotificationType(Enum):
|
|
""" 自动化通知方式 """
|
|
DEFAULT = 0
|
|
DING_TALK = 1
|
|
WECHAT = 2
|
|
EMAIL = 3
|
|
ALL = 4
|
|
|
|
|
|
@unique # 枚举类装饰器，确保只有一个名称绑定到任何一个值。
|
|
class AllureAttachmentType(Enum):
|
|
"""
|
|
allure 报告的文件类型枚举
|
|
"""
|
|
TEXT = "txt"
|
|
CSV = "csv"
|
|
TSV = "tsv"
|
|
URI_LIST = "uri"
|
|
|
|
HTML = "html"
|
|
XML = "xml"
|
|
JSON = "json"
|
|
YAML = "yaml"
|
|
PCAP = "pcap"
|
|
|
|
PNG = "png"
|
|
JPG = "jpg"
|
|
SVG = "svg"
|
|
GIF = "gif"
|
|
BMP = "bmp"
|
|
TIFF = "tiff"
|
|
|
|
MP4 = "mp4"
|
|
OGG = "ogg"
|
|
WEBM = "webm"
|
|
|
|
PDF = "pdf"
|
|
|
|
|
|
class Severity(str, Enum):
|
|
"""
|
|
测试用例优先级
|
|
"""
|
|
BLOCKER = 'BLOCKER' # blocker：阻塞缺陷（中断缺陷，客户端程序无响应，无法执行下一步操作）
|
|
CRITICAL = 'CRITICAL' # critical：严重缺陷（临界缺陷，功能点缺失）
|
|
NORMAL = 'NORMAL' # normal： 一般缺陷（边界情况，格式错误）
|
|
MINOR = 'MINOR' # minor：次要缺陷（界面错误与ui需求不符）
|
|
TRIVIAL = 'TRIVIAL' # trivial： 轻微缺陷（必须项无提示，或者提示不规范）
|
|
|
|
|
|
class TestCaseEnum(Enum):
|
|
"""
|
|
测试用例中字段
|
|
"""
|
|
ID = ("id", True)
|
|
TITLE = ("title", True)
|
|
URL = ("url", True)
|
|
SEVERITY = ("severity", False)
|
|
METHOD = ("method", True)
|
|
HEADERS = ("headers", True)
|
|
COOKIES = ("cookies", False)
|
|
RUN = ("run", False)
|
|
REQUEST_TYPE = ("request_type", True)
|
|
PAYLOAD = ("payload", False)
|
|
FILES = ("files", False)
|
|
THINK_TIME = ("think_time", False)
|
|
EXTRACT = ("extract", False)
|
|
VALIDATE = ("validate", True)
|
|
ASSERT_SQL = ("assert_sql", False)
|
|
CASE_DEPENDENCE = ("case_dependence", False)
|
|
|
|
|
|
class TestCase(BaseModel):
|
|
"""
|
|
测试用例各数据格式要求
|
|
"""
|
|
id: Union[None, Text]
|
|
title: Text
|
|
severity: Text
|
|
url: Text
|
|
method: Text
|
|
headers: Union[None, Dict, Text] = {}
|
|
cookies: Union[None, Dict, Text]
|
|
request_type: Text
|
|
run: Union[None, bool, Text] = None
|
|
payload: Any = None
|
|
files: Any = None
|
|
wait_seconds: int = None
|
|
extract: Union[None, Dict, Text] = None
|
|
assert_response: Union[None, Dict, Text]
|
|
assert_sql: Union[None, Dict, Text] = None
|
|
case_dependence: Union[None, Dict] = None
|
|
|
|
|
|
class Method(Enum):
|
|
"""
|
|
请求方式
|
|
"""
|
|
GET = "GET"
|
|
POST = "POST"
|
|
PUT = "PUT"
|
|
PATCH = "PATCH"
|
|
DELETE = "DELETE"
|
|
HEAD = "HEAD"
|
|
OPTION = "OPTION"
|
|
|
|
|
|
class RequestType(Enum):
|
|
"""
|
|
request请求发送，请求参数的数据类型
|
|
"""
|
|
JSON = "JSON"
|
|
PARAMS = "PARAMS"
|
|
DATA = "DATA"
|
|
FILE = 'FILE'
|
|
EXPORT = "EXPORT"
|
|
NONE = "NONE"
|
|
|
|
|
|
@unique
|
|
class AssertMethod(Enum):
|
|
"""
|
|
断言类型
|
|
注意：这里的类型与assert_type.py中的方法名相同，不要随意改动
|
|
"""
|
|
equals = "=="
|
|
less_than = "lt"
|
|
less_than_or_equals = "le"
|
|
greater_than = "gt"
|
|
greater_than_or_equals = "ge"
|
|
not_equals = "not_eq"
|
|
string_equals = "str_eq"
|
|
length_equals = "len_eq"
|
|
length_greater_than = "len_gt"
|
|
length_greater_than_or_equals = 'len_ge'
|
|
length_less_than = "len_lt"
|
|
length_less_than_or_equals = 'len_le'
|
|
contains = "contains"
|
|
contained_by = 'contained_by'
|
|
startswith = 'startswith'
|
|
endswith = 'endswith'
|