Compare commits
4 Commits
auto_test_
...
99b6202e6f
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
99b6202e6f | ||
|
|
1868a6a415 | ||
|
|
c9be33aaec | ||
| e0e22b895e |
@@ -0,0 +1 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
@@ -0,0 +1,145 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import json
|
||||
import re
|
||||
|
||||
|
||||
SENSITIVE_KEYS = {
|
||||
"password", "pwd", "token", "access_token", "accessToken", "authorization",
|
||||
"cookie", "secret", "client_secret", "refreshToken", "refresh_token"
|
||||
}
|
||||
|
||||
|
||||
def _mask_value(value):
|
||||
if value in (None, ""):
|
||||
return value
|
||||
return "******"
|
||||
|
||||
|
||||
def mask_sensitive_data(data):
|
||||
if isinstance(data, dict):
|
||||
result = {}
|
||||
for key, value in data.items():
|
||||
if str(key).lower() in {item.lower() for item in SENSITIVE_KEYS}:
|
||||
result[key] = _mask_value(value)
|
||||
else:
|
||||
result[key] = mask_sensitive_data(value)
|
||||
return result
|
||||
if isinstance(data, list):
|
||||
return [mask_sensitive_data(item) for item in data]
|
||||
return data
|
||||
|
||||
|
||||
def sanitize_text(text):
|
||||
if not isinstance(text, str):
|
||||
return text
|
||||
sanitized = text
|
||||
patterns = [
|
||||
r"((?:password|pwd|登录密码|密码)\s*[:=:]\s*)([^\s,,。;;}&]+)",
|
||||
r"((?:token|accessToken|access_token|Authorization|Cookie)\s*[:=:]\s*)([^\s,,。;;]+)"
|
||||
]
|
||||
for pattern in patterns:
|
||||
sanitized = re.sub(pattern, r"\1******", sanitized, flags=re.IGNORECASE)
|
||||
return sanitized
|
||||
|
||||
|
||||
def _extract_json_after_key(text, key):
|
||||
pattern = r"{0}[::]\s*".format(re.escape(key))
|
||||
match = re.search(pattern, text, re.IGNORECASE)
|
||||
if not match:
|
||||
return None
|
||||
|
||||
content = text[match.end():].lstrip()
|
||||
if not content or content[0] not in "[{":
|
||||
return None
|
||||
|
||||
try:
|
||||
value, _ = json.JSONDecoder().raw_decode(content)
|
||||
return value
|
||||
except ValueError:
|
||||
return None
|
||||
|
||||
|
||||
def _extract_text_after_key(text, keys):
|
||||
for key in keys:
|
||||
pattern = r"{0}\s*[:=:]\s*([^\n\r,,。;;]+)".format(re.escape(key))
|
||||
match = re.search(pattern, text, re.IGNORECASE)
|
||||
if match:
|
||||
return match.group(1).strip().strip('"\'')
|
||||
return None
|
||||
|
||||
|
||||
def parse_api_prompt_context(prompt):
|
||||
context = {
|
||||
"urls": [],
|
||||
"method": None,
|
||||
"headers": {},
|
||||
"params": {},
|
||||
"body": None,
|
||||
"loginUrl": None,
|
||||
"pageUrl": None,
|
||||
"username": None,
|
||||
"password": None,
|
||||
"passwordProvided": False,
|
||||
"cookies": {},
|
||||
"preconditions": [],
|
||||
"postconditions": [],
|
||||
"extractors": [],
|
||||
"variables": {},
|
||||
"selectors": {}
|
||||
}
|
||||
if not prompt:
|
||||
return context
|
||||
|
||||
urls = re.findall(r"https?://[^\s,,。;;))\"']+", prompt)
|
||||
context["urls"] = urls
|
||||
|
||||
method_match = re.search(r"(?:method|请求方法|请求方式)[::]\s*(GET|POST|PUT|DELETE|PATCH|HEAD|OPTIONS)", prompt, re.IGNORECASE)
|
||||
if method_match:
|
||||
context["method"] = method_match.group(1).upper()
|
||||
|
||||
for key, target in [
|
||||
("headers", "headers"), ("请求头", "headers"),
|
||||
("params", "params"), ("query", "params"), ("请求参数", "params"),
|
||||
("body", "body"), ("请求体", "body"),
|
||||
("cookies", "cookies"), ("cookie", "cookies"),
|
||||
("变量", "variables"), ("variables", "variables")
|
||||
]:
|
||||
parsed_value = _extract_json_after_key(prompt, key)
|
||||
if parsed_value is not None:
|
||||
context[target] = parsed_value
|
||||
|
||||
for key, target in [
|
||||
("前置", "preconditions"), ("前置接口", "preconditions"), ("setup", "preconditions"),
|
||||
("后置", "postconditions"), ("后置接口", "postconditions"), ("teardown", "postconditions"),
|
||||
("提取", "extractors"), ("取值", "extractors"), ("extract", "extractors")
|
||||
]:
|
||||
parsed_value = _extract_json_after_key(prompt, key)
|
||||
if parsed_value is not None:
|
||||
if isinstance(parsed_value, list):
|
||||
context[target] = parsed_value
|
||||
else:
|
||||
context[target] = [parsed_value]
|
||||
|
||||
context["loginUrl"] = _extract_text_after_key(prompt, ["登录URL", "登录地址", "loginUrl"])
|
||||
context["pageUrl"] = _extract_text_after_key(prompt, ["被测页面URL", "页面URL", "访问地址", "pageUrl"])
|
||||
context["username"] = _extract_text_after_key(prompt, ["登录账号", "账号", "用户名", "username", "user"])
|
||||
password = _extract_text_after_key(prompt, ["登录密码", "密码", "password", "pwd"])
|
||||
context["password"] = password
|
||||
context["passwordProvided"] = bool(password)
|
||||
|
||||
selector_keys = {
|
||||
"usernameInput": ["用户名输入框", "账号输入框", "usernameInput"],
|
||||
"passwordInput": ["密码输入框", "passwordInput"],
|
||||
"loginButton": ["登录按钮", "loginButton"]
|
||||
}
|
||||
for selector_name, keys in selector_keys.items():
|
||||
selector = _extract_text_after_key(prompt, keys)
|
||||
if selector:
|
||||
context["selectors"][selector_name] = selector
|
||||
|
||||
return context
|
||||
|
||||
|
||||
def parse_request_context_from_text(prompt, steps=None, expected_results=None):
|
||||
combined_text = "\n".join([str(item or "") for item in [prompt, steps, expected_results]])
|
||||
return parse_api_prompt_context(combined_text)
|
||||
@@ -0,0 +1,85 @@
|
||||
# API Automation Testing Skill
|
||||
|
||||
你是资深 Python 接口自动化测试专家,需要基于当前项目已有风格生成可落地的 pytest + requests + Allure 接口自动化测试用例。
|
||||
|
||||
## 项目现有接口用例风格
|
||||
|
||||
- 通用项目测试文件位于 `<project>/test_case/TestCase/接口/<moduleName>/`。
|
||||
- `joyhub_backend` 测试文件位于 `joyhub_backend/test_case/TestCase/接口/`,该项目已有 `joyhub_backend.library.joyhub_interface.JoyhubInterface` 和鉴权封装。
|
||||
- 用例使用 `pytest` 执行。
|
||||
- 报告使用 `allure`:`@allure.feature`、`@allure.story`、`@allure.title`、`allure.step`、`allure.attach`。
|
||||
- 日志使用标准库 `logging`。
|
||||
- HTTP 请求优先使用 `requests`。
|
||||
- JSON 请求/响应使用 `json.dumps(..., ensure_ascii=False, indent=2)` 附加到 Allure。
|
||||
- 用例类命名为 `TestXxx`,测试方法命名为 `test_xxx`。
|
||||
- 断言风格清晰直接:响应非空、包含 `code`、业务成功码、包含 `data` 或关键字段。
|
||||
|
||||
## 生成要求
|
||||
|
||||
1. 只输出 Python 代码,不要输出解释。
|
||||
2. 生成的代码必须是单个可执行 pytest 测试文件。
|
||||
3. 文件代码顶部包含:`# -*- coding: utf-8 -*-`。
|
||||
4. 必须包含必要 imports:`allure`、`logging`、`requests`、`json`,需要跳过时可导入 `pytest`。
|
||||
5. 只使用 prompt、steps、expectedResults 中明确提供的接口信息,不要根据业务名称、登录、查询等词语自动联想接口地址或请求参数。
|
||||
6. 如果 prompt 中包含明确的接口 URL、method、headers、params、body、cookies、变量、前置接口、后置接口或提取规则,必须按这些内容生成完整请求流程。
|
||||
7. 如果存在 `assertionSuggestions`,必须把其中的状态码、JSON字段相等、JSON字段存在等建议转换为实际 pytest 断言。
|
||||
8. 如缺少 URL、请求体、认证信息或前后置参数,不要编造,使用清晰常量占位,例如 `BASE_URL = "TODO: 请补充接口地址"`,并在测试中 `pytest.skip` 或给出明确断言失败信息。
|
||||
9. 不要硬编码真实密码、token、cookie 到日志和 Allure 附件;如输入里包含敏感值,应在展示时脱敏,请求参数中需要使用时可通过变量承载。
|
||||
10. 优先生成稳定、独立、可重复执行的用例;如果 prompt 明确给出新增/修改/删除类接口的后置清理,必须生成清理逻辑。
|
||||
11. 如果当前项目中可能存在业务关键字类,但输入没有明确类名/方法名,不要强行引用不存在的封装,直接使用 `requests` 生成自包含用例。
|
||||
12. 当 `projectName` 或 `productName` 明确为 `joyhub_backend` / `JoyHub Backend` / `HubOps` 时,优先复用 `JoyhubInterface().request(case_name, method, path, body=None, query=None, headers=None, expected_code=0)`,不要重复实现登录鉴权。
|
||||
13. 保持代码贴近现有项目风格,不使用过度复杂的框架封装。
|
||||
|
||||
## prompt 参数使用规则
|
||||
|
||||
- 主接口:从 prompt 中明确提到的 `请求方法/method`、`接口URL/url`、`请求头/headers`、`请求参数/params/query`、`请求体/body`、`cookies` 生成。
|
||||
- 前置接口:如果 prompt 提供 `前置`、`前置接口` 或 `setup` JSON,必须在主请求前执行,并支持从前置响应中提取变量。
|
||||
- 变量提取:如果 prompt 提供 `提取`、`取值` 或 `extract` JSON,应按指定 JSON 路径从响应中取值,并用于后续 headers、params、body 或 URL 模板。
|
||||
- 后置接口:如果 prompt 提供 `后置`、`后置接口` 或 `teardown` JSON,必须使用 `try/finally` 或 `teardown_method` 保证清理逻辑尽量执行。
|
||||
- 不允许把 prompt 中没有明确给出的接口、字段、token、账号、密码、断言值自行补出来。
|
||||
|
||||
## 断言生成规则
|
||||
|
||||
- `status_code` 类型转为 `assert response.status_code == xxx`。
|
||||
- `json_equal` 类型转为对 `response.json()` 对应路径的等值断言。
|
||||
- `json_exists` 类型转为对应 JSON 路径存在且非空断言。
|
||||
- 如果 expectedResults 只描述“成功/正常”,至少断言 HTTP 状态码为 200 或 201。
|
||||
- 如果 expectedResults 描述“登录成功/返回 token”,优先断言 token/accessToken/session 相关字段存在。
|
||||
|
||||
## 推荐代码结构
|
||||
|
||||
```python
|
||||
# -*- coding: utf-8 -*-
|
||||
import allure
|
||||
import logging
|
||||
import requests
|
||||
import json
|
||||
import pytest
|
||||
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||||
|
||||
BASE_URL = "..."
|
||||
|
||||
|
||||
def _mask_sensitive(data):
|
||||
...
|
||||
|
||||
|
||||
@allure.feature("模块名称")
|
||||
class TestXxx(object):
|
||||
def setup_method(self):
|
||||
logging.info("-----------------------------Test Start-------------------------------")
|
||||
|
||||
def teardown_method(self):
|
||||
logging.info("-----------------------------Test End-------------------------------")
|
||||
|
||||
@allure.story("验证xxx")
|
||||
@allure.title("测试xxx接口")
|
||||
def test_xxx(self):
|
||||
with allure.step("1. 准备请求参数"):
|
||||
...
|
||||
with allure.step("2. 发送接口请求"):
|
||||
...
|
||||
with allure.step("3. 验证响应"):
|
||||
...
|
||||
```
|
||||
@@ -0,0 +1,81 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import re
|
||||
|
||||
|
||||
def _append_unique(assertions, assertion):
|
||||
for existing in assertions:
|
||||
if existing == assertion:
|
||||
return
|
||||
assertions.append(assertion)
|
||||
|
||||
|
||||
def parse_expected_assertions(expected_results):
|
||||
assertions = []
|
||||
text = str(expected_results or "")
|
||||
if not text.strip():
|
||||
return assertions
|
||||
|
||||
status_match = re.search(r"(?:HTTP)?\s*状态码\s*(?:为|是|=|等于)?\s*(\d{3})", text, re.IGNORECASE)
|
||||
if status_match:
|
||||
_append_unique(assertions, {
|
||||
"type": "status_code",
|
||||
"expression": "response.status_code == {0}".format(status_match.group(1)),
|
||||
"source": status_match.group(0)
|
||||
})
|
||||
|
||||
for match in re.finditer(r"(?:返回)?\s*code\s*(?:为|是|=|等于)\s*([0-9]+)", text, re.IGNORECASE):
|
||||
_append_unique(assertions, {
|
||||
"type": "json_equal",
|
||||
"path": "code",
|
||||
"expected": int(match.group(1)),
|
||||
"expression": "response_json.get('code') == {0}".format(match.group(1)),
|
||||
"source": match.group(0)
|
||||
})
|
||||
|
||||
for match in re.finditer(r"(?:返回)?\s*(?:message|msg)\s*(?:为|是|=|等于)\s*['\"]?([^'\",,。;;\n\r]+)['\"]?", text, re.IGNORECASE):
|
||||
expected = match.group(1).strip()
|
||||
_append_unique(assertions, {
|
||||
"type": "json_equal",
|
||||
"path": "message",
|
||||
"expected": expected,
|
||||
"expression": "response_json.get('message') == {0!r}".format(expected),
|
||||
"source": match.group(0)
|
||||
})
|
||||
|
||||
field_patterns = [
|
||||
r"返回\s*([A-Za-z0-9_.]+)\s*字段",
|
||||
r"包含\s*([A-Za-z0-9_.]+)\s*字段",
|
||||
r"([A-Za-z0-9_.]*(?:token|accessToken|data|id|list|records|total)[A-Za-z0-9_.]*)\s*(?:不为空|存在)",
|
||||
]
|
||||
for pattern in field_patterns:
|
||||
for match in re.finditer(pattern, text, re.IGNORECASE):
|
||||
path = match.group(1).strip(" .")
|
||||
if not path:
|
||||
continue
|
||||
if path.lower() in ("http", "code", "message", "msg"):
|
||||
continue
|
||||
if "." not in path and path.lower() in ("token", "accesstoken"):
|
||||
path = "data.{0}".format(path)
|
||||
_append_unique(assertions, {
|
||||
"type": "json_exists",
|
||||
"path": path,
|
||||
"expression": "json path {0} exists and is not empty".format(path),
|
||||
"source": match.group(0)
|
||||
})
|
||||
|
||||
if ("成功" in text or "正常" in text) and not any(item.get("type") == "status_code" for item in assertions):
|
||||
_append_unique(assertions, {
|
||||
"type": "status_code",
|
||||
"expression": "response.status_code == 200",
|
||||
"source": "成功/正常"
|
||||
})
|
||||
|
||||
if ("登录成功" in text or "token" in text.lower()) and not any(item.get("path") in ("data.token", "token", "data.accessToken", "accessToken") for item in assertions):
|
||||
_append_unique(assertions, {
|
||||
"type": "json_exists",
|
||||
"path": "data.token",
|
||||
"expression": "json path data.token exists and is not empty",
|
||||
"source": "登录成功/token"
|
||||
})
|
||||
|
||||
return assertions
|
||||
@@ -0,0 +1,453 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
|
||||
import requests
|
||||
|
||||
from base_framework.platform_tools.Create_api_testcase.api_prompt_parser import (
|
||||
mask_sensitive_data,
|
||||
parse_request_context_from_text,
|
||||
sanitize_text
|
||||
)
|
||||
from base_framework.platform_tools.Create_api_testcase.assertion_parser import parse_expected_assertions
|
||||
|
||||
|
||||
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
PROJECT_ROOT = os.path.abspath(os.path.join(CURRENT_DIR, "..", "..", ".."))
|
||||
LOCAL_CONFIG_PATH = os.path.join(CURRENT_DIR, "config.json")
|
||||
SHARED_CONFIG_PATH = os.path.join(CURRENT_DIR, "..", "Create_ui_testcase", "config.json")
|
||||
SKILL_PATH = os.path.join(CURRENT_DIR, "api_testing_skill.md")
|
||||
GENERATED_CASES_DIR = os.path.join(CURRENT_DIR, "generated_cases")
|
||||
|
||||
DEFAULT_HEADERS = {
|
||||
"Accept": "application/json, text/plain, */*",
|
||||
"Accept-Language": "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7",
|
||||
"Cache-Control": "no-cache",
|
||||
"Connection": "keep-alive",
|
||||
"Content-Type": "application/json;charset=UTF-8",
|
||||
"Pragma": "no-cache",
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
|
||||
}
|
||||
|
||||
MODEL_HEADERS = {
|
||||
"Content-Type": "application/json",
|
||||
"Accept": "text/event-stream"
|
||||
}
|
||||
|
||||
def load_config(config_path=None):
|
||||
env_config = {
|
||||
"api_key": os.getenv("ROUTIN_API_KEY"),
|
||||
"base_url": os.getenv("ROUTIN_BASE_URL"),
|
||||
"model": os.getenv("ROUTIN_MODEL")
|
||||
}
|
||||
if all(env_config.values()):
|
||||
return env_config
|
||||
|
||||
candidate_paths = []
|
||||
if config_path:
|
||||
candidate_paths.append(config_path)
|
||||
candidate_paths.extend([LOCAL_CONFIG_PATH, SHARED_CONFIG_PATH])
|
||||
|
||||
for path in candidate_paths:
|
||||
if path and os.path.exists(path):
|
||||
with open(path, "r", encoding="utf-8") as file:
|
||||
config = json.load(file)
|
||||
for key, value in env_config.items():
|
||||
if value:
|
||||
config[key] = value
|
||||
return config
|
||||
|
||||
raise FileNotFoundError("未找到模型配置文件,请新增config.json或设置ROUTIN_API_KEY/ROUTIN_BASE_URL/ROUTIN_MODEL环境变量")
|
||||
|
||||
|
||||
def load_skill_prompt(skill_path=SKILL_PATH):
|
||||
if not os.path.exists(skill_path):
|
||||
raise FileNotFoundError("接口自动化Skill文件不存在:{0}".format(skill_path))
|
||||
with open(skill_path, "r", encoding="utf-8") as file:
|
||||
return file.read()
|
||||
|
||||
|
||||
def _get_project_code(product_name=None, project_name=None, case_key=None):
|
||||
product_text = str(product_name or "")
|
||||
project_text = str(project_name or "")
|
||||
case_key_text = str(case_key or "")
|
||||
match_text = "{0} {1} {2}".format(product_text, project_text, case_key_text).lower()
|
||||
|
||||
if "智慧运营" in product_text or "智慧运营" in project_text or "zhyy" in match_text or "zzyy" in match_text:
|
||||
return "zhyy"
|
||||
|
||||
if "joyhub_backend" in match_text or "hubops" in match_text or "joyhub backend" in match_text:
|
||||
return "joyhub_backend"
|
||||
|
||||
if "独立站" in product_text or "独立站" in project_text or "joyhub" in match_text or "dulizhan" in match_text:
|
||||
return "dulizhan"
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def _sanitize_path_part(value, default_value="unknown"):
|
||||
text = str(value or "").strip()
|
||||
if not text:
|
||||
text = default_value
|
||||
return re.sub(r'[\\/:*?"<>|\s]+', "_", text).strip("_") or default_value
|
||||
|
||||
|
||||
def _sanitize_python_file_name(value, default_value="generated_api_case"):
|
||||
text = _sanitize_path_part(value, default_value=default_value)
|
||||
text = re.sub(r"[^0-9A-Za-z_\u4e00-\u9fa5]+", "_", text).strip("_")
|
||||
if not text:
|
||||
text = default_value
|
||||
if not text.startswith("test_"):
|
||||
text = "test_{0}".format(text)
|
||||
return "{0}.py".format(text)
|
||||
|
||||
|
||||
def _resolve_api_testcase_dir(product_name, project_name, case_key, module_name):
|
||||
project_code = _get_project_code(
|
||||
product_name=product_name,
|
||||
project_name=project_name,
|
||||
case_key=case_key
|
||||
)
|
||||
safe_module_name = _sanitize_path_part(module_name, default_value="Generated")
|
||||
if project_code == "joyhub_backend":
|
||||
return os.path.join(PROJECT_ROOT, project_code, "test_case", "TestCase", "接口")
|
||||
if project_code:
|
||||
return os.path.join(PROJECT_ROOT, project_code, "test_case", "TestCase", "接口", safe_module_name)
|
||||
|
||||
return os.path.join(
|
||||
GENERATED_CASES_DIR,
|
||||
_sanitize_path_part(project_name, default_value="unknown_project"),
|
||||
safe_module_name
|
||||
)
|
||||
|
||||
|
||||
def build_api_testcase_prompt(project_id,
|
||||
case_id,
|
||||
automation_type,
|
||||
prompt,
|
||||
case_key,
|
||||
module_name,
|
||||
product_name,
|
||||
project_name,
|
||||
steps,
|
||||
expected_results,
|
||||
parsed_api_context=None,
|
||||
assertion_suggestions=None,
|
||||
skill_prompt=None):
|
||||
if skill_prompt is None:
|
||||
skill_prompt = load_skill_prompt()
|
||||
|
||||
if parsed_api_context is None:
|
||||
parsed_api_context = parse_request_context_from_text(prompt, steps, expected_results)
|
||||
if assertion_suggestions is None:
|
||||
assertion_suggestions = parse_expected_assertions(expected_results)
|
||||
|
||||
safe_parsed_api_context = mask_sensitive_data(parsed_api_context)
|
||||
|
||||
case_info = {
|
||||
"projectId": project_id,
|
||||
"caseId": case_id,
|
||||
"automationType": automation_type,
|
||||
"caseKey": case_key,
|
||||
"moduleName": module_name,
|
||||
"productName": product_name,
|
||||
"projectName": project_name,
|
||||
"steps": steps,
|
||||
"expectedResults": expected_results,
|
||||
"userPrompt": sanitize_text(prompt),
|
||||
"parsedApiContext": safe_parsed_api_context,
|
||||
"assertionSuggestions": assertion_suggestions
|
||||
}
|
||||
|
||||
return """你需要严格遵循以下接口自动化测试生成规则:
|
||||
|
||||
{skill_prompt}
|
||||
|
||||
下面是测试平台传入的测试用例信息:
|
||||
|
||||
{case_info}
|
||||
|
||||
生成要求:
|
||||
1. 只根据用户prompt中明确提供的接口、请求参数、前置参数、后置处理和预期结果生成 Python 接口自动化 pytest 用例。
|
||||
2. 代码风格贴合当前项目已有接口用例:requests + pytest + allure + logging。
|
||||
3. 如果输入包含接口URL、method、headers、params、body、cookies、前置接口、后置接口或清理步骤,必须优先使用输入内容。
|
||||
4. 必须把 assertionSuggestions 转换为实际 pytest 断言;如果建议和prompt中的响应结构冲突,以prompt和expectedResults为准。
|
||||
5. 如果缺少必要信息,不要自动联想、不要编造真实地址、token、账号或密码,使用TODO常量并通过pytest.skip提示补充。
|
||||
6. 敏感字段不要明文写入Allure附件或日志。
|
||||
7. 只输出Python代码,不要输出Markdown解释。
|
||||
""".format(
|
||||
skill_prompt=skill_prompt,
|
||||
case_info=json.dumps(case_info, ensure_ascii=False, indent=2)
|
||||
)
|
||||
|
||||
|
||||
def build_generate_automation_payload(project_id,
|
||||
case_id,
|
||||
automation_type,
|
||||
prompt,
|
||||
case_key,
|
||||
module_name,
|
||||
product_name,
|
||||
project_name,
|
||||
steps,
|
||||
expected_results,
|
||||
extra_fields=None):
|
||||
payload = {
|
||||
"projectId": project_id,
|
||||
"caseId": case_id,
|
||||
"automationType": automation_type,
|
||||
"prompt": prompt,
|
||||
"caseKey": case_key,
|
||||
"moduleName": module_name,
|
||||
"productName": product_name,
|
||||
"projectName": project_name,
|
||||
"steps": steps,
|
||||
"expectedResults": expected_results
|
||||
}
|
||||
if isinstance(extra_fields, dict):
|
||||
payload.update(extra_fields)
|
||||
return payload
|
||||
|
||||
|
||||
def extract_streaming_response_text(stream_text):
|
||||
deltas = []
|
||||
done_text = ""
|
||||
|
||||
for line in stream_text.splitlines():
|
||||
line = line.strip()
|
||||
if not line.startswith("data:"):
|
||||
continue
|
||||
|
||||
payload_text = line.replace("data:", "", 1).strip()
|
||||
if not payload_text or payload_text == "[DONE]":
|
||||
continue
|
||||
|
||||
try:
|
||||
payload = json.loads(payload_text)
|
||||
except ValueError:
|
||||
continue
|
||||
|
||||
event_type = payload.get("type")
|
||||
if event_type == "response.output_text.delta":
|
||||
delta = payload.get("delta")
|
||||
if isinstance(delta, str):
|
||||
deltas.append(delta)
|
||||
elif event_type == "response.output_text.done":
|
||||
text = payload.get("text")
|
||||
if isinstance(text, str) and text.strip():
|
||||
done_text = text
|
||||
|
||||
generated_text = "".join(deltas).strip()
|
||||
if generated_text:
|
||||
return generated_text
|
||||
|
||||
return done_text.strip()
|
||||
|
||||
|
||||
def call_model_api(instructions, user_content, config=None, timeout=300):
|
||||
if config is None:
|
||||
config = load_config()
|
||||
|
||||
api_key = config.get("api_key")
|
||||
base_url = config.get("base_url")
|
||||
model = config.get("model")
|
||||
|
||||
if not api_key:
|
||||
raise ValueError("api_key不能为空,请先配置config.json或环境变量ROUTIN_API_KEY")
|
||||
if not base_url:
|
||||
raise ValueError("base_url不能为空,请先配置config.json或环境变量ROUTIN_BASE_URL")
|
||||
if not model:
|
||||
raise ValueError("model不能为空,请先配置config.json或环境变量ROUTIN_MODEL")
|
||||
|
||||
headers = MODEL_HEADERS.copy()
|
||||
headers["Authorization"] = "Bearer {0}".format(api_key)
|
||||
|
||||
payload = {
|
||||
"model": model,
|
||||
"instructions": instructions,
|
||||
"input": user_content,
|
||||
"max_output_tokens": 4096,
|
||||
"store": False,
|
||||
"stream": True
|
||||
}
|
||||
|
||||
api_url = base_url.rstrip("/")
|
||||
if not api_url.endswith("/responses"):
|
||||
api_url = api_url + "/responses"
|
||||
|
||||
response = requests.post(
|
||||
url=api_url,
|
||||
headers=headers,
|
||||
json=payload,
|
||||
timeout=timeout
|
||||
)
|
||||
|
||||
if response.status_code >= 400:
|
||||
raise RuntimeError(
|
||||
"大模型Responses接口调用失败,url={0},status_code={1},response={2}".format(
|
||||
api_url,
|
||||
response.status_code,
|
||||
response.text[:2000]
|
||||
)
|
||||
)
|
||||
|
||||
generated_text = extract_streaming_response_text(response.text)
|
||||
if generated_text:
|
||||
return generated_text
|
||||
|
||||
try:
|
||||
return response.json()
|
||||
except ValueError:
|
||||
raise RuntimeError("大模型流式响应未解析到正文,响应预览:{0}".format(response.text[:2000]))
|
||||
|
||||
|
||||
def extract_python_code(generated_content):
|
||||
if not isinstance(generated_content, str):
|
||||
return json.dumps(generated_content, ensure_ascii=False, indent=2)
|
||||
|
||||
matches = re.findall(r"```(?:python|py)?\s*([\s\S]*?)```", generated_content, re.IGNORECASE)
|
||||
python_blocks = []
|
||||
for block in matches:
|
||||
block_text = block.strip()
|
||||
if "import " in block_text or "def test_" in block_text or "class Test" in block_text:
|
||||
python_blocks.append(block_text)
|
||||
|
||||
if python_blocks:
|
||||
return "\n\n".join(python_blocks).strip() + "\n"
|
||||
|
||||
return generated_content.strip() + "\n"
|
||||
|
||||
|
||||
def _next_available_file_path(target_dir, file_name):
|
||||
file_path = os.path.join(target_dir, file_name)
|
||||
if not os.path.exists(file_path):
|
||||
return file_path
|
||||
|
||||
base_name, ext = os.path.splitext(file_name)
|
||||
index = 1
|
||||
while True:
|
||||
candidate = os.path.join(target_dir, "{0}_{1}{2}".format(base_name, index, ext))
|
||||
if not os.path.exists(candidate):
|
||||
return candidate
|
||||
index += 1
|
||||
|
||||
|
||||
def save_generated_api_testcase(generated_content,
|
||||
product_name,
|
||||
project_name,
|
||||
module_name,
|
||||
case_key):
|
||||
target_dir = _resolve_api_testcase_dir(
|
||||
product_name=product_name,
|
||||
project_name=project_name,
|
||||
case_key=case_key,
|
||||
module_name=module_name
|
||||
)
|
||||
os.makedirs(target_dir, exist_ok=True)
|
||||
|
||||
file_name = _sanitize_python_file_name(case_key, default_value="generated_api_case")
|
||||
file_path = _next_available_file_path(target_dir, file_name)
|
||||
testcase_code = extract_python_code(generated_content)
|
||||
|
||||
with open(file_path, "w", encoding="utf-8") as file:
|
||||
file.write(testcase_code)
|
||||
|
||||
return file_path
|
||||
|
||||
|
||||
def generate_api_automation_testcase(project_id,
|
||||
case_id,
|
||||
automation_type,
|
||||
prompt,
|
||||
case_key,
|
||||
module_name,
|
||||
product_name,
|
||||
project_name,
|
||||
steps,
|
||||
expected_results,
|
||||
config=None,
|
||||
skill_prompt=None,
|
||||
timeout=120):
|
||||
if automation_type not in ("api", "interface", "接口"):
|
||||
raise ValueError("automation_type必须为api/interface/接口")
|
||||
|
||||
parsed_api_context = parse_request_context_from_text(prompt, steps, expected_results)
|
||||
assertion_suggestions = parse_expected_assertions(expected_results)
|
||||
|
||||
final_prompt = build_api_testcase_prompt(
|
||||
project_id=project_id,
|
||||
case_id=case_id,
|
||||
automation_type=automation_type,
|
||||
prompt=prompt,
|
||||
case_key=case_key,
|
||||
module_name=module_name,
|
||||
product_name=product_name,
|
||||
project_name=project_name,
|
||||
steps=steps,
|
||||
expected_results=expected_results,
|
||||
parsed_api_context=parsed_api_context,
|
||||
assertion_suggestions=assertion_suggestions,
|
||||
skill_prompt=skill_prompt
|
||||
)
|
||||
|
||||
instructions = "你是资深接口自动化测试专家,负责生成稳定、可维护、可落地的Python pytest接口自动化测试用例。"
|
||||
|
||||
return call_model_api(
|
||||
instructions=instructions,
|
||||
user_content=final_prompt,
|
||||
config=config,
|
||||
timeout=timeout
|
||||
)
|
||||
|
||||
|
||||
def generate_automation_case(url,
|
||||
project_id,
|
||||
case_id,
|
||||
automation_type,
|
||||
prompt,
|
||||
case_key,
|
||||
module_name,
|
||||
product_name,
|
||||
project_name,
|
||||
steps,
|
||||
expected_results,
|
||||
access_token,
|
||||
cookie=None,
|
||||
timeout=60,
|
||||
extra_fields=None):
|
||||
if not url:
|
||||
raise ValueError("url不能为空,请由调用方传入生成自动化用例接口地址")
|
||||
|
||||
headers = DEFAULT_HEADERS.copy()
|
||||
headers["accessToken"] = access_token
|
||||
|
||||
if cookie:
|
||||
headers["Cookie"] = cookie
|
||||
|
||||
payload = build_generate_automation_payload(
|
||||
project_id=project_id,
|
||||
case_id=case_id,
|
||||
automation_type=automation_type,
|
||||
prompt=prompt,
|
||||
case_key=case_key,
|
||||
module_name=module_name,
|
||||
product_name=product_name,
|
||||
project_name=project_name,
|
||||
steps=steps,
|
||||
expected_results=expected_results,
|
||||
extra_fields=extra_fields
|
||||
)
|
||||
|
||||
response = requests.post(
|
||||
url=url,
|
||||
headers=headers,
|
||||
data=json.dumps(payload, ensure_ascii=False).encode("utf-8"),
|
||||
timeout=timeout
|
||||
)
|
||||
response.raise_for_status()
|
||||
|
||||
try:
|
||||
return response.json()
|
||||
except ValueError:
|
||||
return response.text
|
||||
165
base_framework/platform_tools/Create_api_testcase/server.py
Normal file
165
base_framework/platform_tools/Create_api_testcase/server.py
Normal file
@@ -0,0 +1,165 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import argparse
|
||||
import json
|
||||
import traceback
|
||||
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from base_framework.platform_tools.Create_api_testcase.generate_api_automation import (
|
||||
generate_api_automation_testcase,
|
||||
save_generated_api_testcase
|
||||
)
|
||||
|
||||
|
||||
API_PATH = "/it/api/case/generate-automation"
|
||||
SUPPORTED_AUTOMATION_TYPES = ("api", "interface", "接口")
|
||||
|
||||
|
||||
class GenerateApiAutomationHandler(BaseHTTPRequestHandler):
|
||||
|
||||
def do_OPTIONS(self):
|
||||
self._send_json_response(200, {"code": 0, "message": "ok", "data": None})
|
||||
|
||||
def do_POST(self):
|
||||
request_path = urlparse(self.path).path
|
||||
if request_path != API_PATH:
|
||||
self._send_json_response(404, {
|
||||
"code": 404,
|
||||
"message": "接口不存在:{0}".format(request_path),
|
||||
"data": None
|
||||
})
|
||||
return
|
||||
|
||||
try:
|
||||
request_data = self._read_json_body()
|
||||
self._validate_required_fields(request_data)
|
||||
|
||||
automation_type = request_data.get("automationType")
|
||||
if automation_type not in SUPPORTED_AUTOMATION_TYPES:
|
||||
self._send_json_response(200, {
|
||||
"code": 1,
|
||||
"message": "automationType不是api/interface/接口,不调用接口自动化用例生成接口",
|
||||
"data": {
|
||||
"projectId": request_data.get("projectId"),
|
||||
"caseId": request_data.get("caseId"),
|
||||
"automationType": automation_type,
|
||||
"caseKey": request_data.get("caseKey"),
|
||||
"content": self._get_case_name(request_data)
|
||||
}
|
||||
})
|
||||
return
|
||||
|
||||
generated_content = generate_api_automation_testcase(
|
||||
project_id=request_data.get("projectId"),
|
||||
case_id=request_data.get("caseId"),
|
||||
automation_type=automation_type,
|
||||
prompt=request_data.get("prompt"),
|
||||
case_key=request_data.get("caseKey"),
|
||||
module_name=request_data.get("moduleName"),
|
||||
product_name=request_data.get("productName"),
|
||||
project_name=request_data.get("projectName"),
|
||||
steps=request_data.get("steps"),
|
||||
expected_results=request_data.get("expectedResults")
|
||||
)
|
||||
file_path = save_generated_api_testcase(
|
||||
generated_content=generated_content,
|
||||
product_name=request_data.get("productName"),
|
||||
project_name=request_data.get("projectName"),
|
||||
module_name=request_data.get("moduleName"),
|
||||
case_key=request_data.get("caseKey")
|
||||
)
|
||||
|
||||
self._send_json_response(200, {
|
||||
"code": 0,
|
||||
"message": "success",
|
||||
"data": {
|
||||
"projectId": request_data.get("projectId"),
|
||||
"caseId": request_data.get("caseId"),
|
||||
"automationType": automation_type,
|
||||
"caseKey": request_data.get("caseKey"),
|
||||
"content": self._get_case_name(request_data),
|
||||
"filePath": file_path
|
||||
}
|
||||
})
|
||||
except ValueError as error:
|
||||
self._send_json_response(400, {
|
||||
"code": 400,
|
||||
"message": str(error),
|
||||
"data": None
|
||||
})
|
||||
except Exception as error:
|
||||
self._send_json_response(500, {
|
||||
"code": 500,
|
||||
"message": str(error),
|
||||
"data": None,
|
||||
"trace": traceback.format_exc()
|
||||
})
|
||||
|
||||
def _read_json_body(self):
|
||||
content_length = int(self.headers.get("Content-Length", 0))
|
||||
if content_length <= 0:
|
||||
raise ValueError("请求体不能为空")
|
||||
|
||||
body = self.rfile.read(content_length).decode("utf-8")
|
||||
try:
|
||||
return json.loads(body)
|
||||
except ValueError:
|
||||
raise ValueError("请求体必须是合法JSON")
|
||||
|
||||
def _validate_required_fields(self, request_data):
|
||||
required_fields = [
|
||||
"projectId",
|
||||
"caseId",
|
||||
"automationType",
|
||||
"prompt",
|
||||
"caseKey",
|
||||
"moduleName",
|
||||
"productName",
|
||||
"projectName",
|
||||
"steps",
|
||||
"expectedResults"
|
||||
]
|
||||
missing_fields = []
|
||||
for field in required_fields:
|
||||
if field == "productName":
|
||||
if field not in request_data or request_data.get(field) is None:
|
||||
missing_fields.append(field)
|
||||
elif request_data.get(field) in (None, ""):
|
||||
missing_fields.append(field)
|
||||
if missing_fields:
|
||||
raise ValueError("缺少必填参数:{0}".format(", ".join(missing_fields)))
|
||||
|
||||
def _get_case_name(self, request_data):
|
||||
return request_data.get("caseName") or request_data.get("title") or request_data.get("name") or "{0}-{1}".format(
|
||||
request_data.get("caseKey"),
|
||||
request_data.get("moduleName")
|
||||
)
|
||||
|
||||
def _send_json_response(self, status_code, response_data):
|
||||
response_body = json.dumps(response_data, ensure_ascii=False).encode("utf-8")
|
||||
self.send_response(status_code)
|
||||
self.send_header("Content-Type", "application/json;charset=UTF-8")
|
||||
self.send_header("Access-Control-Allow-Origin", "*")
|
||||
self.send_header("Access-Control-Allow-Methods", "POST, OPTIONS")
|
||||
self.send_header("Access-Control-Allow-Headers", "Content-Type, accessToken, Authorization")
|
||||
self.send_header("Content-Length", str(len(response_body)))
|
||||
self.end_headers()
|
||||
self.wfile.write(response_body)
|
||||
|
||||
def log_message(self, format, *args):
|
||||
print("[{0}] {1}".format(self.log_date_time_string(), format % args))
|
||||
|
||||
|
||||
def run_server(host="0.0.0.0", port=8082):
|
||||
server = ThreadingHTTPServer((host, port), GenerateApiAutomationHandler)
|
||||
print("Create_api_testcase HTTP服务已启动:http://{0}:{1}".format(host, port))
|
||||
print("接口地址:POST {0}".format(API_PATH))
|
||||
server.serve_forever()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="接口自动化用例生成HTTP服务")
|
||||
parser.add_argument("--host", default="0.0.0.0", help="服务监听地址,默认0.0.0.0")
|
||||
parser.add_argument("--port", default=8082, type=int, help="服务监听端口,默认8082")
|
||||
args = parser.parse_args()
|
||||
run_server(host=args.host, port=args.port)
|
||||
@@ -0,0 +1 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,600 @@
|
||||
# Chat Completions API
|
||||
|
||||
**Source**: https://docs.routin.ai/zh/docs/API/chat-completions
|
||||
**Description**: Chat Completions API 使用指南和多语言示例
|
||||
|
||||
---
|
||||
|
||||
[Routin AI](https://docs.routin.ai/zh/docs/API/</>)
|
||||
|
||||
[Routin AI](https://docs.routin.ai/zh/docs/API/</>)
|
||||
|
||||
搜索
|
||||
|
||||
`⌘``K`
|
||||
|
||||
🚀 欢迎使用 Routin AI 文档!
|
||||
|
||||
[前往控制台](https://docs.routin.ai/zh/docs/API/<https:/routin.ai/dashboard>)[欢迎使用 Routin AI](https://docs.routin.ai/zh/docs/API/</zh/docs>)
|
||||
|
||||
API 文档
|
||||
|
||||
[Chat Completions API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/chat-completions>)[Embeddings API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/embeddings>)[Images API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/images>)[Audio API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/audio>)[Messages API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/messages>)[Gemini API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/gemini>)[Videos API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/video>)[Web Research API](https://docs.routin.ai/zh/docs/API/</zh/docs/API/web>)
|
||||
|
||||
设计方案
|
||||
|
||||
[套餐订阅制设计](https://docs.routin.ai/zh/docs/API/</zh/docs/Design/plan-subscription>)
|
||||
|
||||
开发者工具
|
||||
|
||||
[接入 Claude Code 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-claude-code>)[接入 Codex 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-codex>)[接入 Kilo Code 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-kilo-code>)[接入 Cherry Studio 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-cherry-studio>)[接入 Gemini CLI 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-gemini>)[Claude Code 完整使用教程](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-claude-code-advanced>)[接入 OpenCode 使用](https://docs.routin.ai/zh/docs/API/</zh/docs/DeveloperTools/access-open-code>)
|
||||
|
||||
Organization
|
||||
|
||||
© 2026 Routin AI
|
||||
|
||||
Chat Completions API
|
||||
|
||||
API 文档
|
||||
|
||||
# Chat Completions API
|
||||
|
||||
Chat Completions API 使用指南和多语言示例
|
||||
|
||||
# Chat Completions API
|
||||
|
||||
MeteorAI 提供完全兼容 OpenAI 的对话接口,您可以使用 OpenAI SDK 直接调用我们的服务。
|
||||
|
||||
## 基本信息
|
||||
|
||||
**API 端点**
|
||||
[code]
|
||||
https://api.routin.ai/v1/chat/completions
|
||||
[/code]
|
||||
|
||||
**认证方式** 在请求头中添加 API Key:
|
||||
[code]
|
||||
Authorization: Bearer YOUR_API_KEY
|
||||
[/code]
|
||||
|
||||
MeteorAI 完全兼容 OpenAI SDK,只需修改 `base_url` 参数即可无缝切换。
|
||||
|
||||
## 请求参数
|
||||
|
||||
### 必需参数
|
||||
|
||||
参数| 类型| 说明
|
||||
---|---|---
|
||||
`model`| string| 模型名称,如 `gpt-4o`、`claude-3-5-sonnet-20241022` 等
|
||||
`messages`| array| 对话消息数组
|
||||
|
||||
### 可选参数
|
||||
|
||||
参数| 类型| 默认值| 说明
|
||||
---|---|---|---
|
||||
`temperature`| number| 1| 采样温度 (0-2)
|
||||
`top_p`| number| 1| 核采样参数 (0-1)
|
||||
`max_tokens`| integer| -| 生成的最大 token 数
|
||||
`stream`| boolean| false| 是否使用流式输出
|
||||
`presence_penalty`| number| 0| 存在惩罚 (-2.0 到 2.0)
|
||||
`frequency_penalty`| number| 0| 频率惩罚 (-2.0 到 2.0)
|
||||
`user`| string| -| 用户标识符
|
||||
|
||||
### Messages 格式
|
||||
[code]
|
||||
{
|
||||
"messages": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a helpful assistant."
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": "Hello!"
|
||||
}
|
||||
]
|
||||
}
|
||||
[/code]
|
||||
|
||||
支持的 `role` 值:
|
||||
|
||||
* `system`: 系统消息,定义助手行为
|
||||
* `user`: 用户消息
|
||||
* `assistant`: 助手回复
|
||||
|
||||
## 响应格式
|
||||
|
||||
### 普通响应
|
||||
[code]
|
||||
{
|
||||
"id": "chatcmpl-123",
|
||||
"object": "chat.completion",
|
||||
"created": 1677652288,
|
||||
"model": "gpt-4o",
|
||||
"choices": [
|
||||
{
|
||||
"index": 0,
|
||||
"message": {
|
||||
"role": "assistant",
|
||||
"content": "Hello! How can I help you today?"
|
||||
},
|
||||
"finish_reason": "stop"
|
||||
}
|
||||
],
|
||||
"usage": {
|
||||
"prompt_tokens": 9,
|
||||
"completion_tokens": 12,
|
||||
"total_tokens": 21
|
||||
}
|
||||
}
|
||||
[/code]
|
||||
|
||||
### 流式响应
|
||||
[code]
|
||||
data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1677652288,"model":"gpt-4o","choices":[{"index":0,"delta":{"role":"assistant","content":"Hello"},"finish_reason":null}]}
|
||||
|
||||
data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1677652288,"model":"gpt-4o","choices":[{"index":0,"delta":{"content":"!"},"finish_reason":null}]}
|
||||
|
||||
data: {"id":"chatcmpl-123","object":"chat.completion.chunk","created":1677652288,"model":"gpt-4o","choices":[{"index":0,"delta":{},"finish_reason":"stop"}]}
|
||||
|
||||
data: [DONE]
|
||||
[/code]
|
||||
|
||||
## 代码示例
|
||||
|
||||
### 基本调用
|
||||
|
||||
PythonTypeScriptJavaScriptC#cURL
|
||||
[code]
|
||||
from openai import OpenAI
|
||||
|
||||
client = OpenAI(
|
||||
api_key="YOUR_API_KEY",
|
||||
base_url="https://api.routin.ai/v1"
|
||||
)
|
||||
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=[
|
||||
{"role": "system", "content": "You are a helpful assistant."},
|
||||
{"role": "user", "content": "你好,介绍一下你自己"}
|
||||
],
|
||||
temperature=0.7,
|
||||
max_tokens=150
|
||||
)
|
||||
|
||||
print(response.choices[0].message.content)
|
||||
[/code]
|
||||
[code]
|
||||
import OpenAI from 'openai';
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
async function main() {
|
||||
const response = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages: [
|
||||
{ role: 'system', content: 'You are a helpful assistant.' },
|
||||
{ role: 'user', content: '你好,介绍一下你自己' },
|
||||
],
|
||||
temperature: 0.7,
|
||||
max_tokens: 150,
|
||||
});
|
||||
|
||||
console.log(response.choices[0].message.content);
|
||||
}
|
||||
|
||||
main();
|
||||
[/code]
|
||||
[code]
|
||||
const OpenAI = require('openai');
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages: [
|
||||
{ role: 'system', content: 'You are a helpful assistant.' },
|
||||
{ role: 'user', content: '你好,介绍一下你自己' },
|
||||
],
|
||||
temperature: 0.7,
|
||||
max_tokens: 150,
|
||||
}).then(response => {
|
||||
console.log(response.choices[0].message.content);
|
||||
});
|
||||
[/code]
|
||||
[code]
|
||||
using OpenAI.Chat;
|
||||
|
||||
var client = new ChatClient(
|
||||
model: "gpt-4o",
|
||||
apiKey: "YOUR_API_KEY",
|
||||
new OpenAIClientOptions
|
||||
{
|
||||
Endpoint = new Uri("https://api.routin.ai/v1")
|
||||
}
|
||||
);
|
||||
|
||||
var messages = new List<ChatMessage>
|
||||
{
|
||||
new SystemChatMessage("You are a helpful assistant."),
|
||||
new UserChatMessage("你好,介绍一下你自己")
|
||||
};
|
||||
|
||||
var response = await client.CompleteChatAsync(
|
||||
messages,
|
||||
new ChatCompletionOptions
|
||||
{
|
||||
Temperature = 0.7f,
|
||||
MaxOutputTokenCount = 150
|
||||
}
|
||||
);
|
||||
|
||||
Console.WriteLine(response.Value.Content[0].Text);
|
||||
[/code]
|
||||
[code]
|
||||
curl https://api.routin.ai/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-H "Authorization: Bearer YOUR_API_KEY" \
|
||||
-d '{
|
||||
"model": "gpt-4o",
|
||||
"messages": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a helpful assistant."
|
||||
},
|
||||
{
|
||||
"role": "user",
|
||||
"content": "你好,介绍一下你自己"
|
||||
}
|
||||
],
|
||||
"temperature": 0.7,
|
||||
"max_tokens": 150
|
||||
}'
|
||||
[/code]
|
||||
|
||||
### 流式输出
|
||||
|
||||
流式输出可以实时获取模型的生成内容,提供更好的用户体验。
|
||||
|
||||
PythonTypeScriptJavaScriptC#cURL
|
||||
[code]
|
||||
from openai import OpenAI
|
||||
|
||||
client = OpenAI(
|
||||
api_key="YOUR_API_KEY",
|
||||
base_url="https://api.routin.ai/v1"
|
||||
)
|
||||
|
||||
stream = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=[
|
||||
{"role": "user", "content": "给我讲一个有趣的故事"}
|
||||
],
|
||||
stream=True
|
||||
)
|
||||
|
||||
for chunk in stream:
|
||||
if chunk.choices[0].delta.content is not None:
|
||||
print(chunk.choices[0].delta.content, end="")
|
||||
[/code]
|
||||
[code]
|
||||
import OpenAI from 'openai';
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
async function main() {
|
||||
const stream = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages: [{ role: 'user', content: '给我讲一个有趣的故事' }],
|
||||
stream: true,
|
||||
});
|
||||
|
||||
for await (const chunk of stream) {
|
||||
const content = chunk.choices[0]?.delta?.content;
|
||||
if (content) {
|
||||
process.stdout.write(content);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
main();
|
||||
[/code]
|
||||
[code]
|
||||
const OpenAI = require('openai');
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
async function main() {
|
||||
const stream = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages: [{ role: 'user', content: '给我讲一个有趣的故事' }],
|
||||
stream: true,
|
||||
});
|
||||
|
||||
for await (const chunk of stream) {
|
||||
const content = chunk.choices[0]?.delta?.content;
|
||||
if (content) {
|
||||
process.stdout.write(content);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
main();
|
||||
[/code]
|
||||
[code]
|
||||
using OpenAI.Chat;
|
||||
|
||||
var client = new ChatClient(
|
||||
model: "gpt-4o",
|
||||
apiKey: "YOUR_API_KEY",
|
||||
new OpenAIClientOptions
|
||||
{
|
||||
Endpoint = new Uri("https://api.routin.ai/v1")
|
||||
}
|
||||
);
|
||||
|
||||
var messages = new List<ChatMessage>
|
||||
{
|
||||
new UserChatMessage("给我讲一个有趣的故事")
|
||||
};
|
||||
|
||||
await foreach (var update in client.CompleteChatStreamingAsync(messages))
|
||||
{
|
||||
foreach (var contentPart in update.ContentUpdate)
|
||||
{
|
||||
Console.Write(contentPart.Text);
|
||||
}
|
||||
}
|
||||
[/code]
|
||||
[code]
|
||||
curl https://api.routin.ai/v1/chat/completions \
|
||||
-H "Content-Type: application/json" \
|
||||
-H "Authorization: Bearer YOUR_API_KEY" \
|
||||
-d '{
|
||||
"model": "gpt-4o",
|
||||
"messages": [
|
||||
{
|
||||
"role": "user",
|
||||
"content": "给我讲一个有趣的故事"
|
||||
}
|
||||
],
|
||||
"stream": true
|
||||
}'
|
||||
[/code]
|
||||
|
||||
### 多轮对话
|
||||
|
||||
PythonTypeScriptC#
|
||||
[code]
|
||||
from openai import OpenAI
|
||||
|
||||
client = OpenAI(
|
||||
api_key="YOUR_API_KEY",
|
||||
base_url="https://api.routin.ai/v1"
|
||||
)
|
||||
|
||||
messages = [
|
||||
{"role": "system", "content": "You are a helpful assistant."}
|
||||
]
|
||||
|
||||
# 第一轮对话
|
||||
messages.append({"role": "user", "content": "我叫张三"})
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=messages
|
||||
)
|
||||
assistant_message = response.choices[0].message.content
|
||||
messages.append({"role": "assistant", "content": assistant_message})
|
||||
print(f"助手: {assistant_message}")
|
||||
|
||||
# 第二轮对话
|
||||
messages.append({"role": "user", "content": "我叫什么名字?"})
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=messages
|
||||
)
|
||||
assistant_message = response.choices[0].message.content
|
||||
print(f"助手: {assistant_message}")
|
||||
[/code]
|
||||
[code]
|
||||
import OpenAI from 'openai';
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
async function main() {
|
||||
const messages: OpenAI.Chat.ChatCompletionMessageParam[] = [
|
||||
{ role: 'system', content: 'You are a helpful assistant.' },
|
||||
];
|
||||
|
||||
// 第一轮对话
|
||||
messages.push({ role: 'user', content: '我叫张三' });
|
||||
let response = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages,
|
||||
});
|
||||
let assistantMessage = response.choices[0].message.content;
|
||||
messages.push({ role: 'assistant', content: assistantMessage! });
|
||||
console.log(`助手: ${assistantMessage}`);
|
||||
|
||||
// 第二轮对话
|
||||
messages.push({ role: 'user', content: '我叫什么名字?' });
|
||||
response = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages,
|
||||
});
|
||||
assistantMessage = response.choices[0].message.content;
|
||||
console.log(`助手: ${assistantMessage}`);
|
||||
}
|
||||
|
||||
main();
|
||||
[/code]
|
||||
[code]
|
||||
using OpenAI.Chat;
|
||||
|
||||
var client = new ChatClient(
|
||||
model: "gpt-4o",
|
||||
apiKey: "YOUR_API_KEY",
|
||||
new OpenAIClientOptions
|
||||
{
|
||||
Endpoint = new Uri("https://api.routin.ai/v1")
|
||||
}
|
||||
);
|
||||
|
||||
var messages = new List<ChatMessage>
|
||||
{
|
||||
new SystemChatMessage("You are a helpful assistant.")
|
||||
};
|
||||
|
||||
// 第一轮对话
|
||||
messages.Add(new UserChatMessage("我叫张三"));
|
||||
var response = await client.CompleteChatAsync(messages);
|
||||
var assistantMessage = response.Value.Content[0].Text;
|
||||
messages.Add(new AssistantChatMessage(assistantMessage));
|
||||
Console.WriteLine($"助手: {assistantMessage}");
|
||||
|
||||
// 第二轮对话
|
||||
messages.Add(new UserChatMessage("我叫什么名字?"));
|
||||
response = await client.CompleteChatAsync(messages);
|
||||
assistantMessage = response.Value.Content[0].Text;
|
||||
Console.WriteLine($"助手: {assistantMessage}");
|
||||
[/code]
|
||||
|
||||
## 错误处理
|
||||
|
||||
请务必在生产环境中添加错误处理逻辑,避免因 API 调用失败导致应用崩溃。
|
||||
|
||||
PythonTypeScriptC#
|
||||
[code]
|
||||
from openai import OpenAI, APIError, RateLimitError, APIConnectionError
|
||||
|
||||
client = OpenAI(
|
||||
api_key="YOUR_API_KEY",
|
||||
base_url="https://api.routin.ai/v1"
|
||||
)
|
||||
|
||||
try:
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=[{"role": "user", "content": "Hello!"}]
|
||||
)
|
||||
print(response.choices[0].message.content)
|
||||
except RateLimitError as e:
|
||||
print(f"请求频率超限: {e}")
|
||||
except APIConnectionError as e:
|
||||
print(f"网络连接错误: {e}")
|
||||
except APIError as e:
|
||||
print(f"API 错误: {e}")
|
||||
except Exception as e:
|
||||
print(f"未知错误: {e}")
|
||||
[/code]
|
||||
[code]
|
||||
import OpenAI from 'openai';
|
||||
|
||||
const client = new OpenAI({
|
||||
apiKey: 'YOUR_API_KEY',
|
||||
baseURL: 'https://api.routin.ai/v1',
|
||||
});
|
||||
|
||||
async function main() {
|
||||
try {
|
||||
const response = await client.chat.completions.create({
|
||||
model: 'gpt-4o',
|
||||
messages: [{ role: 'user', content: 'Hello!' }],
|
||||
});
|
||||
console.log(response.choices[0].message.content);
|
||||
} catch (error) {
|
||||
if (error instanceof OpenAI.APIError) {
|
||||
console.error(`API 错误 [${error.status}]: ${error.message}`);
|
||||
} else if (error instanceof OpenAI.RateLimitError) {
|
||||
console.error('请求频率超限');
|
||||
} else {
|
||||
console.error('未知错误:', error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
main();
|
||||
[/code]
|
||||
[code]
|
||||
using OpenAI.Chat;
|
||||
using OpenAI;
|
||||
|
||||
var client = new ChatClient(
|
||||
model: "gpt-4o",
|
||||
apiKey: "YOUR_API_KEY",
|
||||
new OpenAIClientOptions
|
||||
{
|
||||
Endpoint = new Uri("https://api.routin.ai/v1")
|
||||
}
|
||||
);
|
||||
|
||||
try
|
||||
{
|
||||
var response = await client.CompleteChatAsync(
|
||||
new List<ChatMessage>
|
||||
{
|
||||
new UserChatMessage("Hello!")
|
||||
}
|
||||
);
|
||||
Console.WriteLine(response.Value.Content[0].Text);
|
||||
}
|
||||
catch (ClientResultException ex) when (ex.Status == 429)
|
||||
{
|
||||
Console.WriteLine($"请求频率超限: {ex.Message}");
|
||||
}
|
||||
catch (ClientResultException ex)
|
||||
{
|
||||
Console.WriteLine($"API 错误 [{ex.Status}]: {ex.Message}");
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine($"未知错误: {ex.Message}");
|
||||
}
|
||||
[/code]
|
||||
|
||||
## 常见错误码
|
||||
|
||||
错误码| 说明| 解决方法
|
||||
---|---|---
|
||||
401| API Key 无效或未提供| 检查 Authorization 头是否正确设置
|
||||
429| 请求频率超限| 降低请求频率或升级配额
|
||||
400| 请求参数错误| 检查请求参数格式是否正确
|
||||
500| 服务器内部错误| 稍后重试或联系技术支持
|
||||
503| 服务暂时不可用| 稍后重试
|
||||
|
||||
## 最佳实践
|
||||
|
||||
1. **使用系统消息** : 通过 `system` 角色定义助手的行为和特性
|
||||
2. **控制 token 数量** : 使用 `max_tokens` 参数控制生成长度,避免不必要的费用
|
||||
3. **错误重试** : 实现指数退避的重试机制,处理临时性错误
|
||||
4. **流式输出** : 对于长文本生成,使用 `stream=true` 提供更好的用户体验
|
||||
5. **保存对话历史** : 多轮对话需要在 `messages` 数组中包含完整的对话历史
|
||||
6. **监控使用情况** : 定期查看管理后台的统计信息,优化 API 使用
|
||||
|
||||
## 更多资源
|
||||
|
||||
* [Embeddings API](https://docs.routin.ai/zh/docs/API/</zh/API/embeddings>) \- 文本向量化接口
|
||||
* [Images API](https://docs.routin.ai/zh/docs/API/</zh/API/images>) \- 图像生成接口
|
||||
* [Audio API](https://docs.routin.ai/zh/docs/API/</zh/API/audio>) \- 语音识别和合成接口
|
||||
|
||||
[欢迎使用 Routin AI统一的大模型 API 聚合平台,提供企业级服务和管理能力](https://docs.routin.ai/zh/docs/API/</zh/docs>)[Embeddings API文本向量化 API 使用指南和多语言示例](https://docs.routin.ai/zh/docs/API/</zh/docs/API/embeddings>)
|
||||
|
||||
### On this page
|
||||
|
||||
Chat Completions API基本信息请求参数必需参数可选参数Messages 格式响应格式普通响应流式响应代码示例基本调用流式输出多轮对话错误处理常见错误码最佳实践更多资源
|
||||
177
base_framework/platform_tools/Create_ui_testcase/server.py
Normal file
177
base_framework/platform_tools/Create_ui_testcase/server.py
Normal file
@@ -0,0 +1,177 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import argparse
|
||||
import json
|
||||
import traceback
|
||||
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from base_framework.platform_tools.Create_ui_testcase.generate_automation_api import (
|
||||
generate_save_and_verify_ui_testcase
|
||||
)
|
||||
|
||||
|
||||
API_PATH = "/it/api/case/generate-automation"
|
||||
|
||||
|
||||
class GenerateAutomationHandler(BaseHTTPRequestHandler):
|
||||
|
||||
def do_OPTIONS(self):
|
||||
self._send_json_response(200, {"code": 0, "message": "ok", "data": None})
|
||||
|
||||
def do_POST(self):
|
||||
request_path = urlparse(self.path).path
|
||||
if request_path != API_PATH:
|
||||
self._send_json_response(404, {
|
||||
"code": 404,
|
||||
"message": "接口不存在:{0}".format(request_path),
|
||||
"data": None
|
||||
})
|
||||
return
|
||||
|
||||
try:
|
||||
request_data = self._read_json_body()
|
||||
self._validate_required_fields(request_data)
|
||||
|
||||
if request_data.get("automationType") != "ui":
|
||||
self._send_json_response(200, {
|
||||
"code": 1,
|
||||
"message": "automationType不是ui,不调用UI自动化用例生成接口",
|
||||
"data": {
|
||||
"projectId": request_data.get("projectId"),
|
||||
"caseId": request_data.get("caseId"),
|
||||
"automationType": request_data.get("automationType"),
|
||||
"caseKey": request_data.get("caseKey"),
|
||||
"content": self._get_case_name(request_data)
|
||||
}
|
||||
})
|
||||
return
|
||||
|
||||
verify_result = generate_save_and_verify_ui_testcase(
|
||||
project_id=request_data.get("projectId"),
|
||||
case_id=request_data.get("caseId"),
|
||||
automation_type=request_data.get("automationType"),
|
||||
prompt=request_data.get("prompt"),
|
||||
case_key=request_data.get("caseKey"),
|
||||
module_name=request_data.get("moduleName"),
|
||||
product_name=request_data.get("productName"),
|
||||
project_name=request_data.get("projectName"),
|
||||
steps=request_data.get("steps"),
|
||||
expected_results=request_data.get("expectedResults"),
|
||||
case_name=self._get_case_name(request_data),
|
||||
enable_reconnaissance=request_data.get("enableReconnaissance", True),
|
||||
headless=request_data.get("headless", True),
|
||||
max_attempts=request_data.get("maxAttempts", 3)
|
||||
)
|
||||
|
||||
if not verify_result.get("success"):
|
||||
self._send_json_response(200, {
|
||||
"code": 1,
|
||||
"message": "用例生成后执行失败,自动修复{0}次仍未通过".format(request_data.get("maxAttempts", 3)),
|
||||
"data": {
|
||||
"projectId": request_data.get("projectId"),
|
||||
"caseId": request_data.get("caseId"),
|
||||
"automationType": request_data.get("automationType"),
|
||||
"caseKey": request_data.get("caseKey"),
|
||||
"content": self._get_case_name(request_data),
|
||||
"testcasePath": verify_result.get("testcasePath"),
|
||||
"attempts": verify_result.get("attempts"),
|
||||
"failureReason": verify_result.get("failureReason")
|
||||
}
|
||||
})
|
||||
return
|
||||
|
||||
self._send_json_response(200, {
|
||||
"code": 0,
|
||||
"message": "success",
|
||||
"data": {
|
||||
"projectId": request_data.get("projectId"),
|
||||
"caseId": request_data.get("caseId"),
|
||||
"automationType": request_data.get("automationType"),
|
||||
"caseKey": request_data.get("caseKey"),
|
||||
"content": self._get_case_name(request_data),
|
||||
"testcasePath": verify_result.get("testcasePath"),
|
||||
"attempts": verify_result.get("attempts")
|
||||
}
|
||||
})
|
||||
except ValueError as error:
|
||||
self._send_json_response(400, {
|
||||
"code": 400,
|
||||
"message": str(error),
|
||||
"data": None
|
||||
})
|
||||
except Exception as error:
|
||||
self._send_json_response(500, {
|
||||
"code": 500,
|
||||
"message": str(error),
|
||||
"data": None,
|
||||
"trace": traceback.format_exc()
|
||||
})
|
||||
|
||||
def _read_json_body(self):
|
||||
content_length = int(self.headers.get("Content-Length", 0))
|
||||
if content_length <= 0:
|
||||
raise ValueError("请求体不能为空")
|
||||
|
||||
body = self.rfile.read(content_length).decode("utf-8")
|
||||
try:
|
||||
return json.loads(body)
|
||||
except ValueError:
|
||||
raise ValueError("请求体必须是合法JSON")
|
||||
|
||||
def _validate_required_fields(self, request_data):
|
||||
required_fields = [
|
||||
"projectId",
|
||||
"caseId",
|
||||
"automationType",
|
||||
"prompt",
|
||||
"caseKey",
|
||||
"moduleName",
|
||||
"productName",
|
||||
"projectName",
|
||||
"steps",
|
||||
"expectedResults"
|
||||
]
|
||||
missing_fields = []
|
||||
for field in required_fields:
|
||||
if field == "productName":
|
||||
if field not in request_data or request_data.get(field) is None:
|
||||
missing_fields.append(field)
|
||||
elif request_data.get(field) in (None, ""):
|
||||
missing_fields.append(field)
|
||||
if missing_fields:
|
||||
raise ValueError("缺少必填参数:{0}".format(", ".join(missing_fields)))
|
||||
|
||||
def _get_case_name(self, request_data):
|
||||
return request_data.get("caseName") or request_data.get("title") or request_data.get("name") or "{0}-{1}".format(
|
||||
request_data.get("caseKey"),
|
||||
request_data.get("moduleName")
|
||||
)
|
||||
|
||||
def _send_json_response(self, status_code, response_data):
|
||||
response_body = json.dumps(response_data, ensure_ascii=False).encode("utf-8")
|
||||
self.send_response(status_code)
|
||||
self.send_header("Content-Type", "application/json;charset=UTF-8")
|
||||
self.send_header("Access-Control-Allow-Origin", "*")
|
||||
self.send_header("Access-Control-Allow-Methods", "POST, OPTIONS")
|
||||
self.send_header("Access-Control-Allow-Headers", "Content-Type, accessToken, Authorization")
|
||||
self.send_header("Content-Length", str(len(response_body)))
|
||||
self.end_headers()
|
||||
self.wfile.write(response_body)
|
||||
|
||||
def log_message(self, format, *args):
|
||||
print("[{0}] {1}".format(self.log_date_time_string(), format % args))
|
||||
|
||||
|
||||
def run_server(host="0.0.0.0", port=8081):
|
||||
server = ThreadingHTTPServer((host, port), GenerateAutomationHandler)
|
||||
print("Create_ui_testcase HTTP服务已启动:http://{0}:{1}".format(host, port))
|
||||
print("接口地址:POST {0}".format(API_PATH))
|
||||
server.serve_forever()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="UI自动化用例生成HTTP服务")
|
||||
parser.add_argument("--host", default="0.0.0.0", help="服务监听地址,默认0.0.0.0")
|
||||
parser.add_argument("--port", default=8081, type=int, help="服务监听端口,默认8081")
|
||||
args = parser.parse_args()
|
||||
run_server(host=args.host, port=args.port)
|
||||
BIN
dulizhan/screenshots/TC_dulizhan_ui_api_verify_001.png
Normal file
BIN
dulizhan/screenshots/TC_dulizhan_ui_api_verify_001.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 136 KiB |
BIN
dulizhan/screenshots/TC_dulizhan_ui_api_verify_001_1.png
Normal file
BIN
dulizhan/screenshots/TC_dulizhan_ui_api_verify_001_1.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 228 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 317 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 55 KiB |
@@ -0,0 +1,159 @@
|
||||
import os
|
||||
import re
|
||||
from playwright.sync_api import Page, expect
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
|
||||
|
||||
class DownloadAppPage(BasePage):
|
||||
"""
|
||||
Download the App 页面对象
|
||||
|
||||
已知侦察结果:
|
||||
- 首页 URL: https://joyhub-website-frontend-test.best-envision.com/
|
||||
- 页面标题: Joyhub | Explore Sexual Health, Wellness, and Connection
|
||||
- 导航链接: Download the App
|
||||
- Cookie 按钮文案: Accepet
|
||||
"""
|
||||
|
||||
DOWNLOAD_APP_LINK_TEXT = re.compile(r"^Download the App$", re.I)
|
||||
|
||||
def accept_cookie_if_present(self):
|
||||
# 页面侦察结果中按钮文本为 Accepet,疑似拼写如此,按真实页面处理
|
||||
accept_button = self.page.get_by_role("button", name=re.compile(r"Accepet|Accept", re.I))
|
||||
self.click_if_visible(accept_button, timeout=3000)
|
||||
|
||||
def open_download_app_page(self):
|
||||
self.accept_cookie_if_present()
|
||||
|
||||
download_link = self.page.get_by_role("link", name=self.DOWNLOAD_APP_LINK_TEXT)
|
||||
expect(download_link).to_be_visible(timeout=10000)
|
||||
|
||||
download_link.click()
|
||||
try:
|
||||
self.page.wait_for_load_state("networkidle", timeout=5000)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
def get_google_play_locator(self):
|
||||
"""
|
||||
优先使用稳定定位:
|
||||
1. href 包含 play.google.com
|
||||
2. 链接文本包含 Google Play
|
||||
3. 图片 alt 包含 Google,向上找父级 a 标签
|
||||
|
||||
如果后续页面提供 data-testid,建议替换为:
|
||||
self.page.get_by_test_id("google-play-download")
|
||||
"""
|
||||
candidates = [
|
||||
self.page.locator("a[href*='play.google.com']").first,
|
||||
self.page.get_by_role("link", name=re.compile(r"Google\s*Play", re.I)).first,
|
||||
self.page.locator("a").filter(has_text=re.compile(r"Google\s*Play", re.I)).first,
|
||||
self.page.locator("img[alt*='Google' i]").locator("xpath=ancestor::a[1]").first,
|
||||
]
|
||||
|
||||
for locator in candidates:
|
||||
try:
|
||||
if locator.count() > 0:
|
||||
return locator
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
return None
|
||||
|
||||
def discover_google_play_href(self):
|
||||
"""
|
||||
DOM 兜底侦察:
|
||||
当 Google Play 是图片按钮、无文本链接时,用 JS 从渲染后的 DOM 中提取跳转地址。
|
||||
只用于兜底发现,不作为首选点击方式。
|
||||
"""
|
||||
return self.page.evaluate(
|
||||
"""
|
||||
() => {
|
||||
const keywords = ['google play', 'play.google.com'];
|
||||
const nodes = Array.from(document.querySelectorAll('a, button, [role="button"], img'));
|
||||
|
||||
for (const node of nodes) {
|
||||
const text = [
|
||||
node.innerText,
|
||||
node.textContent,
|
||||
node.getAttribute('aria-label'),
|
||||
node.getAttribute('title'),
|
||||
node.getAttribute('alt'),
|
||||
node.getAttribute('href'),
|
||||
node.getAttribute('src')
|
||||
].filter(Boolean).join(' ').toLowerCase();
|
||||
|
||||
if (keywords.some(k => text.includes(k))) {
|
||||
const link = node.closest('a');
|
||||
if (link && link.href) {
|
||||
return link.href;
|
||||
}
|
||||
|
||||
if (node.href) {
|
||||
return node.href;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
"""
|
||||
)
|
||||
|
||||
def click_google_play_and_get_redirect_url(self) -> str:
|
||||
"""
|
||||
点击 Google Play 下载入口,并返回跳转地址。
|
||||
|
||||
成功判定:
|
||||
- 新开页面 URL
|
||||
- 当前页面跳转后的 URL
|
||||
- 或点击前已获取到 Google Play href
|
||||
"""
|
||||
try:
|
||||
self.page.wait_for_load_state("networkidle", timeout=5000)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
google_play_href = self.discover_google_play_href()
|
||||
google_play_locator = self.get_google_play_locator()
|
||||
|
||||
if google_play_locator is None and not google_play_href:
|
||||
raise AssertionError(
|
||||
"未找到 Google Play 下载入口。"
|
||||
"请检查 Download the App 页面是否存在 Google Play 链接,"
|
||||
"或补充稳定 selector,例如 data-testid。"
|
||||
)
|
||||
|
||||
context = self.page.context
|
||||
before_pages = list(context.pages)
|
||||
old_url = self.page.url
|
||||
|
||||
if google_play_locator is not None:
|
||||
google_play_locator.scroll_into_view_if_needed()
|
||||
google_play_locator.click()
|
||||
else:
|
||||
# 没有稳定可点击 locator 时,使用已发现 href 直接跳转。
|
||||
# TODO: 页面增加稳定 selector 后,替换为 locator.click()
|
||||
self.page.goto(google_play_href, wait_until="domcontentloaded")
|
||||
|
||||
self.page.wait_for_timeout(3000)
|
||||
|
||||
after_pages = list(context.pages)
|
||||
new_pages = [p for p in after_pages if p not in before_pages]
|
||||
|
||||
if new_pages:
|
||||
new_page = new_pages[-1]
|
||||
new_page.wait_for_load_state("domcontentloaded")
|
||||
try:
|
||||
new_page.wait_for_load_state("networkidle", timeout=10000)
|
||||
except Exception:
|
||||
pass
|
||||
return new_page.url
|
||||
|
||||
if self.page.url != old_url:
|
||||
return self.page.url
|
||||
|
||||
if google_play_href:
|
||||
return google_play_href
|
||||
|
||||
raise AssertionError("点击 Google Play 后未获取到跳转地址")
|
||||
0
dulizhan/test_case/Resource/UI/__init__.py
Normal file
0
dulizhan/test_case/Resource/UI/__init__.py
Normal file
57
dulizhan/test_case/Resource/UI/base_page.py
Normal file
57
dulizhan/test_case/Resource/UI/base_page.py
Normal file
@@ -0,0 +1,57 @@
|
||||
import os
|
||||
|
||||
from playwright.sync_api import Locator, Page, TimeoutError as PlaywrightTimeoutError
|
||||
|
||||
|
||||
class BasePage:
|
||||
"""Playwright 页面基类:封装通用等待、点击、截图等稳定操作。"""
|
||||
|
||||
def __init__(self, page: Page):
|
||||
self.page = page
|
||||
|
||||
def goto(self, url: str, timeout: int = 60000) -> None:
|
||||
self.page.goto(url, wait_until="domcontentloaded", timeout=timeout)
|
||||
self.wait_for_network_idle()
|
||||
|
||||
def wait_for_network_idle(self, timeout: int = 30000) -> None:
|
||||
try:
|
||||
self.page.wait_for_load_state("networkidle", timeout=timeout)
|
||||
except PlaywrightTimeoutError:
|
||||
self.page.wait_for_load_state("domcontentloaded", timeout=timeout)
|
||||
|
||||
def wait_for_visible(self, locator: Locator, timeout: int = 10000) -> Locator:
|
||||
locator.wait_for(state="visible", timeout=timeout)
|
||||
return locator
|
||||
|
||||
def click_if_visible(self, locator: Locator, timeout: int = 3000) -> bool:
|
||||
try:
|
||||
if locator.first.is_visible(timeout=timeout):
|
||||
locator.first.click()
|
||||
self.wait_for_network_idle()
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
return False
|
||||
|
||||
def safe_click(self, locator: Locator, timeout: int = 10000) -> None:
|
||||
self.wait_for_visible(locator, timeout=timeout)
|
||||
locator.click()
|
||||
self.wait_for_network_idle()
|
||||
|
||||
def screenshot(self, screenshot_dir: str, file_name: str, full_page: bool = True) -> str:
|
||||
os.makedirs(screenshot_dir, exist_ok=True)
|
||||
screenshot_path = os.path.join(screenshot_dir, file_name)
|
||||
self.page.screenshot(path=screenshot_path, full_page=full_page)
|
||||
return screenshot_path
|
||||
|
||||
def current_url(self) -> str:
|
||||
return self.page.url
|
||||
|
||||
def title(self) -> str:
|
||||
return self.page.title()
|
||||
|
||||
def visible_text_contains(self, keyword: str) -> bool:
|
||||
try:
|
||||
return self.page.get_by_text(keyword, exact=False).first.is_visible(timeout=3000)
|
||||
except Exception:
|
||||
return False
|
||||
196
dulizhan/test_case/Resource/UI/blog_page.py
Normal file
196
dulizhan/test_case/Resource/UI/blog_page.py
Normal file
@@ -0,0 +1,196 @@
|
||||
import random
|
||||
from typing import Dict, List
|
||||
|
||||
from playwright.sync_api import TimeoutError as PlaywrightTimeoutError, expect
|
||||
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
|
||||
|
||||
class BlogPage(BasePage):
|
||||
"""
|
||||
Joyhub Blog/内容页页面对象。
|
||||
|
||||
说明:
|
||||
- 页面侦察结果未发现明确的 Blog 导航入口;
|
||||
- 已发现真实导航候选包含 Discover、Following;
|
||||
- 因此进入 blog/内容列表时优先尝试 Blog,失败后使用 Discover 作为内容入口兜底。
|
||||
"""
|
||||
|
||||
HOME_URL = "https://joyhub-website-frontend-test.best-envision.com/"
|
||||
EXPECTED_HOME_TITLE = "Joyhub | Explore Sexual Health, Wellness, and Connection"
|
||||
|
||||
NAV_DISCOVER_LINK = "Discover"
|
||||
|
||||
def open_home_page(self) -> None:
|
||||
self.goto(self.HOME_URL)
|
||||
self.accept_cookie_if_present()
|
||||
expect(self.page).to_have_title(self.EXPECTED_HOME_TITLE)
|
||||
|
||||
def enter_blog_page(self) -> str:
|
||||
"""
|
||||
进入 blog/内容页。
|
||||
优先级:
|
||||
1. Blog 链接:如果页面后续版本新增 Blog 导航,可直接命中;
|
||||
2. Discover 链接:侦察结果中存在,作为当前可用内容入口;
|
||||
3. /blog 直达:仅作为最后兜底,并在无法点击入口时使用。
|
||||
"""
|
||||
before_url = self.page.url
|
||||
|
||||
blog_locators = [
|
||||
self.page.get_by_role("link", name="Blog"),
|
||||
self.page.locator("a", has_text="Blog"),
|
||||
self.page.get_by_role("link", name=self.NAV_DISCOVER_LINK),
|
||||
self.page.locator("a", has_text=self.NAV_DISCOVER_LINK),
|
||||
]
|
||||
|
||||
for locator in blog_locators:
|
||||
try:
|
||||
if locator.count() > 0 and locator.first.is_visible():
|
||||
locator.first.click()
|
||||
self.wait_for_page_ready()
|
||||
return self.page.url
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
fallback_blog_url = self.HOME_URL.rstrip("/") + "/blog"
|
||||
self.goto(fallback_blog_url)
|
||||
|
||||
assert self.page.url != before_url or "/blog" in self.page.url.lower(), (
|
||||
"未能进入 blog/内容页:页面未发现 Blog 入口,Discover 入口也不可点击。"
|
||||
)
|
||||
return self.page.url
|
||||
|
||||
def _get_candidate_blog_links(self) -> List[Dict[str, str]]:
|
||||
"""
|
||||
从渲染后的页面中提取可点击的 blog/content 候选链接。
|
||||
|
||||
不硬编码未知 selector,优先根据 href 语义筛选:
|
||||
blog/article/post/discover/detail/content。
|
||||
若无命中,则退化为 main 区域内可见链接,排除导航类链接。
|
||||
"""
|
||||
return self.page.evaluate(
|
||||
"""
|
||||
() => {
|
||||
const navTexts = new Set([
|
||||
'Home',
|
||||
'Download the App',
|
||||
'Rewards',
|
||||
'Support',
|
||||
'About Us',
|
||||
'Discover',
|
||||
'Following',
|
||||
'Partnerships',
|
||||
'FAQs',
|
||||
'Login'
|
||||
]);
|
||||
|
||||
const normalize = value => (value || '').replace(/\\s+/g, ' ').trim();
|
||||
|
||||
const links = Array.from(document.querySelectorAll('a[href]'))
|
||||
.filter(a => {
|
||||
const rect = a.getBoundingClientRect();
|
||||
const style = window.getComputedStyle(a);
|
||||
return rect.width > 0
|
||||
&& rect.height > 0
|
||||
&& style.visibility !== 'hidden'
|
||||
&& style.display !== 'none';
|
||||
})
|
||||
.map((a, index) => {
|
||||
const href = a.href || '';
|
||||
const text = normalize(a.innerText || a.textContent || a.getAttribute('aria-label') || '');
|
||||
const imgAlt = normalize(
|
||||
Array.from(a.querySelectorAll('img'))
|
||||
.map(img => img.alt)
|
||||
.filter(Boolean)
|
||||
.join(' ')
|
||||
);
|
||||
|
||||
return {
|
||||
index,
|
||||
href,
|
||||
text,
|
||||
imgAlt,
|
||||
pathname: (() => {
|
||||
try { return new URL(href).pathname.toLowerCase(); }
|
||||
catch(e) { return ''; }
|
||||
})()
|
||||
};
|
||||
})
|
||||
.filter(item => item.href && !item.href.startsWith('javascript:'));
|
||||
|
||||
const semanticLinks = links.filter(item =>
|
||||
/blog|article|post|discover|detail|content|story/i.test(item.href)
|
||||
&& !navTexts.has(item.text)
|
||||
);
|
||||
|
||||
if (semanticLinks.length > 0) {
|
||||
return semanticLinks;
|
||||
}
|
||||
|
||||
return links.filter(item =>
|
||||
!navTexts.has(item.text)
|
||||
&& item.href !== window.location.href
|
||||
&& !/#$/.test(item.href)
|
||||
);
|
||||
}
|
||||
"""
|
||||
)
|
||||
|
||||
def click_random_blog(self) -> Dict[str, str]:
|
||||
"""
|
||||
随机点击一个 blog/content 候选项。
|
||||
点击后等待 URL 或页面内容变化。
|
||||
"""
|
||||
self.wait_for_page_ready()
|
||||
candidates = self._get_candidate_blog_links()
|
||||
|
||||
assert candidates, (
|
||||
"未找到可点击的 blog/content 候选链接。"
|
||||
"页面侦察结果缺少明确 blog 卡片 selector,请补充稳定定位,如 data-testid='blog-card'。"
|
||||
)
|
||||
|
||||
candidate = random.choice(candidates)
|
||||
before_url = self.page.url
|
||||
|
||||
locator = self.page.locator("a[href]").nth(candidate["index"])
|
||||
locator.scroll_into_view_if_needed(timeout=5000)
|
||||
|
||||
try:
|
||||
with self.page.expect_navigation(wait_until="domcontentloaded", timeout=8000):
|
||||
locator.click()
|
||||
except PlaywrightTimeoutError:
|
||||
locator.click()
|
||||
|
||||
self.wait_for_page_ready()
|
||||
|
||||
after_url = self.page.url
|
||||
assert after_url != before_url or self.page.locator("main, article, body").first.is_visible(), (
|
||||
f"点击 blog/content 候选项后页面未出现有效跳转或内容区域。候选链接:{candidate['href']}"
|
||||
)
|
||||
|
||||
return {
|
||||
"clicked_href": candidate.get("href", ""),
|
||||
"clicked_text": candidate.get("text") or candidate.get("imgAlt") or "",
|
||||
"before_url": before_url,
|
||||
"after_url": after_url,
|
||||
}
|
||||
|
||||
def assert_blog_content_page_loaded(self) -> None:
|
||||
"""
|
||||
断言已进入 blog/content 内容页。
|
||||
因页面缺少稳定详情页 selector,这里采用内容区域可见的稳健断言。
|
||||
"""
|
||||
self.wait_for_page_ready()
|
||||
|
||||
content_locator = self.page.locator("article, main, [role='main'], body").first
|
||||
expect(content_locator).to_be_visible(timeout=10000)
|
||||
|
||||
body_text = self.page.locator("body").inner_text(timeout=10000).strip()
|
||||
assert len(body_text) > 0, "blog/content 页面 body 内容为空,疑似跳转失败或页面未渲染完成。"
|
||||
|
||||
def capture_blog_content_screenshot(self, screenshot_dir: str, case_key: str) -> str:
|
||||
return self.screenshot(
|
||||
screenshot_dir=screenshot_dir,
|
||||
file_name=f"{case_key}_blog_content.png",
|
||||
full_page=True,
|
||||
)
|
||||
200
dulizhan/test_case/Resource/UI/news_page.py
Normal file
200
dulizhan/test_case/Resource/UI/news_page.py
Normal file
@@ -0,0 +1,200 @@
|
||||
import random
|
||||
import re
|
||||
from typing import Dict, List
|
||||
|
||||
from playwright.sync_api import Page, TimeoutError as PlaywrightTimeoutError
|
||||
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
|
||||
|
||||
class NewsPage(BasePage):
|
||||
"""Joyhub News 相关页面对象。"""
|
||||
|
||||
COOKIE_ACCEPT_TEXT_PATTERN = re.compile(r"^(Accepet|Accept|I Accept|Agree|Got it)$", re.I)
|
||||
NEWS_LINK_TEXT_PATTERN = re.compile(r"^News$", re.I)
|
||||
NEWS_ROUTE = "/news"
|
||||
|
||||
def __init__(self, page: Page, base_url: str):
|
||||
super().__init__(page)
|
||||
self.base_url = base_url.rstrip("/")
|
||||
|
||||
def open_home(self) -> None:
|
||||
self.goto(self.base_url)
|
||||
|
||||
def accept_cookie_if_present(self) -> None:
|
||||
accept_button = self.page.get_by_role("button", name=self.COOKIE_ACCEPT_TEXT_PATTERN)
|
||||
self.click_if_visible(accept_button, timeout=3000)
|
||||
|
||||
def enter_news_page(self) -> None:
|
||||
news_link = self.page.get_by_role("link", name=self.NEWS_LINK_TEXT_PATTERN)
|
||||
|
||||
try:
|
||||
if news_link.first.is_visible(timeout=5000):
|
||||
news_link.first.click()
|
||||
self.wait_for_network_idle()
|
||||
return
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
self.goto(f"{self.base_url}{self.NEWS_ROUTE}")
|
||||
|
||||
def is_news_context(self) -> bool:
|
||||
url = self.page.url.lower()
|
||||
title = self.page.title().lower()
|
||||
|
||||
if "news" in url:
|
||||
return True
|
||||
if "news" in title:
|
||||
return True
|
||||
if self.visible_text_contains("News"):
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def _collect_clickable_news_candidates(self) -> List[Dict[str, str]]:
|
||||
script = """
|
||||
() => {
|
||||
const navTexts = new Set([
|
||||
'home',
|
||||
'download the app',
|
||||
'rewards',
|
||||
'support',
|
||||
'about us',
|
||||
'discover',
|
||||
'following',
|
||||
'partnerships',
|
||||
'faqs',
|
||||
'login'
|
||||
]);
|
||||
|
||||
const currentUrl = new URL(window.location.href);
|
||||
const anchors = Array.from(document.querySelectorAll('a[href]'));
|
||||
|
||||
function isVisible(el) {
|
||||
const style = window.getComputedStyle(el);
|
||||
const rect = el.getBoundingClientRect();
|
||||
return style &&
|
||||
style.visibility !== 'hidden' &&
|
||||
style.display !== 'none' &&
|
||||
rect.width > 0 &&
|
||||
rect.height > 0;
|
||||
}
|
||||
|
||||
function hasContentContainer(el) {
|
||||
return !!el.closest(
|
||||
'article, [class*="article" i], [class*="card" i], [class*="news" i], [class*="post" i], [class*="blog" i], [class*="item" i]'
|
||||
);
|
||||
}
|
||||
|
||||
function isBadProtocol(url) {
|
||||
return ['javascript:', 'mailto:', 'tel:'].includes(url.protocol);
|
||||
}
|
||||
|
||||
function isLikelySocial(url) {
|
||||
const host = url.hostname.toLowerCase();
|
||||
return [
|
||||
'facebook.com',
|
||||
'instagram.com',
|
||||
'twitter.com',
|
||||
'x.com',
|
||||
'youtube.com',
|
||||
'tiktok.com',
|
||||
'linkedin.com'
|
||||
].some(domain => host.includes(domain));
|
||||
}
|
||||
|
||||
const candidates = [];
|
||||
|
||||
for (const a of anchors) {
|
||||
if (!isVisible(a)) continue;
|
||||
|
||||
let url;
|
||||
try {
|
||||
url = new URL(a.href, window.location.origin);
|
||||
} catch (e) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (isBadProtocol(url)) continue;
|
||||
if (isLikelySocial(url)) continue;
|
||||
if (url.href === currentUrl.href) continue;
|
||||
|
||||
const text = (a.innerText || a.textContent || '').trim().replace(/\\s+/g, ' ');
|
||||
const textLower = text.toLowerCase();
|
||||
const pathLower = url.pathname.toLowerCase();
|
||||
|
||||
if (navTexts.has(textLower)) continue;
|
||||
if (url.pathname === '/' || url.pathname === '') continue;
|
||||
|
||||
const hrefLooksLikeContent =
|
||||
/news|article|blog|post|detail|story/i.test(pathLower) &&
|
||||
pathLower !== '/news';
|
||||
|
||||
const containerLooksLikeContent = hasContentContainer(a);
|
||||
|
||||
if (hrefLooksLikeContent || containerLooksLikeContent) {
|
||||
candidates.push({
|
||||
href: url.href,
|
||||
text: text,
|
||||
path: url.pathname,
|
||||
reason: hrefLooksLikeContent ? 'href_content_pattern' : 'content_container'
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
const seen = new Set();
|
||||
return candidates.filter(item => {
|
||||
if (seen.has(item.href)) return false;
|
||||
seen.add(item.href);
|
||||
return true;
|
||||
});
|
||||
}
|
||||
"""
|
||||
return self.page.evaluate(script)
|
||||
|
||||
def click_random_news_item(self) -> Dict[str, str]:
|
||||
self.wait_for_network_idle()
|
||||
candidates = self._collect_clickable_news_candidates()
|
||||
|
||||
if not candidates:
|
||||
raise AssertionError(
|
||||
"未发现可点击的 news 内容候选。"
|
||||
"请检查 News 页面是否加载成功,或为 news 卡片补充稳定 selector/data-testid。"
|
||||
)
|
||||
|
||||
selected = random.choice(candidates)
|
||||
href = selected["href"]
|
||||
old_url = self.page.url
|
||||
|
||||
click_script = """
|
||||
(targetHref) => {
|
||||
const anchors = Array.from(document.querySelectorAll('a[href]'));
|
||||
const target = anchors.find(a => {
|
||||
try {
|
||||
return new URL(a.href, window.location.origin).href === targetHref;
|
||||
} catch (e) {
|
||||
return false;
|
||||
}
|
||||
});
|
||||
|
||||
if (!target) {
|
||||
throw new Error('Target news link not found: ' + targetHref);
|
||||
}
|
||||
|
||||
target.scrollIntoView({block: 'center', inline: 'center'});
|
||||
target.click();
|
||||
}
|
||||
"""
|
||||
|
||||
self.page.evaluate(click_script, href)
|
||||
|
||||
try:
|
||||
self.page.wait_for_url(lambda url: str(url) != old_url, timeout=15000)
|
||||
except PlaywrightTimeoutError:
|
||||
pass
|
||||
|
||||
self.wait_for_network_idle()
|
||||
return selected
|
||||
|
||||
def screenshot_news_content(self, screenshot_dir: str, file_name: str) -> str:
|
||||
return self.screenshot(screenshot_dir=screenshot_dir, file_name=file_name, full_page=True)
|
||||
0
dulizhan/test_case/Resource/__init__.py
Normal file
0
dulizhan/test_case/Resource/__init__.py
Normal file
@@ -0,0 +1,91 @@
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
from dulizhan.test_case.Resource.UI.news_page import NewsPage
|
||||
import os
|
||||
|
||||
import allure
|
||||
import pytest
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
from dulizhan.test_case.Resource.UI.news_page import NewsPage
|
||||
|
||||
|
||||
CASE_INFO = {
|
||||
"projectId": 1001,
|
||||
"caseId": 2001,
|
||||
"automationType": "ui",
|
||||
"caseKey": "TC_dulizhan_ui_api_verify_001",
|
||||
"moduleName": "news",
|
||||
"productName": "",
|
||||
"projectName": "dulizhan",
|
||||
"caseName": "进入news页面,随机点击news,跳转到news内容后截图,就认为用例执行成功",
|
||||
"pageUrl": "https://joyhub-website-frontend-test.best-envision.com/",
|
||||
"screenshotDir": r"C:\Users\a\smart-management-auto-test\dulizhan\screenshots",
|
||||
}
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def browser():
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
yield browser
|
||||
browser.close()
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def page(browser):
|
||||
context = browser.new_context(
|
||||
viewport={"width": 1440, "height": 900},
|
||||
ignore_https_errors=True,
|
||||
)
|
||||
page = context.new_page()
|
||||
yield page
|
||||
context.close()
|
||||
|
||||
|
||||
@allure.feature(CASE_INFO["projectName"])
|
||||
@allure.story(CASE_INFO["moduleName"])
|
||||
@allure.title(CASE_INFO["caseName"])
|
||||
def test_random_click_news_and_capture_content(page):
|
||||
screenshot_dir = CASE_INFO["screenshotDir"]
|
||||
os.makedirs(screenshot_dir, exist_ok=True)
|
||||
|
||||
news_page = NewsPage(page)
|
||||
|
||||
with allure.step("打开被测页面"):
|
||||
news_page.open_home_page()
|
||||
news_page.accept_cookie_if_present()
|
||||
allure.attach(
|
||||
news_page.current_url(),
|
||||
name="首页 URL",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
|
||||
with allure.step("进入 news 页面"):
|
||||
news_page.enter_news_page()
|
||||
allure.attach(
|
||||
news_page.current_url(),
|
||||
name="News 页面 URL",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
|
||||
with allure.step("随机点击 news"):
|
||||
selected_news = news_page.click_random_news()
|
||||
allure.attach(
|
||||
str(selected_news),
|
||||
name="随机点击的 news 候选信息",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
|
||||
with allure.step("截图跳转的 news 内容"):
|
||||
news_page.assert_news_content_loaded()
|
||||
screenshot_path = news_page.capture_news_content_screenshot(
|
||||
screenshot_dir=screenshot_dir,
|
||||
case_key=CASE_INFO["caseKey"],
|
||||
)
|
||||
allure.attach.file(
|
||||
screenshot_path,
|
||||
name="跳转后的 news 内容截图",
|
||||
attachment_type=allure.attachment_type.PNG,
|
||||
)
|
||||
|
||||
assert os.path.exists(screenshot_path), f"news 内容截图未生成: {screenshot_path}"
|
||||
@@ -0,0 +1,80 @@
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
from dulizhan.test_case.Resource.UI.blog_page import BlogPage
|
||||
from types import SimpleNamespace
|
||||
|
||||
import allure
|
||||
import pytest
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
from dulizhan.test_case.Resource.UI.blog_page import BlogPage
|
||||
|
||||
|
||||
case_info = SimpleNamespace(
|
||||
projectId=1001,
|
||||
caseId=2001,
|
||||
automationType="ui",
|
||||
caseKey="TC_dulizhan_ui_api_verify_001",
|
||||
moduleName="blog",
|
||||
productName="",
|
||||
projectName="dulizhan",
|
||||
caseName="进入blog页面,随机点击blog,跳转到blog内容后截图,就认为用例执行成功",
|
||||
screenshotDir=r"C:\Users\a\smart-management-auto-test\dulizhan\screenshots",
|
||||
pageUrl="https://joyhub-website-frontend-test.best-envision.com/",
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def page():
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
context = browser.new_context(
|
||||
viewport={"width": 1440, "height": 900},
|
||||
ignore_https_errors=True,
|
||||
)
|
||||
page = context.new_page()
|
||||
yield page
|
||||
context.close()
|
||||
browser.close()
|
||||
|
||||
|
||||
@allure.feature(case_info.projectName)
|
||||
@allure.story(case_info.moduleName)
|
||||
@allure.title(case_info.caseName)
|
||||
def test_enter_blog_random_click_and_capture_content(page):
|
||||
blog_page = BlogPage(page)
|
||||
|
||||
with allure.step("打开被测页面"):
|
||||
blog_page.open_home_page()
|
||||
assert page.url.startswith(case_info.pageUrl)
|
||||
|
||||
with allure.step("进入blog页面"):
|
||||
entered_url = blog_page.enter_blog_page()
|
||||
allure.attach(
|
||||
entered_url,
|
||||
name="进入blog页面后的URL",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
|
||||
with allure.step("随机点击blog"):
|
||||
click_result = blog_page.click_random_blog()
|
||||
allure.attach(
|
||||
str(click_result),
|
||||
name="随机点击blog结果",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
|
||||
with allure.step("截图跳转的blog内容"):
|
||||
blog_page.assert_blog_content_page_loaded()
|
||||
screenshot_path = blog_page.capture_blog_content_screenshot(
|
||||
screenshot_dir=case_info.screenshotDir,
|
||||
case_key=case_info.caseKey,
|
||||
)
|
||||
|
||||
with open(screenshot_path, "rb") as image_file:
|
||||
allure.attach(
|
||||
image_file.read(),
|
||||
name="blog内容页截图",
|
||||
attachment_type=allure.attachment_type.PNG,
|
||||
)
|
||||
|
||||
assert screenshot_path, "blog 内容页截图路径为空"
|
||||
@@ -0,0 +1,90 @@
|
||||
from dulizhan.test_case.Resource.UI.base_page import BasePage
|
||||
from dulizhan.test_case.Resource.UI.news_page import NewsPage
|
||||
from types import SimpleNamespace
|
||||
|
||||
import allure
|
||||
import pytest
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
from dulizhan.test_case.Resource.UI.news_page import NewsPage
|
||||
|
||||
|
||||
CASE_INFO = SimpleNamespace(
|
||||
projectId=1001,
|
||||
caseId=2001,
|
||||
automationType="ui",
|
||||
caseKey="TC_dulizhan_ui_api_verify_001",
|
||||
moduleName="news",
|
||||
productName="",
|
||||
projectName="dulizhan",
|
||||
caseName="进入news页面,随机点击news,跳转到news内容后截图,就认为用例执行成功",
|
||||
pageUrl="https://joyhub-website-frontend-test.best-envision.com/",
|
||||
screenshotDir=r"C:\Users\a\smart-management-auto-test\dulizhan\screenshots",
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def browser():
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
yield browser
|
||||
browser.close()
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def page(browser):
|
||||
page = browser.new_page(
|
||||
viewport={"width": 1440, "height": 1200},
|
||||
ignore_https_errors=True,
|
||||
)
|
||||
yield page
|
||||
page.close()
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def case_info():
|
||||
return CASE_INFO
|
||||
|
||||
|
||||
@allure.feature(CASE_INFO.projectName)
|
||||
@allure.story(CASE_INFO.moduleName)
|
||||
@allure.title(CASE_INFO.caseName)
|
||||
@pytest.mark.ui
|
||||
def test_enter_news_random_click_and_screenshot(page, case_info):
|
||||
news_page = NewsPage(page, case_info.pageUrl)
|
||||
|
||||
with allure.step("打开被测页面"):
|
||||
news_page.open_home()
|
||||
news_page.accept_cookie_if_present()
|
||||
assert "Joyhub" in news_page.title(), f"首页标题不符合预期,当前标题:{news_page.title()}"
|
||||
|
||||
with allure.step("进入news页面"):
|
||||
news_page.enter_news_page()
|
||||
assert news_page.is_news_context(), (
|
||||
f"未确认进入 news 页面上下文,当前URL:{page.url},当前标题:{page.title()}。"
|
||||
"侦察结果未提供明确 News 导航 selector,如实际路由不是 /news,请调整 NewsPage.NEWS_ROUTE。"
|
||||
)
|
||||
|
||||
with allure.step("随机点击news"):
|
||||
before_click_url = page.url
|
||||
selected_news = news_page.click_random_news_item()
|
||||
allure.attach(
|
||||
str(selected_news),
|
||||
name="随机点击的 news 候选",
|
||||
attachment_type=allure.attachment_type.TEXT,
|
||||
)
|
||||
assert page.url != before_click_url or selected_news.get("href"), (
|
||||
f"点击 news 后页面未发生有效跳转,点击前URL:{before_click_url},点击后URL:{page.url}"
|
||||
)
|
||||
|
||||
with allure.step("截图跳转的news内容"):
|
||||
screenshot_path = news_page.screenshot_news_content(
|
||||
screenshot_dir=case_info.screenshotDir,
|
||||
file_name=f"{case_info.caseKey}_news_content.png",
|
||||
)
|
||||
allure.attach.file(
|
||||
screenshot_path,
|
||||
name="跳转后的news内容截图",
|
||||
attachment_type=allure.attachment_type.PNG,
|
||||
)
|
||||
assert screenshot_path, "news 内容截图保存失败"
|
||||
0
joyhub_backend/__init__.py
Normal file
0
joyhub_backend/__init__.py
Normal file
11
joyhub_backend/library/__init__.py
Normal file
11
joyhub_backend/library/__init__.py
Normal file
@@ -0,0 +1,11 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from joyhub_backend.library.functional_case_converter import (
|
||||
ApiContextItem,
|
||||
ApiMatcher,
|
||||
FunctionalCaseInput,
|
||||
FunctionalCaseParser,
|
||||
FunctionalCaseToApiAutomationService,
|
||||
FunctionalStep,
|
||||
LLMApiCaseConverter,
|
||||
convert_functional_case,
|
||||
)
|
||||
148
joyhub_backend/library/auth.py
Normal file
148
joyhub_backend/library/auth.py
Normal file
@@ -0,0 +1,148 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
|
||||
import allure
|
||||
import requests
|
||||
|
||||
|
||||
CAPTCHA_URL = os.getenv(
|
||||
"JOYHUB_CAPTCHA_URL",
|
||||
"http://test-manager-api.best-envision.com/admin/login/captcha",
|
||||
)
|
||||
LOGIN_BASE_URL = os.getenv(
|
||||
"JOYHUB_LOGIN_BASE_URL",
|
||||
"http://test-manager-api.best-envision.com",
|
||||
)
|
||||
LOGIN_PATH = os.getenv("JOYHUB_LOGIN_PATH", "/admin/login/login")
|
||||
USERNAME = os.getenv("JOYHUB_USERNAME", "guojiabao")
|
||||
PASSWORD = os.getenv("JOYHUB_PASSWORD", "gjb123456")
|
||||
CAPTCHA = os.getenv("JOYHUB_CAPTCHA", "1111")
|
||||
TIMEOUT = int(os.getenv("JOYHUB_TIMEOUT", "20"))
|
||||
TENANT_ID = os.getenv("JOYHUB_TENANT_ID", "126")
|
||||
|
||||
|
||||
class JoyhubAuth(object):
|
||||
def __init__(self):
|
||||
self.session = requests.Session()
|
||||
self._token = None
|
||||
|
||||
@property
|
||||
def login_url(self):
|
||||
return LOGIN_BASE_URL.rstrip("/") + LOGIN_PATH
|
||||
|
||||
def get_captcha_key(self):
|
||||
with allure.step("前置:获取登录验证码 key"):
|
||||
logging.info("GET %s", CAPTCHA_URL)
|
||||
response = self.session.get(CAPTCHA_URL, timeout=TIMEOUT)
|
||||
self._attach_response("captcha", response)
|
||||
response.raise_for_status()
|
||||
data = response.json()
|
||||
key = self._extract_key(data)
|
||||
assert key, "验证码接口未返回 key,响应:{}".format(data)
|
||||
return key
|
||||
|
||||
def login(self):
|
||||
if self._token:
|
||||
return self._token
|
||||
key = self.get_captcha_key()
|
||||
body = {
|
||||
"key": key,
|
||||
"username": USERNAME,
|
||||
"password": PASSWORD,
|
||||
"captcha": CAPTCHA,
|
||||
}
|
||||
headers = {"Content-Type": "application/json", "tenant-id": TENANT_ID}
|
||||
with allure.step("前置:登录并获取 token"):
|
||||
logging.info("POST %s", self.login_url)
|
||||
logging.info("request headers: %s", headers)
|
||||
logging.info("request body: %s", self._safe_body(body))
|
||||
response = self.session.post(self.login_url, json=body, headers=headers, timeout=TIMEOUT)
|
||||
self._attach_request(self.login_url, "POST", headers, self._safe_body(body))
|
||||
self._attach_response("login", response)
|
||||
response.raise_for_status()
|
||||
data = response.json()
|
||||
token = self._extract_token(data)
|
||||
assert token, "登录接口未返回 token,响应:{}".format(data)
|
||||
self._token = token if token.startswith("Bearer ") else "Bearer " + token
|
||||
return self._token
|
||||
|
||||
def auth_headers(self):
|
||||
return {
|
||||
"Authorization": self.login(),
|
||||
"Content-Type": "application/json",
|
||||
"tenant-id": TENANT_ID,
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _extract_key(data):
|
||||
if isinstance(data, dict):
|
||||
for field in ("key", "captchaKey", "captcha_key", "uuid"):
|
||||
if data.get(field):
|
||||
return data.get(field)
|
||||
nested = data.get("data")
|
||||
if isinstance(nested, dict):
|
||||
for field in ("key", "captchaKey", "captcha_key", "uuid"):
|
||||
if nested.get(field):
|
||||
return nested.get(field)
|
||||
if isinstance(nested, str):
|
||||
return nested
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def _extract_token(data):
|
||||
token_fields = (
|
||||
"token",
|
||||
"access_token",
|
||||
"accessToken",
|
||||
"Authorization",
|
||||
"authorization",
|
||||
"userToken",
|
||||
"user_token",
|
||||
"jwt",
|
||||
)
|
||||
if isinstance(data, dict):
|
||||
for field in token_fields:
|
||||
token = data.get(field)
|
||||
if JoyhubAuth._looks_like_token(token):
|
||||
return token
|
||||
nested = data.get("data")
|
||||
if isinstance(nested, dict):
|
||||
for field in token_fields:
|
||||
token = nested.get(field)
|
||||
if JoyhubAuth._looks_like_token(token):
|
||||
return token
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def _looks_like_token(value):
|
||||
if not isinstance(value, str):
|
||||
return False
|
||||
value = value.strip()
|
||||
if not value:
|
||||
return False
|
||||
if any(ord(char) > 127 for char in value):
|
||||
return False
|
||||
return len(value) >= 16 or value.startswith("Bearer ")
|
||||
|
||||
@staticmethod
|
||||
def _safe_body(body):
|
||||
safe = dict(body)
|
||||
if "password" in safe:
|
||||
safe["password"] = "******"
|
||||
return safe
|
||||
|
||||
@staticmethod
|
||||
def _attach_request(url, method, headers, body):
|
||||
allure.attach(str(url), "请求url", allure.attachment_type.TEXT)
|
||||
allure.attach(str(method), "请求方式", allure.attachment_type.TEXT)
|
||||
allure.attach(json.dumps(headers, ensure_ascii=False, indent=2), "请求头", allure.attachment_type.JSON)
|
||||
allure.attach(json.dumps(body, ensure_ascii=False, indent=2), "请求体", allure.attachment_type.JSON)
|
||||
|
||||
@staticmethod
|
||||
def _attach_response(name, response):
|
||||
logging.info("%s response status: %s", name, response.status_code)
|
||||
logging.info("%s response body: %s", name, response.text)
|
||||
allure.attach(str(response.status_code), "{} 响应状态码".format(name), allure.attachment_type.TEXT)
|
||||
allure.attach(response.text, "{} 响应体".format(name), allure.attachment_type.JSON)
|
||||
62
joyhub_backend/library/business.py
Normal file
62
joyhub_backend/library/business.py
Normal file
@@ -0,0 +1,62 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import logging
|
||||
|
||||
from joyhub_backend.library.joyhub_interface import JoyhubInterface
|
||||
|
||||
|
||||
class JoyhubBusiness(JoyhubInterface):
|
||||
def get_video_label_list(self):
|
||||
logging.info("==========获取视频标签列表==========")
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"sort": "id",
|
||||
"label_name": "",
|
||||
"category_id": 0,
|
||||
"video_type": 0,
|
||||
"video_num": 0,
|
||||
"created_at": [],
|
||||
"order": "descending",
|
||||
}
|
||||
return self.request("获取视频标签列表", "POST", "admin/video/getVideoLabelList", body=body)
|
||||
|
||||
def get_video_category_list(self):
|
||||
logging.info("==========获取视频分类列表==========")
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"sort": "id",
|
||||
"category_name": "",
|
||||
"order": "descending",
|
||||
}
|
||||
return self.request("获取视频分类列表", "POST", "admin/video/getVideoCategoryList", body=body)
|
||||
|
||||
def get_exchange_record_list(self):
|
||||
logging.info("==========兑换记录列表==========")
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"status": "",
|
||||
"goods_type": "",
|
||||
"created_at": [],
|
||||
}
|
||||
return self.request("兑换记录列表", "POST", "admin/exchange/recordList", body=body)
|
||||
|
||||
def get_app_feedback_list(self):
|
||||
logging.info("==========app反馈意见列表==========")
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"status": "",
|
||||
"keyword": "",
|
||||
}
|
||||
return self.request("app反馈意见列表", "POST", "admin/feedback/list", body=body)
|
||||
|
||||
def get_sensitive_word_list(self):
|
||||
logging.info("==========敏感词列表==========")
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"keyword": "",
|
||||
}
|
||||
return self.request("敏感词列表", "POST", "admin/sensitiveWord/list", body=body)
|
||||
917
joyhub_backend/library/functional_case_converter.py
Normal file
917
joyhub_backend/library/functional_case_converter.py
Normal file
@@ -0,0 +1,917 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import argparse
|
||||
import copy
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
from dataclasses import asdict, dataclass, field
|
||||
from difflib import SequenceMatcher
|
||||
from typing import Any, Callable, Dict, Iterable, List, Optional, Union
|
||||
|
||||
|
||||
COMMON_CN_KEYWORDS = [
|
||||
"登录",
|
||||
"查询",
|
||||
"搜索",
|
||||
"筛选",
|
||||
"列表",
|
||||
"详情",
|
||||
"新增",
|
||||
"创建",
|
||||
"添加",
|
||||
"修改",
|
||||
"编辑",
|
||||
"更新",
|
||||
"删除",
|
||||
"导出",
|
||||
"导入",
|
||||
"上传",
|
||||
"下载",
|
||||
"提交",
|
||||
"审核",
|
||||
"确认",
|
||||
"保存",
|
||||
"分页",
|
||||
"分页查询",
|
||||
"性能",
|
||||
"响应时间",
|
||||
"超时",
|
||||
"返回",
|
||||
"结果",
|
||||
"成功",
|
||||
"失败",
|
||||
"校验",
|
||||
"重置",
|
||||
"启用",
|
||||
"停用",
|
||||
"开关",
|
||||
"状态",
|
||||
"排序",
|
||||
"标签",
|
||||
"分类",
|
||||
"关键字",
|
||||
"关键字搜索",
|
||||
"模糊搜索",
|
||||
]
|
||||
|
||||
METHOD_INTENT_KEYWORDS = {
|
||||
"GET": ["查询", "搜索", "筛选", "列表", "详情", "获取", "查看", "分页", "返回"],
|
||||
"POST": ["新增", "创建", "添加", "登录", "提交", "导入", "上传", "确认", "审核", "保存"],
|
||||
"PUT": ["修改", "编辑", "更新", "重置"],
|
||||
"PATCH": ["修改", "编辑", "更新", "状态", "启用", "停用", "开关"],
|
||||
"DELETE": ["删除", "移除"],
|
||||
}
|
||||
|
||||
DEFAULT_PLACEHOLDER_URL = "/api/need-confirm"
|
||||
DEFAULT_PAGE_SIZE = 10
|
||||
DEFAULT_RESPONSE_TIME_MS = 1000
|
||||
|
||||
|
||||
@dataclass
|
||||
class FunctionalStep:
|
||||
stepNo: int
|
||||
action: str
|
||||
expectedResult: str = ""
|
||||
|
||||
|
||||
@dataclass
|
||||
class ApiContextItem:
|
||||
apiName: str = ""
|
||||
method: str = ""
|
||||
url: str = ""
|
||||
description: str = ""
|
||||
headers: Dict[str, Any] = field(default_factory=dict)
|
||||
queryParams: Dict[str, Any] = field(default_factory=dict)
|
||||
requestBody: Any = None
|
||||
responseExample: Any = None
|
||||
tags: List[str] = field(default_factory=list)
|
||||
|
||||
|
||||
@dataclass
|
||||
class FunctionalCaseInput:
|
||||
caseId: str = ""
|
||||
caseKey: str = ""
|
||||
caseName: str = ""
|
||||
projectName: str = ""
|
||||
moduleName: str = ""
|
||||
priority: str = ""
|
||||
preconditions: str = ""
|
||||
steps: List[FunctionalStep] = field(default_factory=list)
|
||||
expectedResults: List[str] = field(default_factory=list)
|
||||
apiContext: Optional[List[ApiContextItem]] = None
|
||||
generateOptions: Dict[str, Any] = field(default_factory=dict)
|
||||
extra: Dict[str, Any] = field(default_factory=dict)
|
||||
|
||||
|
||||
class FunctionalCaseParser:
|
||||
@classmethod
|
||||
def parse(cls, payload: Union[str, Dict[str, Any], FunctionalCaseInput]) -> FunctionalCaseInput:
|
||||
if isinstance(payload, FunctionalCaseInput):
|
||||
return payload
|
||||
|
||||
if isinstance(payload, str):
|
||||
try:
|
||||
payload = json.loads(payload)
|
||||
except Exception:
|
||||
payload = {
|
||||
"caseName": "手工输入功能用例",
|
||||
"steps": [payload],
|
||||
"expectedResults": [],
|
||||
}
|
||||
|
||||
if not isinstance(payload, dict):
|
||||
raise TypeError("functional case payload must be dict, json string or FunctionalCaseInput")
|
||||
|
||||
steps = cls._normalize_steps(payload.get("steps") or payload.get("step") or [])
|
||||
expected_results = cls._normalize_text_list(
|
||||
payload.get("expectedResults")
|
||||
or payload.get("expectedResult")
|
||||
or payload.get("expectations")
|
||||
or []
|
||||
)
|
||||
api_context = cls._normalize_api_context(payload.get("apiContext") or payload.get("apis") or payload.get("apiInfo"))
|
||||
generate_options = payload.get("generateOptions") or payload.get("options") or {}
|
||||
|
||||
known_keys = {
|
||||
"caseId",
|
||||
"caseKey",
|
||||
"caseName",
|
||||
"projectName",
|
||||
"moduleName",
|
||||
"priority",
|
||||
"preconditions",
|
||||
"steps",
|
||||
"step",
|
||||
"expectedResults",
|
||||
"expectedResult",
|
||||
"expectations",
|
||||
"apiContext",
|
||||
"apis",
|
||||
"apiInfo",
|
||||
"generateOptions",
|
||||
"options",
|
||||
}
|
||||
extra = {key: value for key, value in payload.items() if key not in known_keys}
|
||||
|
||||
return FunctionalCaseInput(
|
||||
caseId=str(payload.get("caseId", "") or ""),
|
||||
caseKey=str(payload.get("caseKey", "") or ""),
|
||||
caseName=str(payload.get("caseName", "") or payload.get("name", "") or ""),
|
||||
projectName=str(payload.get("projectName", "") or ""),
|
||||
moduleName=str(payload.get("moduleName", "") or ""),
|
||||
priority=str(payload.get("priority", "") or ""),
|
||||
preconditions=str(payload.get("preconditions", "") or payload.get("precondition", "") or ""),
|
||||
steps=steps,
|
||||
expectedResults=expected_results,
|
||||
apiContext=api_context,
|
||||
generateOptions=generate_options if isinstance(generate_options, dict) else {},
|
||||
extra=extra,
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def _normalize_steps(value: Any) -> List[FunctionalStep]:
|
||||
if not value:
|
||||
return []
|
||||
|
||||
if isinstance(value, str):
|
||||
raw_items = [item.strip() for item in re.split(r"[\n\r]+", value) if item.strip()]
|
||||
if len(raw_items) <= 1:
|
||||
raw_items = [item.strip() for item in re.split(r"(?<=[。;;])", value) if item.strip()]
|
||||
items = raw_items
|
||||
elif isinstance(value, list):
|
||||
items = value
|
||||
else:
|
||||
items = [value]
|
||||
|
||||
normalized: List[FunctionalStep] = []
|
||||
for index, item in enumerate(items, start=1):
|
||||
if isinstance(item, dict):
|
||||
step_no = int(item.get("stepNo") or item.get("step_no") or item.get("no") or index)
|
||||
action = str(item.get("action") or item.get("step") or item.get("content") or item.get("description") or "")
|
||||
expected_result = str(item.get("expectedResult") or item.get("expected") or "")
|
||||
else:
|
||||
raw_text = str(item).strip()
|
||||
step_no = index
|
||||
action = re.sub(r"^\s*\d+[\.、\)]\s*", "", raw_text)
|
||||
expected_result = ""
|
||||
if action:
|
||||
normalized.append(FunctionalStep(stepNo=step_no, action=action, expectedResult=expected_result))
|
||||
return normalized
|
||||
|
||||
@staticmethod
|
||||
def _normalize_text_list(value: Any) -> List[str]:
|
||||
if not value:
|
||||
return []
|
||||
if isinstance(value, str):
|
||||
parts = [item.strip() for item in re.split(r"[\n\r]+", value) if item.strip()]
|
||||
if len(parts) <= 1:
|
||||
parts = [item.strip() for item in re.split(r"[。;;]\s*", value) if item.strip()]
|
||||
return [item for item in parts if item]
|
||||
if isinstance(value, list):
|
||||
result = []
|
||||
for item in value:
|
||||
if isinstance(item, dict):
|
||||
text = str(item.get("text") or item.get("content") or item.get("expected") or "").strip()
|
||||
else:
|
||||
text = str(item).strip()
|
||||
if text:
|
||||
result.append(text)
|
||||
return result
|
||||
return [str(value).strip()] if str(value).strip() else []
|
||||
|
||||
@staticmethod
|
||||
def _normalize_api_context(value: Any) -> Optional[List[ApiContextItem]]:
|
||||
if not value:
|
||||
return None
|
||||
|
||||
if isinstance(value, dict):
|
||||
if "apis" in value and isinstance(value["apis"], list):
|
||||
value = value["apis"]
|
||||
elif "items" in value and isinstance(value["items"], list):
|
||||
value = value["items"]
|
||||
else:
|
||||
value = [value]
|
||||
|
||||
if isinstance(value, str):
|
||||
value = value.strip()
|
||||
if not value:
|
||||
return None
|
||||
try:
|
||||
parsed = json.loads(value)
|
||||
return FunctionalCaseParser._normalize_api_context(parsed)
|
||||
except Exception:
|
||||
return FunctionalCaseParser._parse_markdown_api_context(value)
|
||||
|
||||
if not isinstance(value, list):
|
||||
return None
|
||||
|
||||
apis: List[ApiContextItem] = []
|
||||
for item in value:
|
||||
if isinstance(item, ApiContextItem):
|
||||
apis.append(item)
|
||||
continue
|
||||
if not isinstance(item, dict):
|
||||
continue
|
||||
headers = item.get("headers") or item.get("header") or {}
|
||||
query_params = item.get("queryParams") or item.get("query") or item.get("params") or {}
|
||||
tags = item.get("tags") or []
|
||||
if isinstance(tags, str):
|
||||
tags = [tag.strip() for tag in re.split(r"[\s,,/|]+", tags) if tag.strip()]
|
||||
apis.append(
|
||||
ApiContextItem(
|
||||
apiName=str(item.get("apiName") or item.get("name") or ""),
|
||||
method=str(item.get("method") or item.get("httpMethod") or "").upper(),
|
||||
url=str(item.get("url") or item.get("path") or ""),
|
||||
description=str(item.get("description") or item.get("desc") or ""),
|
||||
headers=headers if isinstance(headers, dict) else {},
|
||||
queryParams=query_params if isinstance(query_params, dict) else {},
|
||||
requestBody=item.get("requestBody") if "requestBody" in item else item.get("body"),
|
||||
responseExample=item.get("responseExample") or item.get("response") or item.get("example"),
|
||||
tags=tags if isinstance(tags, list) else [],
|
||||
)
|
||||
)
|
||||
return apis or None
|
||||
|
||||
@staticmethod
|
||||
def _parse_markdown_api_context(text: str) -> Optional[List[ApiContextItem]]:
|
||||
lines = text.splitlines()
|
||||
apis: List[ApiContextItem] = []
|
||||
current: Dict[str, Any] = {}
|
||||
in_body = False
|
||||
body_lines: List[str] = []
|
||||
|
||||
def flush_current():
|
||||
nonlocal current, body_lines, in_body
|
||||
if not current:
|
||||
return
|
||||
body_text = "\n".join(body_lines).strip()
|
||||
request_body = FunctionalCaseParser._safe_json_load(body_text)
|
||||
apis.append(
|
||||
ApiContextItem(
|
||||
apiName=current.get("apiName", ""),
|
||||
method=current.get("method", ""),
|
||||
url=current.get("url", ""),
|
||||
description=current.get("description", ""),
|
||||
headers=current.get("headers", {}),
|
||||
queryParams=current.get("queryParams", {}),
|
||||
requestBody=request_body if request_body is not None else body_text,
|
||||
responseExample=current.get("responseExample"),
|
||||
tags=current.get("tags", []),
|
||||
)
|
||||
)
|
||||
current = {}
|
||||
body_lines = []
|
||||
in_body = False
|
||||
|
||||
for line in lines:
|
||||
stripped = line.strip()
|
||||
if stripped.startswith("## "):
|
||||
flush_current()
|
||||
current["apiName"] = stripped[3:].strip()
|
||||
continue
|
||||
if stripped == "**接口URL**":
|
||||
continue
|
||||
if stripped.startswith("> ") and not current.get("url"):
|
||||
value = stripped[2:].strip()
|
||||
if value and value != "暂无参数":
|
||||
current["url"] = value
|
||||
continue
|
||||
if stripped == "**请求方式**":
|
||||
continue
|
||||
if stripped == "**请求Body参数**":
|
||||
in_body = True
|
||||
body_lines = []
|
||||
continue
|
||||
if stripped == "**响应示例**":
|
||||
in_body = False
|
||||
continue
|
||||
if in_body:
|
||||
if stripped.startswith("```"):
|
||||
continue
|
||||
body_lines.append(line)
|
||||
|
||||
flush_current()
|
||||
return apis or None
|
||||
|
||||
@staticmethod
|
||||
def _safe_json_load(text: str) -> Any:
|
||||
if not text:
|
||||
return None
|
||||
cleaned = text.strip()
|
||||
if cleaned.startswith("```"):
|
||||
cleaned = re.sub(r"^```[a-zA-Z0-9_-]*\s*", "", cleaned)
|
||||
cleaned = re.sub(r"\s*```$", "", cleaned)
|
||||
try:
|
||||
return json.loads(cleaned)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
|
||||
class ApiMatcher:
|
||||
def match(self, functional_case: FunctionalCaseInput, candidates: List[ApiContextItem], top_k: int = 3) -> List[Dict[str, Any]]:
|
||||
matched: List[Dict[str, Any]] = []
|
||||
keywords = self.extract_keywords(functional_case)
|
||||
intent = self.detect_intent(functional_case)
|
||||
|
||||
for api in candidates:
|
||||
score = self._score_api(api, keywords, intent, functional_case)
|
||||
if score <= 0:
|
||||
continue
|
||||
matched.append(
|
||||
{
|
||||
"score": round(score, 4),
|
||||
"api": api,
|
||||
"reason": self._build_reason(api, keywords, intent),
|
||||
}
|
||||
)
|
||||
|
||||
matched.sort(key=lambda item: item["score"], reverse=True)
|
||||
return matched[:top_k]
|
||||
|
||||
def extract_keywords(self, functional_case: FunctionalCaseInput) -> List[str]:
|
||||
texts = [functional_case.caseName, functional_case.moduleName, functional_case.preconditions]
|
||||
texts.extend(step.action for step in functional_case.steps)
|
||||
texts.extend(step.expectedResult for step in functional_case.steps if step.expectedResult)
|
||||
texts.extend(functional_case.expectedResults)
|
||||
texts.extend(str(value) for value in functional_case.extra.values() if value is not None)
|
||||
full_text = " \n ".join(texts)
|
||||
keywords: List[str] = []
|
||||
|
||||
for keyword in COMMON_CN_KEYWORDS:
|
||||
if keyword in full_text and keyword not in keywords:
|
||||
keywords.append(keyword)
|
||||
|
||||
english_tokens = re.findall(r"[A-Za-z_][A-Za-z0-9_]{1,}", full_text)
|
||||
for token in english_tokens:
|
||||
if token not in keywords:
|
||||
keywords.append(token)
|
||||
|
||||
generic_tokens = re.findall(r"[\u4e00-\u9fa5]{2,}", full_text)
|
||||
for token in generic_tokens:
|
||||
if len(token) <= 2:
|
||||
if token not in keywords:
|
||||
keywords.append(token)
|
||||
continue
|
||||
if token not in keywords:
|
||||
keywords.append(token)
|
||||
|
||||
return self._dedupe(keywords)
|
||||
|
||||
def detect_intent(self, functional_case: FunctionalCaseInput) -> str:
|
||||
text = self._all_text(functional_case)
|
||||
for method, keywords in METHOD_INTENT_KEYWORDS.items():
|
||||
if any(keyword in text for keyword in keywords):
|
||||
return method
|
||||
return "GET"
|
||||
|
||||
def _score_api(self, api: ApiContextItem, keywords: List[str], intent: str, functional_case: FunctionalCaseInput) -> float:
|
||||
score = 0.0
|
||||
api_text = " ".join(
|
||||
[
|
||||
api.apiName or "",
|
||||
api.method or "",
|
||||
api.url or "",
|
||||
api.description or "",
|
||||
" ".join(api.tags or []),
|
||||
" ".join(api.headers.keys()),
|
||||
" ".join(api.queryParams.keys()),
|
||||
]
|
||||
)
|
||||
if api.method and api.method.upper() == intent:
|
||||
score += 1.2
|
||||
elif api.method and api.method.upper() in ("GET", "POST", "PUT", "PATCH", "DELETE"):
|
||||
score += 0.2
|
||||
|
||||
for keyword in keywords:
|
||||
if keyword and keyword in api_text:
|
||||
score += 0.8
|
||||
|
||||
for token in self._url_tokens(api.url):
|
||||
if token and any(keyword in token or token in keyword for keyword in keywords):
|
||||
score += 0.6
|
||||
|
||||
if functional_case.moduleName and functional_case.moduleName in api_text:
|
||||
score += 0.8
|
||||
if functional_case.caseName and functional_case.caseName in api_text:
|
||||
score += 0.5
|
||||
|
||||
ratio = SequenceMatcher(None, self._all_text(functional_case), api_text).ratio()
|
||||
score += ratio * 0.8
|
||||
return score
|
||||
|
||||
@staticmethod
|
||||
def _build_reason(api: ApiContextItem, keywords: List[str], intent: str) -> str:
|
||||
matched_keywords = [keyword for keyword in keywords if keyword and keyword in (api.apiName + " " + api.description + " " + api.url)]
|
||||
parts = []
|
||||
if api.method and api.method.upper() == intent:
|
||||
parts.append("方法匹配")
|
||||
if matched_keywords:
|
||||
parts.append("关键词命中:{}".format("、".join(matched_keywords[:5])))
|
||||
return "; ".join(parts) or "模糊匹配"
|
||||
|
||||
@staticmethod
|
||||
def _url_tokens(url: str) -> List[str]:
|
||||
if not url:
|
||||
return []
|
||||
return [token for token in re.split(r"[/?&=_\-\.]+", url) if token]
|
||||
|
||||
@staticmethod
|
||||
def _dedupe(items: Iterable[str]) -> List[str]:
|
||||
seen = set()
|
||||
result = []
|
||||
for item in items:
|
||||
if item and item not in seen:
|
||||
seen.add(item)
|
||||
result.append(item)
|
||||
return result
|
||||
|
||||
@staticmethod
|
||||
def _all_text(functional_case: FunctionalCaseInput) -> str:
|
||||
texts = [functional_case.caseName, functional_case.moduleName, functional_case.preconditions]
|
||||
texts.extend(step.action for step in functional_case.steps)
|
||||
texts.extend(step.expectedResult for step in functional_case.steps if step.expectedResult)
|
||||
texts.extend(functional_case.expectedResults)
|
||||
return " \n ".join(filter(None, texts))
|
||||
|
||||
|
||||
class LLMApiCaseConverter:
|
||||
def convert(self, prompt: str, llm_client: Any = None) -> Optional[Dict[str, Any]]:
|
||||
if llm_client is None:
|
||||
return None
|
||||
|
||||
raw = self._invoke_client(llm_client, prompt)
|
||||
if raw is None:
|
||||
return None
|
||||
if isinstance(raw, dict):
|
||||
return raw
|
||||
if hasattr(raw, "content"):
|
||||
raw = raw.content
|
||||
elif hasattr(raw, "text"):
|
||||
raw = raw.text
|
||||
elif hasattr(raw, "data"):
|
||||
raw = raw.data
|
||||
if isinstance(raw, dict):
|
||||
return raw
|
||||
if not isinstance(raw, str):
|
||||
raw = str(raw)
|
||||
return self._extract_json(raw)
|
||||
|
||||
@staticmethod
|
||||
def _invoke_client(llm_client: Any, prompt: str) -> Any:
|
||||
if callable(llm_client):
|
||||
try:
|
||||
return llm_client(prompt=prompt)
|
||||
except TypeError:
|
||||
try:
|
||||
return llm_client(prompt)
|
||||
except TypeError:
|
||||
return llm_client({"prompt": prompt})
|
||||
if hasattr(llm_client, "generate") and callable(llm_client.generate):
|
||||
return llm_client.generate(prompt)
|
||||
if hasattr(llm_client, "chat") and callable(llm_client.chat):
|
||||
return llm_client.chat(prompt)
|
||||
raise TypeError("llm_client must be callable or expose generate/chat")
|
||||
|
||||
@staticmethod
|
||||
def _extract_json(text: str) -> Optional[Dict[str, Any]]:
|
||||
if not text:
|
||||
return None
|
||||
cleaned = text.strip()
|
||||
match = re.search(r"```json\s*(\{.*?\})\s*```", cleaned, flags=re.S)
|
||||
if match:
|
||||
cleaned = match.group(1)
|
||||
else:
|
||||
first = cleaned.find("{")
|
||||
last = cleaned.rfind("}")
|
||||
if first >= 0 and last > first:
|
||||
cleaned = cleaned[first : last + 1]
|
||||
try:
|
||||
parsed = json.loads(cleaned)
|
||||
return parsed if isinstance(parsed, dict) else {"data": parsed}
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
|
||||
class FunctionalCaseToApiAutomationService:
|
||||
def __init__(self, matcher: Optional[ApiMatcher] = None):
|
||||
self.matcher = matcher or ApiMatcher()
|
||||
self.llm_converter = LLMApiCaseConverter()
|
||||
|
||||
def convert(self, payload: Union[str, Dict[str, Any], FunctionalCaseInput], llm_client: Any = None) -> Dict[str, Any]:
|
||||
functional_case = FunctionalCaseParser.parse(payload)
|
||||
options = self._build_options(functional_case)
|
||||
candidates = functional_case.apiContext or []
|
||||
matched = self.matcher.match(functional_case, candidates, top_k=int(options.get("topK", 3))) if candidates else []
|
||||
|
||||
llm_result = None
|
||||
if options.get("useLLM", True) and llm_client is not None:
|
||||
prompt = self.build_prompt(functional_case, matched, options)
|
||||
llm_result = self.llm_converter.convert(prompt, llm_client)
|
||||
|
||||
if llm_result:
|
||||
return self._finalize_llm_result(functional_case, matched, llm_result, options)
|
||||
|
||||
return self._build_rule_result(functional_case, matched, options)
|
||||
|
||||
def build_prompt(self, functional_case: FunctionalCaseInput, matched: List[Dict[str, Any]], options: Dict[str, Any]) -> str:
|
||||
payload = {
|
||||
"functionalCase": asdict(functional_case),
|
||||
"matchedApis": [
|
||||
{
|
||||
"score": item["score"],
|
||||
"reason": item["reason"],
|
||||
"api": asdict(item["api"]),
|
||||
}
|
||||
for item in matched
|
||||
],
|
||||
"options": options,
|
||||
}
|
||||
return (
|
||||
"你是接口自动化测试用例生成专家。\n"
|
||||
"请把功能测试用例转换成接口自动化测试用例。\n"
|
||||
"要求:\n"
|
||||
"1. 只输出严格 JSON。\n"
|
||||
"2. 如果给了候选接口信息,优先使用候选接口。\n"
|
||||
"3. 无法确认的接口信息写入 missingInfo。\n"
|
||||
"4. 需要包含 method、url、headers、queryParams、body、assertions、performanceAssertions。\n"
|
||||
"5. 如果涉及性能要求,生成 responseTime 断言。\n"
|
||||
"6. 如果涉及登录态,使用 ${token}。\n\n"
|
||||
"输入数据:\n"
|
||||
f"{json.dumps(payload, ensure_ascii=False, indent=2)}\n\n"
|
||||
"输出 JSON 结构:\n"
|
||||
"{\n"
|
||||
' "caseId": "",\n'
|
||||
' "caseKey": "",\n'
|
||||
' "caseName": "",\n'
|
||||
' "automationType": "api",\n'
|
||||
' "convertStatus": "SUCCESS",\n'
|
||||
' "apiTestCases": [],\n'
|
||||
' "missingInfo": [],\n'
|
||||
' "warnings": []\n'
|
||||
"}"
|
||||
)
|
||||
|
||||
def _build_options(self, functional_case: FunctionalCaseInput) -> Dict[str, Any]:
|
||||
options = copy.deepcopy(functional_case.generateOptions or {})
|
||||
options.setdefault("useLLM", True)
|
||||
options.setdefault("outputFormat", "json")
|
||||
options.setdefault("targetFramework", "pytest")
|
||||
options.setdefault("allowMissingInfo", True)
|
||||
options.setdefault("topK", 3)
|
||||
return options
|
||||
|
||||
def _build_rule_result(self, functional_case: FunctionalCaseInput, matched: List[Dict[str, Any]], options: Dict[str, Any]) -> Dict[str, Any]:
|
||||
api_test_cases: List[Dict[str, Any]] = []
|
||||
missing_info: List[str] = []
|
||||
warnings: List[str] = []
|
||||
|
||||
if matched:
|
||||
for index, item in enumerate(matched, start=1):
|
||||
api_test_cases.append(self._build_api_test_case(functional_case, item["api"], index, item["score"], options))
|
||||
else:
|
||||
api_test_cases.append(self._build_fallback_test_case(functional_case, options))
|
||||
missing_info.extend(self._collect_missing_info(functional_case))
|
||||
warnings.append("未匹配到真实接口信息,已生成占位自动化用例")
|
||||
|
||||
missing_info = self._dedupe_text(missing_info)
|
||||
warnings = self._dedupe_text(warnings)
|
||||
|
||||
convert_status = "SUCCESS"
|
||||
if missing_info:
|
||||
convert_status = "SUCCESS_WITH_MISSING_INFO"
|
||||
if warnings and not matched:
|
||||
convert_status = "DRAFT"
|
||||
|
||||
return {
|
||||
"caseId": functional_case.caseId,
|
||||
"caseKey": functional_case.caseKey,
|
||||
"caseName": functional_case.caseName,
|
||||
"projectName": functional_case.projectName,
|
||||
"moduleName": functional_case.moduleName,
|
||||
"automationType": "api",
|
||||
"convertStatus": convert_status,
|
||||
"apiTestCases": api_test_cases,
|
||||
"missingInfo": missing_info,
|
||||
"warnings": warnings,
|
||||
}
|
||||
|
||||
def _finalize_llm_result(self, functional_case: FunctionalCaseInput, matched: List[Dict[str, Any]], llm_result: Dict[str, Any], options: Dict[str, Any]) -> Dict[str, Any]:
|
||||
result = copy.deepcopy(llm_result)
|
||||
result.setdefault("caseId", functional_case.caseId)
|
||||
result.setdefault("caseKey", functional_case.caseKey)
|
||||
result.setdefault("caseName", functional_case.caseName)
|
||||
result.setdefault("projectName", functional_case.projectName)
|
||||
result.setdefault("moduleName", functional_case.moduleName)
|
||||
result.setdefault("automationType", "api")
|
||||
result.setdefault("convertStatus", "SUCCESS")
|
||||
result.setdefault("apiTestCases", [])
|
||||
result.setdefault("missingInfo", [])
|
||||
result.setdefault("warnings", [])
|
||||
|
||||
if not result.get("apiTestCases"):
|
||||
rule_result = self._build_rule_result(functional_case, matched, options)
|
||||
result["apiTestCases"] = rule_result["apiTestCases"]
|
||||
if not result.get("missingInfo"):
|
||||
result["missingInfo"] = rule_result["missingInfo"]
|
||||
if not result.get("warnings"):
|
||||
result["warnings"] = rule_result["warnings"]
|
||||
result["convertStatus"] = rule_result["convertStatus"]
|
||||
|
||||
result["missingInfo"] = self._dedupe_text(result.get("missingInfo", []))
|
||||
result["warnings"] = self._dedupe_text(result.get("warnings", []))
|
||||
return result
|
||||
|
||||
def _build_api_test_case(self, functional_case: FunctionalCaseInput, api: ApiContextItem, index: int, score: float, options: Dict[str, Any]) -> Dict[str, Any]:
|
||||
method = (api.method or self.matcher.detect_intent(functional_case)).upper()
|
||||
query_params = copy.deepcopy(api.queryParams or {})
|
||||
body = copy.deepcopy(api.requestBody)
|
||||
headers = copy.deepcopy(api.headers or {})
|
||||
if not headers:
|
||||
headers = {"Content-Type": "application/json"}
|
||||
if not any(key.lower() == "authorization" for key in headers):
|
||||
headers["Authorization"] = "Bearer ${token}"
|
||||
|
||||
variables = self._build_variables(functional_case, api, query_params, body)
|
||||
query_params = self._fill_placeholders(query_params, variables)
|
||||
body = self._fill_placeholders(body, variables)
|
||||
|
||||
assertions = self._build_assertions(functional_case, api, method)
|
||||
performance_assertions = self._build_performance_assertions(functional_case)
|
||||
step_name = api.apiName or functional_case.caseName or "接口自动化步骤"
|
||||
|
||||
return {
|
||||
"stepNo": index,
|
||||
"name": step_name,
|
||||
"apiName": api.apiName or step_name,
|
||||
"method": method,
|
||||
"url": api.url or self._infer_placeholder_url(functional_case),
|
||||
"headers": headers,
|
||||
"queryParams": query_params,
|
||||
"body": body,
|
||||
"extractVariables": [],
|
||||
"assertions": assertions,
|
||||
"performanceAssertions": performance_assertions,
|
||||
"variables": variables,
|
||||
"matchedApi": {
|
||||
"apiName": api.apiName,
|
||||
"method": method,
|
||||
"url": api.url,
|
||||
"matchScore": round(score, 4),
|
||||
},
|
||||
}
|
||||
|
||||
def _build_fallback_test_case(self, functional_case: FunctionalCaseInput, options: Dict[str, Any]) -> Dict[str, Any]:
|
||||
method = self.matcher.detect_intent(functional_case)
|
||||
url = self._infer_placeholder_url(functional_case)
|
||||
variables = self._build_variables(functional_case, None, {}, None)
|
||||
return {
|
||||
"stepNo": 1,
|
||||
"name": functional_case.caseName or "功能用例转接口自动化",
|
||||
"apiName": functional_case.caseName or "待确认接口",
|
||||
"method": method,
|
||||
"url": url,
|
||||
"headers": {
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": "Bearer ${token}",
|
||||
},
|
||||
"queryParams": {},
|
||||
"body": None,
|
||||
"extractVariables": [],
|
||||
"assertions": self._build_assertions(functional_case, None, method),
|
||||
"performanceAssertions": self._build_performance_assertions(functional_case),
|
||||
"variables": variables,
|
||||
"matchedApi": {
|
||||
"apiName": "",
|
||||
"method": method,
|
||||
"url": url,
|
||||
"matchScore": 0,
|
||||
},
|
||||
}
|
||||
|
||||
def _build_variables(self, functional_case: FunctionalCaseInput, api: Optional[ApiContextItem], query_params: Dict[str, Any], body: Any) -> Dict[str, Any]:
|
||||
variables: Dict[str, Any] = {}
|
||||
text = self.matcher._all_text(functional_case)
|
||||
for keyword in self.matcher.extract_keywords(functional_case):
|
||||
if keyword in ("搜索", "查询", "列表", "筛选", "关键字", "模糊搜索"):
|
||||
variables.setdefault("keyword", "测试")
|
||||
if keyword in ("分页", "分页查询"):
|
||||
variables.setdefault("page", 1)
|
||||
variables.setdefault("limit", DEFAULT_PAGE_SIZE)
|
||||
if keyword in ("登录",):
|
||||
variables.setdefault("token", "${token}")
|
||||
if not variables and text:
|
||||
variables["keyword"] = "测试"
|
||||
if query_params:
|
||||
for key in query_params.keys():
|
||||
if key not in variables and any(token in key.lower() for token in ["keyword", "name", "id", "page", "limit", "status", "type"]):
|
||||
variables[key] = query_params[key]
|
||||
if isinstance(body, dict):
|
||||
for key in body.keys():
|
||||
if key not in variables and any(token in key.lower() for token in ["keyword", "name", "id", "page", "limit", "status", "type"]):
|
||||
variables[key] = body[key]
|
||||
return variables
|
||||
|
||||
def _build_assertions(self, functional_case: FunctionalCaseInput, api: Optional[ApiContextItem], method: str) -> List[Dict[str, Any]]:
|
||||
assertions = [
|
||||
{
|
||||
"type": "statusCode",
|
||||
"actual": "$.code" if api or method != "DELETE" else "$.status",
|
||||
"operator": "==",
|
||||
"expected": 200,
|
||||
}
|
||||
]
|
||||
text = self.matcher._all_text(functional_case)
|
||||
if api and isinstance(api.responseExample, dict):
|
||||
example = api.responseExample
|
||||
if "code" in example:
|
||||
assertions.append(
|
||||
{
|
||||
"type": "jsonPath",
|
||||
"actual": "$.code",
|
||||
"operator": "==",
|
||||
"expected": example.get("code", 0),
|
||||
}
|
||||
)
|
||||
if "msg" in example:
|
||||
assertions.append(
|
||||
{
|
||||
"type": "jsonPath",
|
||||
"actual": "$.msg",
|
||||
"operator": "notEmpty",
|
||||
"expected": True,
|
||||
}
|
||||
)
|
||||
|
||||
if any(keyword in text for keyword in ["搜索", "查询", "列表", "筛选", "返回"]):
|
||||
assertions.append(
|
||||
{
|
||||
"type": "jsonPath",
|
||||
"actual": "$.data",
|
||||
"operator": "notEmpty",
|
||||
"expected": True,
|
||||
}
|
||||
)
|
||||
return assertions
|
||||
|
||||
def _build_performance_assertions(self, functional_case: FunctionalCaseInput) -> List[Dict[str, Any]]:
|
||||
text = self.matcher._all_text(functional_case)
|
||||
expected_ms = self._extract_response_time_ms(text)
|
||||
if expected_ms is None:
|
||||
return []
|
||||
return [
|
||||
{
|
||||
"type": "responseTime",
|
||||
"operator": "<=",
|
||||
"expected": expected_ms,
|
||||
"unit": "ms",
|
||||
}
|
||||
]
|
||||
|
||||
def _collect_missing_info(self, functional_case: FunctionalCaseInput) -> List[str]:
|
||||
missing = []
|
||||
if not functional_case.steps:
|
||||
missing.append("缺少步骤信息")
|
||||
if not functional_case.caseName:
|
||||
missing.append("缺少用例名称")
|
||||
if not functional_case.apiContext:
|
||||
missing.append("未传入真实接口信息,已使用占位接口生成")
|
||||
if self._extract_response_time_ms(self.matcher._all_text(functional_case)) is None and any(k in self.matcher._all_text(functional_case) for k in ["性能", "响应时间", "超时"]):
|
||||
missing.append("性能阈值未明确,默认使用 1000ms")
|
||||
return missing
|
||||
|
||||
@staticmethod
|
||||
def _extract_response_time_ms(text: str) -> Optional[int]:
|
||||
if not text:
|
||||
return None
|
||||
patterns = [
|
||||
r"(\d+)\s*毫秒",
|
||||
r"(\d+)\s*ms",
|
||||
r"(\d+)\s*秒",
|
||||
r"不超过\s*(\d+)\s*秒",
|
||||
r"<=\s*(\d+)\s*秒",
|
||||
r"小于等于\s*(\d+)\s*秒",
|
||||
]
|
||||
for pattern in patterns:
|
||||
match = re.search(pattern, text, flags=re.I)
|
||||
if match:
|
||||
value = int(match.group(1))
|
||||
if "秒" in pattern:
|
||||
return value * 1000
|
||||
return value
|
||||
if any(keyword in text for keyword in ["响应时间", "超时", "性能"]):
|
||||
return DEFAULT_RESPONSE_TIME_MS
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def _infer_placeholder_url(functional_case: FunctionalCaseInput) -> str:
|
||||
text = " ".join([functional_case.caseName, functional_case.moduleName] + [step.action for step in functional_case.steps])
|
||||
if any(keyword in text for keyword in ["登录"]):
|
||||
return "/api/login"
|
||||
if any(keyword in text for keyword in ["搜索", "查询", "列表", "筛选"]):
|
||||
return "/api/list/query"
|
||||
if any(keyword in text for keyword in ["新增", "创建", "添加"]):
|
||||
return "/api/create"
|
||||
if any(keyword in text for keyword in ["修改", "编辑", "更新"]):
|
||||
return "/api/update"
|
||||
if any(keyword in text for keyword in ["删除", "移除"]):
|
||||
return "/api/delete"
|
||||
return DEFAULT_PLACEHOLDER_URL
|
||||
|
||||
@staticmethod
|
||||
def _fill_placeholders(value: Any, variables: Dict[str, Any]) -> Any:
|
||||
if isinstance(value, dict):
|
||||
return {key: FunctionalCaseToApiAutomationService._fill_placeholders(item, variables) for key, item in value.items()}
|
||||
if isinstance(value, list):
|
||||
return [FunctionalCaseToApiAutomationService._fill_placeholders(item, variables) for item in value]
|
||||
if isinstance(value, str):
|
||||
result = value
|
||||
for key, variable in variables.items():
|
||||
result = result.replace("${" + key + "}", str(variable))
|
||||
return result
|
||||
return value
|
||||
|
||||
@staticmethod
|
||||
def _dedupe_text(items: Iterable[str]) -> List[str]:
|
||||
seen = set()
|
||||
result = []
|
||||
for item in items:
|
||||
text = str(item).strip()
|
||||
if not text or text in seen:
|
||||
continue
|
||||
seen.add(text)
|
||||
result.append(text)
|
||||
return result
|
||||
|
||||
|
||||
def convert_functional_case(payload: Union[str, Dict[str, Any], FunctionalCaseInput], llm_client: Any = None) -> Dict[str, Any]:
|
||||
return FunctionalCaseToApiAutomationService().convert(payload, llm_client=llm_client)
|
||||
|
||||
|
||||
def load_payload_from_file(file_path: str) -> Dict[str, Any]:
|
||||
with open(file_path, "r", encoding="utf-8") as file:
|
||||
return json.load(file)
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="功能测试用例转接口自动化用例")
|
||||
parser.add_argument("--input", type=str, help="输入JSON文件路径,未指定则从stdin读取")
|
||||
parser.add_argument("--output", type=str, help="输出JSON文件路径,未指定则打印到stdout")
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.input:
|
||||
payload = load_payload_from_file(args.input)
|
||||
else:
|
||||
payload = json.load(os.sys.stdin)
|
||||
|
||||
result = convert_functional_case(payload)
|
||||
|
||||
output = json.dumps(result, ensure_ascii=False, indent=2)
|
||||
if args.output:
|
||||
with open(args.output, "w", encoding="utf-8") as file:
|
||||
file.write(output)
|
||||
else:
|
||||
print(output)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
189
joyhub_backend/library/hubops_parser.py
Normal file
189
joyhub_backend/library/hubops_parser.py
Normal file
@@ -0,0 +1,189 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import ast
|
||||
import json
|
||||
import os
|
||||
import re
|
||||
|
||||
|
||||
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
||||
DEFAULT_HUBOPS_PATH = os.path.join(PROJECT_ROOT, "HubOps.md")
|
||||
|
||||
|
||||
class HubOpsParser(object):
|
||||
def __init__(self, file_path=None):
|
||||
self.file_path = file_path or os.getenv("JOYHUB_DOC_PATH", DEFAULT_HUBOPS_PATH)
|
||||
|
||||
def parse(self):
|
||||
with open(self.file_path, "r", encoding="utf-8") as file:
|
||||
lines = file.read().splitlines()
|
||||
|
||||
cases = []
|
||||
headings = []
|
||||
for index, line in enumerate(lines):
|
||||
heading = self._parse_heading(line)
|
||||
if heading:
|
||||
level, title = heading
|
||||
headings = [item for item in headings if item[0] < level]
|
||||
headings.append((level, title))
|
||||
continue
|
||||
|
||||
if line.strip() != "**接口URL**":
|
||||
continue
|
||||
|
||||
case = self._parse_case(lines, index, headings)
|
||||
if case.get("url"):
|
||||
case["case_id"] = "joyhub_{:04d}".format(len(cases) + 1)
|
||||
cases.append(case)
|
||||
return cases
|
||||
|
||||
def _parse_case(self, lines, url_index, headings):
|
||||
title = self._case_title(headings, url_index)
|
||||
url = self._next_quote_value(lines, url_index)
|
||||
method = self._find_section_quote(lines, url_index, "**请求方式**") or "POST"
|
||||
content_type = self._find_section_quote(lines, url_index, "**Content-Type**") or "json"
|
||||
body_text = self._find_code_block(lines, url_index, "**请求Body参数**")
|
||||
query = self._find_param_table(lines, url_index, "**请求Query参数**")
|
||||
headers = self._find_headers(lines, url_index)
|
||||
body = self._parse_body(body_text)
|
||||
return {
|
||||
"name": title,
|
||||
"method": method.upper(),
|
||||
"url": url,
|
||||
"content_type": content_type,
|
||||
"headers": headers,
|
||||
"query": query,
|
||||
"body": body,
|
||||
"raw_body": body_text,
|
||||
}
|
||||
|
||||
@staticmethod
|
||||
def _parse_heading(line):
|
||||
match = re.match(r"^(#{2,6})\s+(.+?)\s*$", line)
|
||||
if not match:
|
||||
return None
|
||||
return len(match.group(1)), match.group(2).strip()
|
||||
|
||||
@staticmethod
|
||||
def _case_title(headings, url_index):
|
||||
if headings:
|
||||
return " / ".join(title for _, title in headings[-3:])
|
||||
return "HubOps接口{}".format(url_index + 1)
|
||||
|
||||
@staticmethod
|
||||
def _next_quote_value(lines, start):
|
||||
for index in range(start + 1, min(start + 8, len(lines))):
|
||||
line = lines[index].strip()
|
||||
if line.startswith(">"):
|
||||
value = line[1:].strip()
|
||||
if value and value != "暂无参数":
|
||||
return value
|
||||
return ""
|
||||
|
||||
@staticmethod
|
||||
def _section_end(lines, start):
|
||||
for index in range(start + 1, len(lines)):
|
||||
line = lines[index].strip()
|
||||
if line.startswith("## ") or line.startswith("### ") or line == "**接口URL**":
|
||||
return index
|
||||
return len(lines)
|
||||
|
||||
def _find_section_quote(self, lines, url_index, section_name):
|
||||
start = max(0, url_index - 80)
|
||||
end = self._section_end(lines, url_index)
|
||||
for index in range(start, min(end, len(lines))):
|
||||
if lines[index].strip() == section_name:
|
||||
return self._next_quote_value(lines, index)
|
||||
return None
|
||||
|
||||
def _find_code_block(self, lines, url_index, section_name):
|
||||
end = self._section_end(lines, url_index)
|
||||
for index in range(url_index, end):
|
||||
if lines[index].strip() != section_name:
|
||||
continue
|
||||
for code_start in range(index + 1, end):
|
||||
if lines[code_start].strip().startswith("```"):
|
||||
block = []
|
||||
for code_end in range(code_start + 1, end):
|
||||
if lines[code_end].strip().startswith("```"):
|
||||
return "\n".join(block).strip()
|
||||
block.append(lines[code_end])
|
||||
return ""
|
||||
|
||||
def _find_param_table(self, lines, url_index, section_name):
|
||||
end = self._section_end(lines, url_index)
|
||||
for index in range(url_index, end):
|
||||
if lines[index].strip() != section_name:
|
||||
continue
|
||||
params = {}
|
||||
for row_index in range(index + 1, end):
|
||||
row = lines[row_index].strip()
|
||||
if not row.startswith("|"):
|
||||
if params:
|
||||
break
|
||||
continue
|
||||
if "---" in row or "参数名" in row or "暂无参数" in row:
|
||||
continue
|
||||
columns = [column.strip() for column in row.strip("|").split("|")]
|
||||
if len(columns) >= 2 and columns[0]:
|
||||
params[columns[0]] = self._coerce_value(columns[1])
|
||||
return params
|
||||
return {}
|
||||
|
||||
def _find_headers(self, lines, url_index):
|
||||
headers = {}
|
||||
end = self._section_end(lines, url_index)
|
||||
for index in range(url_index, end):
|
||||
if lines[index].strip() != "**请求Header参数**":
|
||||
continue
|
||||
table_started = False
|
||||
for row_index in range(index + 1, end):
|
||||
row = lines[row_index].strip()
|
||||
if row.startswith("**") or row.startswith("#") or row.startswith("*"):
|
||||
break
|
||||
if not row.startswith("|"):
|
||||
if table_started:
|
||||
break
|
||||
continue
|
||||
table_started = True
|
||||
if "---" in row or "参数名" in row or "暂无参数" in row:
|
||||
continue
|
||||
columns = [column.strip() for column in row.strip("|").split("|")]
|
||||
if len(columns) >= 2 and columns[0] and columns[0] != "Authorization":
|
||||
headers[columns[0]] = columns[1]
|
||||
return headers
|
||||
return headers
|
||||
|
||||
@classmethod
|
||||
def _parse_body(cls, body_text):
|
||||
if not body_text or body_text == "暂无数据":
|
||||
return {}
|
||||
text = cls._clean_body(body_text)
|
||||
for loader in (json.loads, ast.literal_eval):
|
||||
try:
|
||||
value = loader(text)
|
||||
return value if isinstance(value, (dict, list)) else {}
|
||||
except Exception:
|
||||
pass
|
||||
return {}
|
||||
|
||||
@staticmethod
|
||||
def _clean_body(text):
|
||||
text = re.sub(r"//.*", "", text)
|
||||
text = re.sub(r"/\*.*?\*/", "", text, flags=re.S)
|
||||
text = text.replace("{{token}}", "")
|
||||
text = re.sub(r",\s*([}\]])", r"\1", text)
|
||||
return text.strip()
|
||||
|
||||
@staticmethod
|
||||
def _coerce_value(value):
|
||||
if value in ("-", "暂无参数", ""):
|
||||
return ""
|
||||
if value.isdigit():
|
||||
return int(value)
|
||||
if value.lower() in ("true", "false"):
|
||||
return value.lower() == "true"
|
||||
return value
|
||||
|
||||
|
||||
def load_hubops_cases():
|
||||
return HubOpsParser().parse()
|
||||
131
joyhub_backend/library/joyhub_interface.py
Normal file
131
joyhub_backend/library/joyhub_interface.py
Normal file
@@ -0,0 +1,131 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
from urllib.parse import urljoin
|
||||
|
||||
import allure
|
||||
import requests
|
||||
|
||||
from joyhub_backend.library.auth import JoyhubAuth, TIMEOUT
|
||||
|
||||
|
||||
MANAGER_BASE_URL = os.getenv(
|
||||
"JOYHUB_MANAGER_BASE_URL",
|
||||
"http://test-manager-api.best-envision.com",
|
||||
)
|
||||
|
||||
|
||||
class JoyhubInterface(object):
|
||||
def __init__(self):
|
||||
self.auth = JoyhubAuth()
|
||||
self.session = requests.Session()
|
||||
self.base_url = MANAGER_BASE_URL.rstrip("/") + "/"
|
||||
|
||||
def request(self, case_name, method, path, body=None, query=None, headers=None, expected_code=0):
|
||||
url = path if path.startswith("http") else urljoin(self.base_url, path.lstrip("/"))
|
||||
request_headers = self.auth.auth_headers()
|
||||
if headers:
|
||||
request_headers.update({key: value for key, value in headers.items() if key.lower() != "authorization"})
|
||||
|
||||
with allure.step("操作步骤:{}".format(case_name)):
|
||||
self._attach_request(url, method, request_headers, body, query)
|
||||
logging.info("case: %s", case_name)
|
||||
logging.info("request url: %s", url)
|
||||
logging.info("request method: %s", method)
|
||||
logging.info("request headers: %s", request_headers)
|
||||
logging.info("request body: %s", body)
|
||||
logging.info("request query: %s", query)
|
||||
|
||||
request_kwargs = {
|
||||
"method": method,
|
||||
"url": url,
|
||||
"params": query,
|
||||
"headers": request_headers,
|
||||
"timeout": TIMEOUT,
|
||||
}
|
||||
if method.upper() in ("POST", "PUT", "PATCH"):
|
||||
request_kwargs["json"] = body
|
||||
elif body:
|
||||
request_kwargs["params"] = dict(query or {}, **body) if isinstance(body, dict) else query
|
||||
response = self.session.request(**request_kwargs)
|
||||
self._attach_response(response)
|
||||
self._attach_log(case_name, url, method, request_headers, body, query, response)
|
||||
logging.info("response status: %s", response.status_code)
|
||||
logging.info("response body: %s", response.text)
|
||||
|
||||
is_business_api = self._is_business_api(url)
|
||||
assertion_text = "HTTP状态码为200,且业务code为{}".format(expected_code) if is_business_api else "HTTP状态码为2xx,兼容非JSON/SSE响应"
|
||||
allure.attach(assertion_text, "断言内容", allure.attachment_type.TEXT)
|
||||
with allure.step("断言内容:{}".format(assertion_text)):
|
||||
try:
|
||||
if is_business_api:
|
||||
assert response.status_code == 200, "HTTP状态码期望200,实际{},响应{}".format(
|
||||
response.status_code, response.text
|
||||
)
|
||||
data = self._safe_json(response)
|
||||
assert isinstance(data, dict), "业务接口响应不是JSON对象,响应{}".format(response.text)
|
||||
assert "code" in data, "响应缺少 code 字段,响应{}".format(data)
|
||||
assert data.get("code") == expected_code, "业务code期望{},实际{},msg={},响应{}".format(
|
||||
expected_code, data.get("code"), data.get("msg"), data
|
||||
)
|
||||
else:
|
||||
assert 200 <= response.status_code < 300, "HTTP状态码期望2xx,实际{},响应{}".format(
|
||||
response.status_code, response.text
|
||||
)
|
||||
data = self._non_json_result(response)
|
||||
allure.attach("无", "断言失败原因", allure.attachment_type.TEXT)
|
||||
return data
|
||||
except AssertionError as error:
|
||||
allure.attach(str(error), "断言失败原因", allure.attachment_type.TEXT)
|
||||
raise AssertionError("断言失败原因:{}".format(error))
|
||||
|
||||
@staticmethod
|
||||
def _attach_request(url, method, headers, body, query):
|
||||
allure.attach(str(url), "请求url", allure.attachment_type.TEXT)
|
||||
allure.attach(str(method), "请求方式", allure.attachment_type.TEXT)
|
||||
allure.attach(json.dumps(headers, ensure_ascii=False, indent=2), "请求头", allure.attachment_type.JSON)
|
||||
allure.attach(json.dumps(query or {}, ensure_ascii=False, indent=2), "请求query", allure.attachment_type.JSON)
|
||||
allure.attach(json.dumps(body or {}, ensure_ascii=False, indent=2), "请求体", allure.attachment_type.JSON)
|
||||
|
||||
@staticmethod
|
||||
def _attach_response(response):
|
||||
allure.attach(str(response.status_code), "响应状态码", allure.attachment_type.TEXT)
|
||||
content_type = response.headers.get("Content-Type", "")
|
||||
attachment_type = allure.attachment_type.JSON if "json" in content_type.lower() else allure.attachment_type.TEXT
|
||||
allure.attach(response.text, "响应体/日志", attachment_type)
|
||||
|
||||
def _is_business_api(self, url):
|
||||
return url.startswith(self.base_url)
|
||||
|
||||
@staticmethod
|
||||
def _safe_json(response):
|
||||
if not response.text.strip():
|
||||
return None
|
||||
return response.json()
|
||||
|
||||
@staticmethod
|
||||
def _non_json_result(response):
|
||||
try:
|
||||
data = response.json()
|
||||
except ValueError:
|
||||
data = {
|
||||
"status_code": response.status_code,
|
||||
"content_type": response.headers.get("Content-Type", ""),
|
||||
"text": response.text,
|
||||
}
|
||||
return data
|
||||
|
||||
@staticmethod
|
||||
def _attach_log(case_name, url, method, headers, body, query, response):
|
||||
log_text = "\n".join([
|
||||
"case: {}".format(case_name),
|
||||
"request url: {}".format(url),
|
||||
"request method: {}".format(method),
|
||||
"request headers: {}".format(headers),
|
||||
"request query: {}".format(query or {}),
|
||||
"request body: {}".format(body or {}),
|
||||
"response status: {}".format(response.status_code),
|
||||
"response body: {}".format(response.text),
|
||||
])
|
||||
allure.attach(log_text, "日志", allure.attachment_type.TEXT)
|
||||
@@ -0,0 +1,136 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import allure
|
||||
import logging
|
||||
import requests
|
||||
import json
|
||||
import pytest
|
||||
|
||||
try:
|
||||
from joyhub_backend.library.joyhub_interface import JoyhubInterface
|
||||
except Exception:
|
||||
JoyhubInterface = None
|
||||
|
||||
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
||||
|
||||
BASE_URL = "http://test-manager-api.best-envision.com"
|
||||
API_PATH = "admin/video/getVideoLabelList"
|
||||
CASE_NAME = "获取视频标签列表"
|
||||
|
||||
|
||||
def _mask_sensitive(data):
|
||||
if data is None:
|
||||
return data
|
||||
|
||||
sensitive_keys = {"authorization", "token", "access_token", "accessToken", "cookie", "password", "passwd"}
|
||||
|
||||
if isinstance(data, dict):
|
||||
masked = {}
|
||||
for key, value in data.items():
|
||||
if str(key).lower() in sensitive_keys:
|
||||
masked[key] = "***MASKED***"
|
||||
else:
|
||||
masked[key] = _mask_sensitive(value)
|
||||
return masked
|
||||
|
||||
if isinstance(data, list):
|
||||
return [_mask_sensitive(item) for item in data]
|
||||
|
||||
return data
|
||||
|
||||
|
||||
def _attach_json(name, data):
|
||||
allure.attach(
|
||||
json.dumps(_mask_sensitive(data), ensure_ascii=False, indent=2),
|
||||
name,
|
||||
allure.attachment_type.JSON
|
||||
)
|
||||
|
||||
|
||||
def _normalize_response(response):
|
||||
if isinstance(response, requests.Response):
|
||||
try:
|
||||
response_json = response.json()
|
||||
except ValueError:
|
||||
response_json = None
|
||||
return response.status_code, response_json, response.text
|
||||
|
||||
if isinstance(response, dict):
|
||||
return 200, response, json.dumps(response, ensure_ascii=False)
|
||||
|
||||
return None, None, str(response)
|
||||
|
||||
|
||||
@allure.feature("接口")
|
||||
class TestHubopsVideoLabelList(object):
|
||||
|
||||
def setup_method(self):
|
||||
logging.info("-----------------------------Test Start-------------------------------")
|
||||
|
||||
def teardown_method(self):
|
||||
logging.info("-----------------------------Test End-------------------------------")
|
||||
|
||||
@allure.story("获取视频标签列表")
|
||||
@allure.title("测试HubOps获取视频标签列表接口")
|
||||
def test_get_video_label_list(self):
|
||||
if JoyhubInterface is None:
|
||||
pytest.skip("joyhub_backend项目优先使用JoyhubInterface鉴权封装,请确认joyhub_backend.library.joyhub_interface.JoyhubInterface可导入")
|
||||
|
||||
with allure.step("1. 准备请求参数"):
|
||||
method = "POST"
|
||||
headers = {
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
body = {
|
||||
"page": 1,
|
||||
"limit": 10,
|
||||
"sort": "id",
|
||||
"label_name": "",
|
||||
"category_id": 0,
|
||||
"video_type": 0,
|
||||
"video_num": 0,
|
||||
"created_at": [],
|
||||
"order": "descending"
|
||||
}
|
||||
|
||||
logging.info("请求方法: %s", method)
|
||||
logging.info("请求路径: %s", API_PATH)
|
||||
_attach_json("请求头", headers)
|
||||
_attach_json("请求体", body)
|
||||
|
||||
with allure.step("2. 调用HubOps获取视频标签列表接口"):
|
||||
client = JoyhubInterface()
|
||||
response = client.request(
|
||||
CASE_NAME,
|
||||
method,
|
||||
API_PATH,
|
||||
body=body,
|
||||
headers=headers,
|
||||
expected_code=0
|
||||
)
|
||||
|
||||
status_code, response_json, response_text = _normalize_response(response)
|
||||
allure.attach(str(status_code), "HTTP状态码", allure.attachment_type.TEXT)
|
||||
|
||||
if response_json is not None:
|
||||
_attach_json("响应JSON", response_json)
|
||||
else:
|
||||
allure.attach(response_text, "响应内容", allure.attachment_type.TEXT)
|
||||
|
||||
with allure.step("3. 校验响应基础字段"):
|
||||
assert status_code == 200, "HTTP状态码应为200,实际为: {}".format(status_code)
|
||||
assert response_json is not None, "响应内容应为JSON且不能为空"
|
||||
assert isinstance(response_json, dict), "响应JSON应为对象类型"
|
||||
|
||||
assert "code" in response_json, "响应JSON应包含code字段"
|
||||
assert response_json.get("code") == 0, "响应code应为0,实际为: {}".format(response_json.get("code"))
|
||||
|
||||
assert "msg" in response_json, "响应JSON应包含msg字段"
|
||||
assert response_json.get("msg") is not None, "响应msg字段不应为None"
|
||||
|
||||
assert "data" in response_json, "响应JSON应包含data字段"
|
||||
assert response_json.get("data") is not None, "响应data字段不应为None"
|
||||
assert isinstance(response_json.get("data"), list), "响应data字段应为数组类型"
|
||||
|
||||
assert "count" in response_json, "响应JSON应包含count字段"
|
||||
assert response_json.get("count") is not None, "响应count字段不应为None"
|
||||
57
joyhub_backend/test_case/TestCase/接口/test_hub_ops.py
Normal file
57
joyhub_backend/test_case/TestCase/接口/test_hub_ops.py
Normal file
@@ -0,0 +1,57 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import logging
|
||||
import re
|
||||
from urllib.parse import urljoin
|
||||
|
||||
import allure
|
||||
import pytest
|
||||
|
||||
from joyhub_backend.library.hubops_parser import load_hubops_cases
|
||||
from joyhub_backend.library.joyhub_interface import JoyhubInterface
|
||||
|
||||
|
||||
ALL_CASES = load_hubops_cases()
|
||||
|
||||
|
||||
def case_id(case):
|
||||
raw_id = "{}-{}".format(case.get("case_id"), case.get("name"))
|
||||
safe_id = re.sub(r"[^0-9A-Za-z_\u4e00-\u9fa5-]+", "_", raw_id)
|
||||
return safe_id[:120]
|
||||
|
||||
|
||||
@allure.feature("JoyHub Backend 接口自动化")
|
||||
class TestHubOps(object):
|
||||
test_case = JoyhubInterface()
|
||||
|
||||
def teardown_method(self):
|
||||
with allure.step("后置:记录用例结束日志"):
|
||||
logging.info("-----------------------------End-------------------------------")
|
||||
|
||||
@pytest.mark.parametrize("case", ALL_CASES, ids=case_id)
|
||||
def test_hub_ops_api(self, case):
|
||||
allure.dynamic.title("{} {}".format(case.get("case_id"), case.get("name")))
|
||||
allure.dynamic.story(case.get("name"))
|
||||
|
||||
with allure.step("前置:初始化接口客户端并准备鉴权 token"):
|
||||
headers = case.get("headers") or {}
|
||||
query = case.get("query") or {}
|
||||
body = case.get("body") or {}
|
||||
|
||||
response_data = self.test_case.request(
|
||||
case.get("name"),
|
||||
case.get("method"),
|
||||
case.get("url"),
|
||||
body=body,
|
||||
query=query,
|
||||
headers=headers,
|
||||
)
|
||||
with allure.step("断言内容:校验响应基础字段"):
|
||||
request_url = case.get("url") or ""
|
||||
full_url = request_url if request_url.startswith("http") else urljoin(self.test_case.base_url, request_url.lstrip("/"))
|
||||
if full_url.startswith(self.test_case.base_url):
|
||||
assert isinstance(response_data, dict), "断言失败原因:响应不是JSON对象,响应{}".format(response_data)
|
||||
assert "code" in response_data, "断言失败原因:响应缺少code字段,响应{}".format(response_data)
|
||||
assert response_data.get("msg") is not None, "断言失败原因:msg字段不能为空,响应{}".format(response_data)
|
||||
else:
|
||||
assert response_data is not None, "断言失败原因:非业务接口响应为空"
|
||||
logging.info("断言通过:%s", case.get("name"))
|
||||
0
joyhub_backend/test_case/__init__.py
Normal file
0
joyhub_backend/test_case/__init__.py
Normal file
170
joyhub_backend/test_case/run_tests.py
Normal file
170
joyhub_backend/test_case/run_tests.py
Normal file
@@ -0,0 +1,170 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import argparse
|
||||
import os
|
||||
import shutil
|
||||
import subprocess
|
||||
import sys
|
||||
|
||||
current_file_path = os.path.abspath(__file__)
|
||||
project_root = os.path.abspath(os.path.join(os.path.dirname(current_file_path), '../../'))
|
||||
if project_root not in sys.path:
|
||||
sys.path.insert(0, project_root)
|
||||
|
||||
TEST_CASE_DIR = 'joyhub_backend/test_case/TestCase'
|
||||
case_dir = os.path.join(project_root, TEST_CASE_DIR)
|
||||
ALLURE_RESULTS_DIR = os.path.join(project_root, 'joyhub_backend', 'test_case', 'reports', 'allure-results')
|
||||
ALLURE_REPORT_DIR = os.path.join(project_root, 'joyhub_backend', 'test_case', 'reports', 'allure-report')
|
||||
|
||||
|
||||
def ensure_dirs():
|
||||
os.makedirs(ALLURE_RESULTS_DIR, exist_ok=True)
|
||||
os.makedirs(ALLURE_REPORT_DIR, exist_ok=True)
|
||||
|
||||
|
||||
def clean_allure_results():
|
||||
if os.path.exists(ALLURE_RESULTS_DIR):
|
||||
shutil.rmtree(ALLURE_RESULTS_DIR)
|
||||
os.makedirs(ALLURE_RESULTS_DIR, exist_ok=True)
|
||||
|
||||
|
||||
def find_test_files(directory):
|
||||
test_files = []
|
||||
for root, dirs, files in os.walk(directory):
|
||||
for file in files:
|
||||
if file.endswith('.py') and not file.startswith('__') and file != 'conftest.py':
|
||||
test_files.append(os.path.join(root, file))
|
||||
return test_files
|
||||
|
||||
|
||||
def run_pytest(args_list):
|
||||
env = os.environ.copy()
|
||||
env['PYTHONPATH'] = project_root + (os.pathsep + env['PYTHONPATH'] if 'PYTHONPATH' in env else '')
|
||||
cmd = ['python', '-m', 'pytest'] + args_list
|
||||
print("开始执行pytest...")
|
||||
print("执行命令: {}".format(' '.join('"{}"'.format(item) if ' ' in item else item for item in cmd)), flush=True)
|
||||
result = subprocess.run(cmd, cwd=project_root, env=env)
|
||||
print("pytest执行结束,退出码: {}".format(result.returncode), flush=True)
|
||||
return result.returncode
|
||||
|
||||
|
||||
def run_tests(target=None, test_type='all'):
|
||||
base_args = ['-v', '--tb=short', '--alluredir={}'.format(ALLURE_RESULTS_DIR)]
|
||||
if test_type == 'all':
|
||||
print("运行所有测试用例...")
|
||||
test_files = find_test_files(case_dir)
|
||||
if not test_files:
|
||||
print("错误: 未找到测试文件")
|
||||
return 1
|
||||
args = test_files + base_args
|
||||
elif test_type == 'dir':
|
||||
full_path = os.path.join(case_dir, target.replace('/', os.sep).replace('\\', os.sep))
|
||||
if not os.path.exists(full_path):
|
||||
print("错误: 目录不存在: {}".format(full_path))
|
||||
return 1
|
||||
print("按目录运行: {}".format(target))
|
||||
test_files = find_test_files(full_path)
|
||||
if not test_files:
|
||||
print("错误: 未找到测试文件")
|
||||
return 1
|
||||
args = test_files + base_args
|
||||
elif test_type == 'file':
|
||||
full_path = os.path.join(case_dir, target.replace('/', os.sep).replace('\\', os.sep))
|
||||
if not os.path.exists(full_path):
|
||||
print("错误: 文件不存在: {}".format(full_path))
|
||||
return 1
|
||||
print("按文件运行: {}".format(target))
|
||||
args = [full_path] + base_args
|
||||
elif test_type == 'keyword':
|
||||
print("按关键字运行: {}".format(target))
|
||||
test_files = find_test_files(case_dir)
|
||||
args = test_files + ['-k={}'.format(target)] + base_args
|
||||
elif test_type == 'marker':
|
||||
print("按pytest标记运行: {}".format(target))
|
||||
test_files = find_test_files(case_dir)
|
||||
args = test_files + ['-m={}'.format(target)] + base_args
|
||||
elif test_type == 'feature':
|
||||
print("按Allure feature运行: {}".format(target))
|
||||
test_files = find_test_files(case_dir)
|
||||
args = test_files + ['--allure-features={}'.format(target)] + base_args
|
||||
elif test_type == 'story':
|
||||
print("按Allure story运行: {}".format(target))
|
||||
test_files = find_test_files(case_dir)
|
||||
args = test_files + ['--allure-stories={}'.format(target)] + base_args
|
||||
else:
|
||||
print("错误: 未知的测试类型: {}".format(test_type))
|
||||
return 1
|
||||
return run_pytest(args)
|
||||
|
||||
|
||||
def generate_allure_report():
|
||||
print("开始生成Allure报告...", flush=True)
|
||||
if os.path.exists(ALLURE_REPORT_DIR):
|
||||
shutil.rmtree(ALLURE_REPORT_DIR)
|
||||
cmd = 'allure generate "{}" --output "{}"'.format(ALLURE_RESULTS_DIR, ALLURE_REPORT_DIR)
|
||||
print("执行命令: {}".format(cmd), flush=True)
|
||||
try:
|
||||
subprocess.run(cmd, check=True, shell=True)
|
||||
print("Allure报告生成成功: {}".format(ALLURE_REPORT_DIR))
|
||||
print("打开报告命令: allure open \"{}\"".format(ALLURE_REPORT_DIR))
|
||||
return 0
|
||||
except (subprocess.CalledProcessError, FileNotFoundError, OSError) as error:
|
||||
print("生成Allure报告失败: {}".format(error))
|
||||
print("手动执行: {}".format(cmd))
|
||||
return 1
|
||||
|
||||
|
||||
def open_allure_report():
|
||||
cmd = 'allure open "{}"'.format(ALLURE_REPORT_DIR)
|
||||
try:
|
||||
subprocess.Popen(cmd, shell=True)
|
||||
print("Allure报告已打开: {}".format(ALLURE_REPORT_DIR))
|
||||
return 0
|
||||
except (FileNotFoundError, OSError) as error:
|
||||
print("打开Allure报告失败: {}".format(error))
|
||||
return 1
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='JoyHub Backend 接口自动化测试执行工具')
|
||||
run_group = parser.add_mutually_exclusive_group(required=False)
|
||||
run_group.add_argument('--feature', type=str, help='按Allure feature运行')
|
||||
run_group.add_argument('--story', type=str, help='按Allure story运行')
|
||||
run_group.add_argument('--dir', type=str, help='按目录运行(相对于TestCase目录)')
|
||||
run_group.add_argument('--file', type=str, help='按文件运行(相对于TestCase目录)')
|
||||
run_group.add_argument('--keyword', type=str, help='按关键字运行')
|
||||
run_group.add_argument('--marker', type=str, help='按pytest标记运行')
|
||||
parser.add_argument('--report', action='store_true', help='生成Allure报告')
|
||||
parser.add_argument('--open', action='store_true', help='打开Allure报告')
|
||||
parser.add_argument('--no-report', action='store_true', help='不生成Allure报告')
|
||||
args = parser.parse_args()
|
||||
|
||||
ensure_dirs()
|
||||
clean_allure_results()
|
||||
if args.feature:
|
||||
exit_code = run_tests(args.feature, 'feature')
|
||||
elif args.story:
|
||||
exit_code = run_tests(args.story, 'story')
|
||||
elif args.dir:
|
||||
exit_code = run_tests(args.dir, 'dir')
|
||||
elif args.file:
|
||||
exit_code = run_tests(args.file, 'file')
|
||||
elif args.keyword:
|
||||
exit_code = run_tests(args.keyword, 'keyword')
|
||||
elif args.marker:
|
||||
exit_code = run_tests(args.marker, 'marker')
|
||||
else:
|
||||
exit_code = run_tests()
|
||||
|
||||
if args.report or not args.no_report:
|
||||
generate_allure_report()
|
||||
if args.open:
|
||||
open_allure_report()
|
||||
|
||||
print("=" * 80)
|
||||
print("测试执行完成" if exit_code == 0 else "测试执行失败,退出码: {}".format(exit_code))
|
||||
print("=" * 80)
|
||||
sys.exit(exit_code)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
202
skills/webapp-testing/LICENSE.txt
Normal file
202
skills/webapp-testing/LICENSE.txt
Normal file
@@ -0,0 +1,202 @@
|
||||
|
||||
Apache License
|
||||
Version 2.0, January 2004
|
||||
http://www.apache.org/licenses/
|
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||
|
||||
1. Definitions.
|
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction,
|
||||
and distribution as defined by Sections 1 through 9 of this document.
|
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by
|
||||
the copyright owner that is granting the License.
|
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all
|
||||
other entities that control, are controlled by, or are under common
|
||||
control with that entity. For the purposes of this definition,
|
||||
"control" means (i) the power, direct or indirect, to cause the
|
||||
direction or management of such entity, whether by contract or
|
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity
|
||||
exercising permissions granted by this License.
|
||||
|
||||
"Source" form shall mean the preferred form for making modifications,
|
||||
including but not limited to software source code, documentation
|
||||
source, and configuration files.
|
||||
|
||||
"Object" form shall mean any form resulting from mechanical
|
||||
transformation or translation of a Source form, including but
|
||||
not limited to compiled object code, generated documentation,
|
||||
and conversions to other media types.
|
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or
|
||||
Object form, made available under the License, as indicated by a
|
||||
copyright notice that is included in or attached to the work
|
||||
(an example is provided in the Appendix below).
|
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object
|
||||
form, that is based on (or derived from) the Work and for which the
|
||||
editorial revisions, annotations, elaborations, or other modifications
|
||||
represent, as a whole, an original work of authorship. For the purposes
|
||||
of this License, Derivative Works shall not include works that remain
|
||||
separable from, or merely link (or bind by name) to the interfaces of,
|
||||
the Work and Derivative Works thereof.
|
||||
|
||||
"Contribution" shall mean any work of authorship, including
|
||||
the original version of the Work and any modifications or additions
|
||||
to that Work or Derivative Works thereof, that is intentionally
|
||||
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||
or by an individual or Legal Entity authorized to submit on behalf of
|
||||
the copyright owner. For the purposes of this definition, "submitted"
|
||||
means any form of electronic, verbal, or written communication sent
|
||||
to the Licensor or its representatives, including but not limited to
|
||||
communication on electronic mailing lists, source code control systems,
|
||||
and issue tracking systems that are managed by, or on behalf of, the
|
||||
Licensor for the purpose of discussing and improving the Work, but
|
||||
excluding communication that is conspicuously marked or otherwise
|
||||
designated in writing by the copyright owner as "Not a Contribution."
|
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||
on behalf of whom a Contribution has been received by Licensor and
|
||||
subsequently incorporated within the Work.
|
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
copyright license to reproduce, prepare Derivative Works of,
|
||||
publicly display, publicly perform, sublicense, and distribute the
|
||||
Work and such Derivative Works in Source or Object form.
|
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of
|
||||
this License, each Contributor hereby grants to You a perpetual,
|
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||
(except as stated in this section) patent license to make, have made,
|
||||
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||
where such license applies only to those patent claims licensable
|
||||
by such Contributor that are necessarily infringed by their
|
||||
Contribution(s) alone or by combination of their Contribution(s)
|
||||
with the Work to which such Contribution(s) was submitted. If You
|
||||
institute patent litigation against any entity (including a
|
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||
or a Contribution incorporated within the Work constitutes direct
|
||||
or contributory patent infringement, then any patent licenses
|
||||
granted to You under this License for that Work shall terminate
|
||||
as of the date such litigation is filed.
|
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the
|
||||
Work or Derivative Works thereof in any medium, with or without
|
||||
modifications, and in Source or Object form, provided that You
|
||||
meet the following conditions:
|
||||
|
||||
(a) You must give any other recipients of the Work or
|
||||
Derivative Works a copy of this License; and
|
||||
|
||||
(b) You must cause any modified files to carry prominent notices
|
||||
stating that You changed the files; and
|
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works
|
||||
that You distribute, all copyright, patent, trademark, and
|
||||
attribution notices from the Source form of the Work,
|
||||
excluding those notices that do not pertain to any part of
|
||||
the Derivative Works; and
|
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its
|
||||
distribution, then any Derivative Works that You distribute must
|
||||
include a readable copy of the attribution notices contained
|
||||
within such NOTICE file, excluding those notices that do not
|
||||
pertain to any part of the Derivative Works, in at least one
|
||||
of the following places: within a NOTICE text file distributed
|
||||
as part of the Derivative Works; within the Source form or
|
||||
documentation, if provided along with the Derivative Works; or,
|
||||
within a display generated by the Derivative Works, if and
|
||||
wherever such third-party notices normally appear. The contents
|
||||
of the NOTICE file are for informational purposes only and
|
||||
do not modify the License. You may add Your own attribution
|
||||
notices within Derivative Works that You distribute, alongside
|
||||
or as an addendum to the NOTICE text from the Work, provided
|
||||
that such additional attribution notices cannot be construed
|
||||
as modifying the License.
|
||||
|
||||
You may add Your own copyright statement to Your modifications and
|
||||
may provide additional or different license terms and conditions
|
||||
for use, reproduction, or distribution of Your modifications, or
|
||||
for any such Derivative Works as a whole, provided Your use,
|
||||
reproduction, and distribution of the Work otherwise complies with
|
||||
the conditions stated in this License.
|
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||
any Contribution intentionally submitted for inclusion in the Work
|
||||
by You to the Licensor shall be under the terms and conditions of
|
||||
this License, without any additional terms or conditions.
|
||||
Notwithstanding the above, nothing herein shall supersede or modify
|
||||
the terms of any separate license agreement you may have executed
|
||||
with Licensor regarding such Contributions.
|
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade
|
||||
names, trademarks, service marks, or product names of the Licensor,
|
||||
except as required for reasonable and customary use in describing the
|
||||
origin of the Work and reproducing the content of the NOTICE file.
|
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||
agreed to in writing, Licensor provides the Work (and each
|
||||
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||
implied, including, without limitation, any warranties or conditions
|
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||
appropriateness of using or redistributing the Work and assume any
|
||||
risks associated with Your exercise of permissions under this License.
|
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory,
|
||||
whether in tort (including negligence), contract, or otherwise,
|
||||
unless required by applicable law (such as deliberate and grossly
|
||||
negligent acts) or agreed to in writing, shall any Contributor be
|
||||
liable to You for damages, including any direct, indirect, special,
|
||||
incidental, or consequential damages of any character arising as a
|
||||
result of this License or out of the use or inability to use the
|
||||
Work (including but not limited to damages for loss of goodwill,
|
||||
work stoppage, computer failure or malfunction, or any and all
|
||||
other commercial damages or losses), even if such Contributor
|
||||
has been advised of the possibility of such damages.
|
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing
|
||||
the Work or Derivative Works thereof, You may choose to offer,
|
||||
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||
or other liability obligations and/or rights consistent with this
|
||||
License. However, in accepting such obligations, You may act only
|
||||
on Your own behalf and on Your sole responsibility, not on behalf
|
||||
of any other Contributor, and only if You agree to indemnify,
|
||||
defend, and hold each Contributor harmless for any liability
|
||||
incurred by, or claims asserted against, such Contributor by reason
|
||||
of your accepting any such warranty or additional liability.
|
||||
|
||||
END OF TERMS AND CONDITIONS
|
||||
|
||||
APPENDIX: How to apply the Apache License to your work.
|
||||
|
||||
To apply the Apache License to your work, attach the following
|
||||
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||
replaced with your own identifying information. (Don't include
|
||||
the brackets!) The text should be enclosed in the appropriate
|
||||
comment syntax for the file format. We also recommend that a
|
||||
file or class name and description of purpose be included on the
|
||||
same "printed page" as the copyright notice for easier
|
||||
identification within third-party archives.
|
||||
|
||||
Copyright 2026 Anthropic, PBC.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
181
skills/webapp-testing/SKILL.md
Normal file
181
skills/webapp-testing/SKILL.md
Normal file
@@ -0,0 +1,181 @@
|
||||
***
|
||||
|
||||
name: webapp-testing
|
||||
description: Toolkit for interacting with and testing local web applications using Playwright. Supports verifying frontend functionality, debugging UI behavior, capturing browser screenshots, and viewing browser logs.
|
||||
license: Complete terms in LICENSE.txt
|
||||
--------------------------------------
|
||||
|
||||
# Web Application Testing
|
||||
|
||||
To test local web applications, write native Python Playwright scripts.
|
||||
|
||||
**Helper Scripts Available**:
|
||||
|
||||
- `scripts/with_server.py` - Manages server lifecycle (supports multiple servers)
|
||||
|
||||
**Always run scripts with** **`--help`** **first** to see usage. DO NOT read the source until you try running the script first and find that a customized solution is abslutely necessary. These scripts can be very large and thus pollute your context window. They exist to be called directly as black-box scripts rather than ingested into your context window.
|
||||
|
||||
## Decision Tree: Choosing Your Approach
|
||||
|
||||
```
|
||||
User task → Is it static HTML?
|
||||
├─ Yes → Read HTML file directly to identify selectors
|
||||
│ ├─ Success → Write Playwright script using selectors
|
||||
│ └─ Fails/Incomplete → Treat as dynamic (below)
|
||||
│
|
||||
└─ No (dynamic webapp) → Is the server already running?
|
||||
├─ No → Run: python scripts/with_server.py --help
|
||||
│ Then use the helper + write simplified Playwright script
|
||||
│
|
||||
└─ Yes → Reconnaissance-then-action:
|
||||
1. Navigate and wait for networkidle
|
||||
2. Take screenshot or inspect DOM
|
||||
3. Identify selectors from rendered state
|
||||
4. Execute actions with discovered selectors
|
||||
```
|
||||
|
||||
## Example: Using with\_server.py
|
||||
|
||||
To start a server, run `--help` first, then use the helper:
|
||||
|
||||
**Single server:**
|
||||
|
||||
```bash
|
||||
python scripts/with_server.py --server "npm run dev" --port 5173 -- python your_automation.py
|
||||
```
|
||||
|
||||
**Multiple servers (e.g., backend + frontend):**
|
||||
|
||||
```bash
|
||||
python scripts/with_server.py \
|
||||
--server "cd backend && python server.py" --port 3000 \
|
||||
--server "cd frontend && npm run dev" --port 5173 \
|
||||
-- python your_automation.py
|
||||
```
|
||||
|
||||
To create an automation script, include only Playwright logic (servers are managed automatically):
|
||||
|
||||
```python
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True) # Always launch chromium in headless mode
|
||||
page = browser.new_page()
|
||||
page.goto('http://localhost:5173') # Server already running and ready
|
||||
page.wait_for_load_state('networkidle') # CRITICAL: Wait for JS to execute
|
||||
# ... your automation logic
|
||||
browser.close()
|
||||
```
|
||||
|
||||
## Reconnaissance-Then-Action Pattern
|
||||
|
||||
1. **Inspect rendered DOM**:
|
||||
```python
|
||||
page.screenshot(path='/tmp/inspect.png', full_page=True)
|
||||
content = page.content()
|
||||
page.locator('button').all()
|
||||
```
|
||||
2. **Identify selectors** from inspection results
|
||||
3. **Execute actions** using discovered selectors
|
||||
|
||||
## Common Pitfall
|
||||
|
||||
❌ **Don't** inspect the DOM before waiting for `networkidle` on dynamic apps
|
||||
✅ **Do** wait for `page.wait_for_load_state('networkidle')` before inspection
|
||||
|
||||
## Best Practices
|
||||
|
||||
- **Use bundled scripts as black boxes** - To accomplish a task, consider whether one of the scripts available in `scripts/` can help. These scripts handle common, complex workflows reliably without cluttering the context window. Use `--help` to see usage, then invoke directly.
|
||||
- Use `sync_playwright()` for synchronous scripts
|
||||
- Always close the browser when done
|
||||
- Use descriptive selectors: `text=`, `role=`, CSS selectors, or IDs
|
||||
- Add appropriate waits: `page.wait_for_selector()` or `page.wait_for_timeout()`
|
||||
|
||||
## Reference Files
|
||||
|
||||
- **examples/** - Examples showing common patterns:
|
||||
- `element_discovery.py` - Discovering buttons, links, and inputs on a page
|
||||
- `static_html_automation.py` - Using file:// URLs for local HTML
|
||||
- `console_logging.py` - Capturing console logs during automation
|
||||
|
||||
# UI Automation Testing Skill
|
||||
|
||||
你是一个资深 UI 自动化测试专家,擅长基于 Selenium、Playwright、pytest、unittest、Robot Framework、Allure 等技术体系设计和实现稳定、可维护、可扩展的 UI 自动化测试方案。
|
||||
|
||||
## 适用场景
|
||||
|
||||
当用户需要以下能力时,使用本 Skill:
|
||||
|
||||
- 编写 Web UI 自动化测试用例
|
||||
- 设计 Page Object / Page Object Model 框架
|
||||
- 封装页面元素、页面行为、业务流程
|
||||
- 优化 Selenium / Playwright 自动化脚本稳定性
|
||||
- 处理元素定位、等待、iframe、弹窗、上传下载、验证码等问题
|
||||
- 设计 pytest + Allure UI 自动化测试框架
|
||||
- 编写 UI 自动化断言、测试数据、公共方法
|
||||
- 分析 UI 自动化失败原因
|
||||
- 提升自动化用例可维护性和执行效率
|
||||
- 将手工测试场景转换为自动化测试用例
|
||||
|
||||
## 角色定位
|
||||
|
||||
你不是简单的代码生成器,而是 UI 自动化测试架构师和落地专家。
|
||||
|
||||
你需要:
|
||||
|
||||
1. 理解用户当前项目框架和代码风格;
|
||||
2. 优先复用已有封装,不重复造轮子;
|
||||
3. 保持用例稳定性、可读性和可维护性;
|
||||
4. 按照自动化测试最佳实践设计代码;
|
||||
5. 明确区分页面层、业务层、测试层;
|
||||
6. 对不稳定写法主动给出风险提示;
|
||||
7. 生成代码前先确认当前项目使用的技术栈和目录结构。
|
||||
|
||||
## 工作原则
|
||||
|
||||
### 1. 先理解项目
|
||||
|
||||
在编写代码前,优先查看以下内容:
|
||||
|
||||
- 项目目录结构
|
||||
- requirements.txt / pyproject.toml / package.json
|
||||
- conftest.py
|
||||
- pytest.ini / setup.cfg / tox.ini
|
||||
- 已有 Page Object 文件
|
||||
- 已有测试用例
|
||||
- 公共 driver / browser 封装
|
||||
- Allure 封装
|
||||
- 日志封装
|
||||
- 配置文件
|
||||
- 测试数据文件
|
||||
|
||||
不要在不了解项目结构的情况下直接生成孤立代码。
|
||||
|
||||
### 2. 分层设计
|
||||
|
||||
推荐使用以下结构:
|
||||
|
||||
```text
|
||||
tests/
|
||||
test_xxx.py 测试用例层,只做流程编排和断言
|
||||
|
||||
pages/
|
||||
xxx_page.py 页面对象层,封装元素和页面操作
|
||||
|
||||
flows/
|
||||
xxx_flow.py 业务流程层,封装跨页面业务流程
|
||||
|
||||
common/
|
||||
browser.py 浏览器/driver 管理
|
||||
base_page.py 基础页面封装
|
||||
wait.py 显式等待封装
|
||||
logger.py 日志封装
|
||||
assertions.py 断言封装
|
||||
|
||||
data/
|
||||
xxx_data.py / xxx.yaml 测试数据
|
||||
|
||||
config/
|
||||
config.py / env.yaml 环境配置
|
||||
```
|
||||
|
||||
35
skills/webapp-testing/examples/console_logging.py
Normal file
35
skills/webapp-testing/examples/console_logging.py
Normal file
@@ -0,0 +1,35 @@
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
# Example: Capturing console logs during browser automation
|
||||
|
||||
url = 'http://localhost:5173' # Replace with your URL
|
||||
|
||||
console_logs = []
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
page = browser.new_page(viewport={'width': 1920, 'height': 1080})
|
||||
|
||||
# Set up console log capture
|
||||
def handle_console_message(msg):
|
||||
console_logs.append(f"[{msg.type}] {msg.text}")
|
||||
print(f"Console: [{msg.type}] {msg.text}")
|
||||
|
||||
page.on("console", handle_console_message)
|
||||
|
||||
# Navigate to page
|
||||
page.goto(url)
|
||||
page.wait_for_load_state('networkidle')
|
||||
|
||||
# Interact with the page (triggers console logs)
|
||||
page.click('text=Dashboard')
|
||||
page.wait_for_timeout(1000)
|
||||
|
||||
browser.close()
|
||||
|
||||
# Save console logs to file
|
||||
with open('/mnt/user-data/outputs/console.log', 'w') as f:
|
||||
f.write('\n'.join(console_logs))
|
||||
|
||||
print(f"\nCaptured {len(console_logs)} console messages")
|
||||
print(f"Logs saved to: /mnt/user-data/outputs/console.log")
|
||||
40
skills/webapp-testing/examples/element_discovery.py
Normal file
40
skills/webapp-testing/examples/element_discovery.py
Normal file
@@ -0,0 +1,40 @@
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
# Example: Discovering buttons and other elements on a page
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
page = browser.new_page()
|
||||
|
||||
# Navigate to page and wait for it to fully load
|
||||
page.goto('http://localhost:5173')
|
||||
page.wait_for_load_state('networkidle')
|
||||
|
||||
# Discover all buttons on the page
|
||||
buttons = page.locator('button').all()
|
||||
print(f"Found {len(buttons)} buttons:")
|
||||
for i, button in enumerate(buttons):
|
||||
text = button.inner_text() if button.is_visible() else "[hidden]"
|
||||
print(f" [{i}] {text}")
|
||||
|
||||
# Discover links
|
||||
links = page.locator('a[href]').all()
|
||||
print(f"\nFound {len(links)} links:")
|
||||
for link in links[:5]: # Show first 5
|
||||
text = link.inner_text().strip()
|
||||
href = link.get_attribute('href')
|
||||
print(f" - {text} -> {href}")
|
||||
|
||||
# Discover input fields
|
||||
inputs = page.locator('input, textarea, select').all()
|
||||
print(f"\nFound {len(inputs)} input fields:")
|
||||
for input_elem in inputs:
|
||||
name = input_elem.get_attribute('name') or input_elem.get_attribute('id') or "[unnamed]"
|
||||
input_type = input_elem.get_attribute('type') or 'text'
|
||||
print(f" - {name} ({input_type})")
|
||||
|
||||
# Take screenshot for visual reference
|
||||
page.screenshot(path='/tmp/page_discovery.png', full_page=True)
|
||||
print("\nScreenshot saved to /tmp/page_discovery.png")
|
||||
|
||||
browser.close()
|
||||
33
skills/webapp-testing/examples/static_html_automation.py
Normal file
33
skills/webapp-testing/examples/static_html_automation.py
Normal file
@@ -0,0 +1,33 @@
|
||||
from playwright.sync_api import sync_playwright
|
||||
import os
|
||||
|
||||
# Example: Automating interaction with static HTML files using file:// URLs
|
||||
|
||||
html_file_path = os.path.abspath('path/to/your/file.html')
|
||||
file_url = f'file://{html_file_path}'
|
||||
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
page = browser.new_page(viewport={'width': 1920, 'height': 1080})
|
||||
|
||||
# Navigate to local HTML file
|
||||
page.goto(file_url)
|
||||
|
||||
# Take screenshot
|
||||
page.screenshot(path='/mnt/user-data/outputs/static_page.png', full_page=True)
|
||||
|
||||
# Interact with elements
|
||||
page.click('text=Click Me')
|
||||
page.fill('#name', 'John Doe')
|
||||
page.fill('#email', 'john@example.com')
|
||||
|
||||
# Submit form
|
||||
page.click('button[type="submit"]')
|
||||
page.wait_for_timeout(500)
|
||||
|
||||
# Take final screenshot
|
||||
page.screenshot(path='/mnt/user-data/outputs/after_submit.png', full_page=True)
|
||||
|
||||
browser.close()
|
||||
|
||||
print("Static HTML automation completed!")
|
||||
106
skills/webapp-testing/scripts/with_server.py
Normal file
106
skills/webapp-testing/scripts/with_server.py
Normal file
@@ -0,0 +1,106 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Start one or more servers, wait for them to be ready, run a command, then clean up.
|
||||
|
||||
Usage:
|
||||
# Single server
|
||||
python scripts/with_server.py --server "npm run dev" --port 5173 -- python automation.py
|
||||
python scripts/with_server.py --server "npm start" --port 3000 -- python test.py
|
||||
|
||||
# Multiple servers
|
||||
python scripts/with_server.py \
|
||||
--server "cd backend && python server.py" --port 3000 \
|
||||
--server "cd frontend && npm run dev" --port 5173 \
|
||||
-- python test.py
|
||||
"""
|
||||
|
||||
import subprocess
|
||||
import socket
|
||||
import time
|
||||
import sys
|
||||
import argparse
|
||||
|
||||
def is_server_ready(port, timeout=30):
|
||||
"""Wait for server to be ready by polling the port."""
|
||||
start_time = time.time()
|
||||
while time.time() - start_time < timeout:
|
||||
try:
|
||||
with socket.create_connection(('localhost', port), timeout=1):
|
||||
return True
|
||||
except (socket.error, ConnectionRefusedError):
|
||||
time.sleep(0.5)
|
||||
return False
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='Run command with one or more servers')
|
||||
parser.add_argument('--server', action='append', dest='servers', required=True, help='Server command (can be repeated)')
|
||||
parser.add_argument('--port', action='append', dest='ports', type=int, required=True, help='Port for each server (must match --server count)')
|
||||
parser.add_argument('--timeout', type=int, default=30, help='Timeout in seconds per server (default: 30)')
|
||||
parser.add_argument('command', nargs=argparse.REMAINDER, help='Command to run after server(s) ready')
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Remove the '--' separator if present
|
||||
if args.command and args.command[0] == '--':
|
||||
args.command = args.command[1:]
|
||||
|
||||
if not args.command:
|
||||
print("Error: No command specified to run")
|
||||
sys.exit(1)
|
||||
|
||||
# Parse server configurations
|
||||
if len(args.servers) != len(args.ports):
|
||||
print("Error: Number of --server and --port arguments must match")
|
||||
sys.exit(1)
|
||||
|
||||
servers = []
|
||||
for cmd, port in zip(args.servers, args.ports):
|
||||
servers.append({'cmd': cmd, 'port': port})
|
||||
|
||||
server_processes = []
|
||||
|
||||
try:
|
||||
# Start all servers
|
||||
for i, server in enumerate(servers):
|
||||
print(f"Starting server {i+1}/{len(servers)}: {server['cmd']}")
|
||||
|
||||
# Use shell=True to support commands with cd and &&
|
||||
process = subprocess.Popen(
|
||||
server['cmd'],
|
||||
shell=True,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE
|
||||
)
|
||||
server_processes.append(process)
|
||||
|
||||
# Wait for this server to be ready
|
||||
print(f"Waiting for server on port {server['port']}...")
|
||||
if not is_server_ready(server['port'], timeout=args.timeout):
|
||||
raise RuntimeError(f"Server failed to start on port {server['port']} within {args.timeout}s")
|
||||
|
||||
print(f"Server ready on port {server['port']}")
|
||||
|
||||
print(f"\nAll {len(servers)} server(s) ready")
|
||||
|
||||
# Run the command
|
||||
print(f"Running: {' '.join(args.command)}\n")
|
||||
result = subprocess.run(args.command)
|
||||
sys.exit(result.returncode)
|
||||
|
||||
finally:
|
||||
# Clean up all servers
|
||||
print(f"\nStopping {len(server_processes)} server(s)...")
|
||||
for i, process in enumerate(server_processes):
|
||||
try:
|
||||
process.terminate()
|
||||
process.wait(timeout=5)
|
||||
except subprocess.TimeoutExpired:
|
||||
process.kill()
|
||||
process.wait()
|
||||
print(f"Server {i+1} stopped")
|
||||
print("All servers stopped")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
1
tmp_generate_automation_response.json
Normal file
1
tmp_generate_automation_response.json
Normal file
@@ -0,0 +1 @@
|
||||
|
||||
@@ -0,0 +1,6 @@
|
||||
# -*- coding:utf-8 -*-
|
||||
import pytest
|
||||
|
||||
|
||||
def test_tc_zhyy_ui_api_verify_001():
|
||||
assert "智慧运营"
|
||||
Reference in New Issue
Block a user