iflow-mcp_galaxyxieyu_api-auto-test 0.1.0__py3-none-any.whl

atf/__init__.py

@@ -0,0 +1,48 @@
+"""API Test Framework
+自动化测试框架主包
+
+使用延迟导入，避免在导入 atf 包时加载所有依赖。
+"""
+
+from importlib import import_module
+
+__version__ = '1.0.0'
+__package_name__ = 'atf'
+
+__all__ = [
+    "Globals",
+    "ConfigManager",
+    "LogManager",
+    "RequestHandler",
+    "VariableResolver",
+    "AssertHandler",
+    "LoginHandler",
+    "CaseGenerator",
+    "run_tests",
+]
+
+_LAZY_IMPORTS = {
+    "Globals": "atf.core.globals",
+    "ConfigManager": "atf.core.config_manager",
+    "LogManager": "atf.core.log_manager",
+    "RequestHandler": "atf.core.request_handler",
+    "VariableResolver": "atf.core.variable_resolver",
+    "AssertHandler": "atf.core.assert_handler",
+    "LoginHandler": "atf.core.login_handler",
+    "CaseGenerator": "atf.case_generator",
+    "run_tests": "atf.runner",
+}
+
+
+def __getattr__(name):
+    module_path = _LAZY_IMPORTS.get(name)
+    if not module_path:
+        raise AttributeError(f"module 'atf' has no attribute '{name}'")
+    module = import_module(module_path)
+    value = getattr(module, name)
+    globals()[name] = value
+    return value
+
+
+def __dir__():
+    return sorted(list(globals().keys()) + list(_LAZY_IMPORTS.keys()))

atf/assets/report.css

@@ -0,0 +1,243 @@
+/* API Auto Test - 现代化报告主题 */
+
+:root {
+    --primary: #4f46e5;
+    --success: #10b981;
+    --danger: #ef4444;
+    --warning: #f59e0b;
+    --bg: #f8fafc;
+    --card: #ffffff;
+    --text: #1e293b;
+    --border: #e2e8f0;
+    --muted: #64748b;
+}
+
+body {
+    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+    background: var(--bg);
+    color: var(--text);
+    line-height: 1.6;
+    margin: 0;
+    padding: 24px;
+}
+
+h1 {
+    color: var(--primary);
+    font-size: 26px;
+    font-weight: 700;
+    margin: 0 0 8px 0;
+    padding-bottom: 16px;
+    border-bottom: 3px solid var(--primary);
+}
+
+h2 {
+    font-size: 18px;
+    font-weight: 600;
+    margin: 24px 0 12px 0;
+    color: var(--text);
+}
+
+/* 卡片容器 */
+#environment, .summary {
+    background: var(--card);
+    border-radius: 12px;
+    padding: 20px 24px;
+    margin: 20px 0;
+    box-shadow: 0 1px 3px rgba(0,0,0,0.08);
+    border: 1px solid var(--border);
+}
+
+/* 统计标签 */
+.summary__data {
+    display: flex;
+    flex-wrap: wrap;
+    gap: 12px;
+    margin-top: 12px;
+}
+
+.summary p, p.passed, p.failed, p.error, p.skipped, p.rerun {
+    display: inline-flex;
+    align-items: center;
+    margin: 0;
+    padding: 8px 16px;
+    border-radius: 8px;
+    font-weight: 600;
+    font-size: 14px;
+}
+
+p.passed, .passed { background: #dcfce7; color: #166534; }
+p.failed, .failed { background: #fee2e2; color: #991b1b; }
+p.error, .error { background: #fef3c7; color: #92400e; }
+p.skipped, .skipped { background: #e0e7ff; color: #3730a3; }
+p.rerun { background: #f1f5f9; color: #475569; }
+
+/* 表格 */
+table {
+    width: 100%;
+    border-collapse: separate;
+    border-spacing: 0;
+    background: var(--card);
+    border-radius: 12px;
+    overflow: hidden;
+    box-shadow: 0 1px 3px rgba(0,0,0,0.08);
+    margin: 20px 0;
+}
+
+thead tr {
+    background: linear-gradient(135deg, var(--primary) 0%, #6366f1 100%);
+}
+
+th {
+    color: white;
+    font-weight: 600;
+    padding: 14px 16px;
+    text-align: left;
+    font-size: 13px;
+    text-transform: uppercase;
+    letter-spacing: 0.5px;
+    border: none;
+}
+
+th.sortable {
+    cursor: pointer;
+    user-select: none;
+}
+
+th.sortable:hover {
+    background: rgba(255,255,255,0.1);
+}
+
+td {
+    padding: 14px 16px;
+    border-bottom: 1px solid var(--border);
+    font-size: 14px;
+    vertical-align: middle;
+}
+
+tbody tr:hover {
+    background: #f8fafc;
+}
+
+tbody tr:last-child td {
+    border-bottom: none;
+}
+
+/* 结果状态 */
+td.col-result {
+    font-weight: 700;
+    text-transform: uppercase;
+    font-size: 11px;
+    letter-spacing: 0.5px;
+}
+
+td.col-result.passed { color: #059669; }
+td.col-result.failed { color: #dc2626; }
+td.col-result.error { color: #d97706; }
+td.col-result.skipped { color: #6366f1; }
+
+/* 展开详情 */
+.extra, .log {
+    background: #f8fafc;
+    border-left: 4px solid var(--primary);
+    padding: 16px;
+    margin: 12px 0;
+    font-family: 'SF Mono', Monaco, Consolas, monospace;
+    font-size: 13px;
+    line-height: 1.5;
+    white-space: pre-wrap;
+    word-break: break-word;
+    overflow-x: auto;
+    border-radius: 0 8px 8px 0;
+}
+
+/* 环境信息 */
+#environment td {
+    padding: 10px 16px;
+}
+
+#environment td:first-child {
+    font-weight: 600;
+    color: var(--muted);
+    width: 180px;
+}
+
+/* 链接 */
+a {
+    color: var(--primary);
+    text-decoration: none;
+    font-weight: 500;
+}
+
+a:hover {
+    text-decoration: underline;
+}
+
+/* 筛选按钮 */
+.filter {
+    display: flex;
+    gap: 8px;
+    margin: 20px 0;
+    flex-wrap: wrap;
+}
+
+.filter button, button {
+    background: var(--card);
+    border: 1px solid var(--border);
+    padding: 8px 16px;
+    border-radius: 8px;
+    cursor: pointer;
+    font-size: 13px;
+    font-weight: 500;
+    transition: all 0.15s ease;
+    color: var(--text);
+}
+
+.filter button:hover, button:hover {
+    background: var(--primary);
+    color: white;
+    border-color: var(--primary);
+}
+
+.filter button.active {
+    background: var(--primary);
+    color: white;
+    border-color: var(--primary);
+}
+
+/* 折叠/展开按钮 */
+.expander {
+    cursor: pointer;
+    padding: 4px 8px;
+    border-radius: 4px;
+    font-size: 12px;
+    background: #f1f5f9;
+    border: none;
+    color: var(--muted);
+}
+
+.expander:hover {
+    background: var(--primary);
+    color: white;
+}
+
+/* 时间/持续时间 */
+td.col-time, td.col-duration {
+    font-family: 'SF Mono', Monaco, monospace;
+    font-size: 13px;
+    color: var(--muted);
+}
+
+/* 响应式 */
+@media (max-width: 768px) {
+    body { padding: 12px; }
+    th, td { padding: 10px 12px; font-size: 13px; }
+    h1 { font-size: 22px; }
+    .filter { flex-direction: column; }
+}
+
+/* 打印优化 */
+@media print {
+    body { background: white; padding: 0; }
+    table, #environment, .summary { box-shadow: none; border: 1px solid #ddd; }
+    .filter, button { display: none; }
+}

atf/auth.py

@@ -0,0 +1,99 @@
+# @time:    2024-08-15
+# @author:  xiaoqq
+
+import requests
+from atf.core.globals import Globals
+from atf.core.log_manager import log
+import base64
+
+
+class EncryptionManager:
+    """
+    加密管理器，用于处理不同项目的加密逻辑。
+    PS: 需要根据被测项目登录接口的加密方式进行定义，此处只是一个demo。
+    """
+    
+    def __init__(self):
+        self.public_keys = {
+            "merchant": {
+                "test": "test_public_key",
+                "online": "online_public_key"
+            },
+            "user": {
+                "test": "test_public_key",
+                "online": "test_public_key"
+            }
+        }
+    
+    def encrypt(self, src_str, project_name, env="pre"):
+        """
+        根据给定的项目名称使用RSA加密并转换为Base64格式。
+
+        :param src_str: 需要加密的原始字符串
+        :param project_name: 项目名称（"merchant" 或 "user"）
+        :param env: 环境名称（"test", "pre", 等），默认为 "pre"
+        :return: 加密后的字符串
+        """
+        from Crypto.Cipher import PKCS1_v1_5 as Cipher_pksc1_v1_5
+        from Crypto.PublicKey import RSA
+        
+        if env == "pre": # 预发与线上的签名一样
+            env = "online"
+        public_key = self.public_keys[project_name].get(env)
+        if not public_key:
+            log.error(f"No public key found for project '{project_name}' in environment '{env}'")
+            raise ValueError(f"No public key found for project '{project_name}' in environment '{env}'")
+
+        public_key_str = '-----BEGIN PUBLIC KEY-----\n' + public_key + '\n-----END PUBLIC KEY-----'
+        rsakey = RSA.importKey(public_key_str)
+        cipher = Cipher_pksc1_v1_5.new(rsakey)
+        cipher_text = base64.b64encode(cipher.encrypt(src_str.encode()))
+        return cipher_text.decode()
+
+
+class Auth:
+    """
+    项目登录
+    PS: 需要根据被测项目登录接口的加密方式进行定义，此处只是一个demo。
+    """
+    
+    def __init__(self):
+        self.encryption_manager = EncryptionManager()
+    
+    def login(self, project_name, login_config, env='pre', timeout=10):
+        url = login_config['url']
+        method = login_config['method']
+        headers = login_config.get('headers', {})
+        data = login_config['data'].copy()
+        
+        # 加密用户名和密码
+        if project_name in ["merchant", "user"]:
+            data['account'] = self.encryption_manager.encrypt(data['account'], project_name, env)
+            data['password'] = self.encryption_manager.encrypt(data['password'], project_name, env)
+        
+        # 执行登录请求
+        response = requests.request(method, url, json=data, headers=headers, timeout=timeout)
+        
+        # 提取token
+        token = response.json().get('data').get('token')
+        return token
+    
+if __name__ == '__main__':
+    # pass
+    project_name = "user"
+    login_config = {
+  "url": "https://xxxxxx.com/login",
+  "method": "POST",
+  "headers": {
+    "Content-Type": "application/json"
+  },
+  "data": {
+    "account": "13500000000",
+    "password": "xxxxxx",
+    "authorizationClientType": 2,
+    "loginType": 3,
+    "loginVersion": 2
+  }
+}
+    
+    print(Auth().login(project_name, login_config))