#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 陪诊系统后台API接口测试脚本 支持更详细的测试和结果输出 """ import requests import json import sys from datetime import datetime from typing import Dict, Optional # 服务器地址配置 # 默认使用公网地址，如果无法访问会自动尝试本地地址 SERVER_URL = "http://101.43.95.130:8039" LOCAL_URL = "http://localhost:8039" # 测试结果统计 class TestResult: def __init__(self): self.total = 0 self.passed = 0 self.failed = 0 self.results = [] # 颜色输出 class Colors: GREEN = '\033[0;32m' RED = '\033[0;31m' YELLOW = '\033[1;33m' BLUE = '\033[0;34m' NC = '\033[0m' # No Color def print_colored(text, color=Colors.NC): """打印带颜色的文本""" print(f"{color}{text}{Colors.NC}") def test_api(name: str, method: str, url: str, data: Optional[Dict] = None, headers: Optional[Dict] = None, expected_status: list = [200, 401, 403]) -> tuple: """ 测试API接口 Args: name: 接口名称 method: 请求方法 (GET/POST) url: 接口URL data: 请求数据（POST用） headers: 请求头 expected_status: 期望的HTTP状态码列表 Returns: (success: bool, status_code: int, response: dict) """ try: if headers is None: headers = {"Content-Type": "application/json"} if method.upper() == "GET": response = requests.get(url, headers=headers, timeout=10) else: response = requests.post(url, json=data, headers=headers, timeout=10) status_code = response.status_code success = status_code in expected_status # 尝试解析JSON响应 try: response_data = response.json() except: response_data = {"raw": response.text[:200]} return success, status_code, response_data except requests.exceptions.ConnectionError: return False, 0, {"error": "连接失败，无法访问服务器"} except requests.exceptions.Timeout: return False, 0, {"error": "请求超时"} except Exception as e: return False, 0, {"error": str(e)} def check_server_available(url): """检查服务器是否可用""" try: response = requests.get(f"{url}/", timeout=3) return True, response.status_code except: return False, 0 def run_tests(): """运行所有测试""" result = TestResult() # 自动检测可用的服务器地址 global SERVER_URL print("正在检测服务器地址...") server_available, status = check_server_available(SERVER_URL) if not server_available: print_colored(f"公网地址 {SERVER_URL} 无法访问，尝试本地地址...", Colors.YELLOW) local_available, _ = check_server_available(LOCAL_URL) if local_available: SERVER_URL = LOCAL_URL print_colored(f"使用本地地址: {SERVER_URL}", Colors.GREEN) else: print_colored("本地地址也无法访问，请检查服务是否运行", Colors.RED) return 1 print("=" * 60) print_colored("陪诊系统后台API接口测试", Colors.BLUE) print("=" * 60) print(f"服务器地址: {SERVER_URL}") print(f"开始时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print() # 测试用例列表 test_cases = [ # 基础接口（无需认证） { "name": "首页接口", "method": "GET", "url": f"{SERVER_URL}/", "data": None, "headers": None }, { "name": "验证码接口", "method": "GET", "url": f"{SERVER_URL}/captchaImage", "data": None, "headers": None }, # 登录相关接口 { "name": "微信登录接口", "method": "POST", "url": f"{SERVER_URL}/weixinLogin", "data": { "wxcode": "test_code", "encryptedData": "test_data", "iv": "test_iv" }, "headers": {"Content-Type": "application/x-www-form-urlencoded"} }, { "name": "获取用户Token接口", "method": "GET", "url": f"{SERVER_URL}/getUserToken", "data": {"phonenumber": "13800138000"}, "headers": None }, # 系统配置接口（需要认证） { "name": "价格配置接口", "method": "GET", "url": f"{SERVER_URL}/system/config/configKey/price", "data": None, "headers": {"Authorization": "Bearer test_token"} }, { "name": "价格类型配置接口", "method": "GET", "url": f"{SERVER_URL}/system/config/configKey/priceType", "data": None, "headers": {"Authorization": "Bearer test_token"} }, # 医院相关接口 { "name": "医院列表接口", "method": "GET", "url": f"{SERVER_URL}/system/hospital/list", "data": None, "headers": {"Authorization": "Bearer test_token"} }, # 订单相关接口（需要认证） { "name": "订单列表接口", "method": "GET", "url": f"{SERVER_URL}/system/view/list", "data": None, "headers": {"Authorization": "Bearer test_token"} }, { "name": "创建订单接口", "method": "POST", "url": f"{SERVER_URL}/system/order/insertOrderPz", "data": { "bId": 104, "cId": 1, "yuguMoney": "200", "yuliu9": "2", "yuliu10": "1", "hospitalId": 1, "startTime": "2024-01-01", "yuliu11": "13800138000" }, "headers": {"Authorization": "Bearer test_token"} }, # 用户相关接口（需要认证） { "name": "用户列表接口", "method": "GET", "url": f"{SERVER_URL}/system/user/list", "data": None, "headers": {"Authorization": "Bearer test_token"} }, { "name": "获取用户信息接口", "method": "GET", "url": f"{SERVER_URL}/getInfo", "data": None, "headers": {"Authorization": "Bearer test_token"} }, ] # 执行测试 print("-" * 60) print_colored("1. 基础接口测试（无需认证）", Colors.YELLOW) print("-" * 60) for i, test in enumerate(test_cases[:2], 1): result.total += 1 print(f"测试 {result.total}: {test['name']} ... ", end="", flush=True) success, status_code, response_data = test_api( test['name'], test['method'], test['url'], test['data'], test['headers'] ) if success: result.passed += 1 print_colored("✓ 通过", Colors.GREEN) print(f" HTTP状态码: {status_code}") if isinstance(response_data, dict) and 'code' in response_data: print(f" 响应码: {response_data.get('code')}") if 'msg' in response_data: print(f" 消息: {response_data.get('msg')}") else: result.failed += 1 print_colored("✗ 失败", Colors.RED) print(f" HTTP状态码: {status_code}") if isinstance(response_data, dict) and 'error' in response_data: print(f" 错误: {response_data.get('error')}") result.results.append({ "name": test['name'], "success": success, "status_code": status_code, "response": response_data }) print() print("-" * 60) print_colored("2. 登录相关接口测试", Colors.YELLOW) print("-" * 60) for test in test_cases[2:4]: result.total += 1 print(f"测试 {result.total}: {test['name']} ... ", end="", flush=True) success, status_code, response_data = test_api( test['name'], test['method'], test['url'], test['data'], test['headers'] ) if success: result.passed += 1 print_colored("✓ 通过", Colors.GREEN) print(f" HTTP状态码: {status_code}") if isinstance(response_data, dict): if 'code' in response_data: print(f" 响应码: {response_data.get('code')}") if 'msg' in response_data: print(f" 消息: {response_data.get('msg')}") else: result.failed += 1 print_colored("✗ 失败", Colors.RED) print(f" HTTP状态码: {status_code}") if isinstance(response_data, dict) and 'error' in response_data: print(f" 错误: {response_data.get('error')}") result.results.append({ "name": test['name'], "success": success, "status_code": status_code, "response": response_data }) print() print("-" * 60) print_colored("3. 系统配置接口测试（需要认证）", Colors.YELLOW) print("-" * 60) for test in test_cases[4:6]: result.total += 1 print(f"测试 {result.total}: {test['name']} ... ", end="", flush=True) success, status_code, response_data = test_api( test['name'], test['method'], test['url'], test['data'], test['headers'] ) if success: result.passed += 1 print_colored("✓ 通过", Colors.GREEN) print(f" HTTP状态码: {status_code}") if status_code == 401 or status_code == 403: print_colored(" (接口存在，但需要有效Token)", Colors.YELLOW) else: result.failed += 1 print_colored("✗ 失败", Colors.RED) print(f" HTTP状态码: {status_code}") result.results.append({ "name": test['name'], "success": success, "status_code": status_code, "response": response_data }) print() print("-" * 60) print_colored("4. 业务接口测试（需要认证）", Colors.YELLOW) print("-" * 60) for test in test_cases[6:]: result.total += 1 print(f"测试 {result.total}: {test['name']} ... ", end="", flush=True) success, status_code, response_data = test_api( test['name'], test['method'], test['url'], test['data'], test['headers'] ) if success: result.passed += 1 print_colored("✓ 通过", Colors.GREEN) print(f" HTTP状态码: {status_code}") if status_code == 401 or status_code == 403: print_colored(" (接口存在，但需要有效Token)", Colors.YELLOW) else: result.failed += 1 print_colored("✗ 失败", Colors.RED) print(f" HTTP状态码: {status_code}") result.results.append({ "name": test['name'], "success": success, "status_code": status_code, "response": response_data }) print() # 输出测试结果统计 print("=" * 60) print_colored("测试结果统计", Colors.BLUE) print("=" * 60) print(f"总测试数: {result.total}") print_colored(f"通过: {result.passed}", Colors.GREEN) print_colored(f"失败: {result.failed}", Colors.RED) if result.total > 0: pass_rate = (result.passed * 100) // result.total print(f"通过率: {pass_rate}%") print() print(f"结束时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print() # 输出详细结果 if result.failed > 0: print_colored("失败的接口详情:", Colors.YELLOW) for r in result.results: if not r['success']: print(f" - {r['name']}: HTTP {r['status_code']}") print() # 判断整体结果 if result.failed == 0: print_colored("所有接口测试通过！", Colors.GREEN) return 0 else: print_colored("部分接口测试失败，请检查服务器状态和网络连接", Colors.YELLOW) print_colored("提示：401/403状态码表示接口存在但需要有效认证", Colors.YELLOW) return 1 if __name__ == "__main__": try: exit_code = run_tests() sys.exit(exit_code) except KeyboardInterrupt: print("\n\n测试被用户中断") sys.exit(1) except Exception as e: print_colored(f"\n测试过程中发生错误: {str(e)}", Colors.RED) import traceback traceback.print_exc() sys.exit(1)