233 lines
8.4 KiB
Python
233 lines
8.4 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
DMS合规性测试工具 - 完整测试脚本（包含mock服务）
|
||
1. 启动mock_dms_server.py在5001端口
|
||
2. 调用Docker化的API服务器进行测试
|
||
3. 清理mock服务
|
||
"""
|
||
|
||
import requests
|
||
import json
|
||
import time
|
||
import sys
|
||
import subprocess
|
||
import signal
|
||
import os
|
||
from pathlib import Path
|
||
|
||
class MockServerManager:
|
||
def __init__(self, mock_script="mock_dms_server.py", port=5001):
|
||
self.mock_script = mock_script
|
||
self.port = port
|
||
self.process = None
|
||
|
||
def start(self):
|
||
"""启动mock服务器"""
|
||
print(f"[信息] 启动mock服务器: {self.mock_script} (端口 {self.port})")
|
||
|
||
if not Path(self.mock_script).exists():
|
||
print(f"[错误] Mock脚本不存在: {self.mock_script}")
|
||
return False
|
||
|
||
try:
|
||
# 启动mock服务器
|
||
self.process = subprocess.Popen(
|
||
[sys.executable, self.mock_script],
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.PIPE,
|
||
text=True
|
||
)
|
||
|
||
# 等待服务启动
|
||
print(f"[信息] 等待mock服务器启动...")
|
||
for i in range(10): # 最多等待10秒
|
||
try:
|
||
response = requests.get(f"http://127.0.0.1:{self.port}/", timeout=2)
|
||
if response.status_code in [200, 404]: # 404也算正常，说明服务在运行
|
||
print(f"[成功] Mock服务器启动成功 (尝试 {i+1}/10)")
|
||
return True
|
||
except:
|
||
pass
|
||
time.sleep(1)
|
||
|
||
print(f"[错误] Mock服务器启动超时")
|
||
return False
|
||
|
||
except Exception as e:
|
||
print(f"[错误] 启动mock服务器失败: {e}")
|
||
return False
|
||
|
||
def stop(self):
|
||
"""停止mock服务器"""
|
||
if self.process:
|
||
print(f"[信息] 停止mock服务器...")
|
||
try:
|
||
self.process.terminate()
|
||
self.process.wait(timeout=5)
|
||
print(f"[成功] Mock服务器已停止")
|
||
except subprocess.TimeoutExpired:
|
||
print(f"[警告] Mock服务器强制终止")
|
||
self.process.kill()
|
||
self.process.wait()
|
||
except Exception as e:
|
||
print(f"[错误] 停止mock服务器失败: {e}")
|
||
|
||
def test_api_server(api_server, target_api, output_log="log_dms_docker.txt"):
|
||
"""测试API服务器"""
|
||
print(f"\n[测试] API服务器功能测试...")
|
||
|
||
# 测试配置（对应原始命令行参数）
|
||
test_config = {
|
||
"base-url": target_api,
|
||
"dms": "./assets/doc/dms/domain.json",
|
||
"stages-dir": "./custom_stages",
|
||
"custom-test-cases-dir": "./custom_testcases",
|
||
"verbose": True,
|
||
"output": "./test_reports/",
|
||
"format": "json",
|
||
"generate-pdf": True,
|
||
"strictness-level": "CRITICAL",
|
||
"ignore-ssl": True
|
||
}
|
||
|
||
print(f"[信息] 测试配置:")
|
||
print(json.dumps(test_config, indent=2, ensure_ascii=False))
|
||
|
||
try:
|
||
print(f"[信息] 发送测试请求到API服务器...")
|
||
start_time = time.time()
|
||
|
||
response = requests.post(
|
||
f"{api_server}/run",
|
||
json=test_config,
|
||
headers={"Content-Type": "application/json"},
|
||
timeout=120 # 2分钟超时
|
||
)
|
||
|
||
end_time = time.time()
|
||
duration = end_time - start_time
|
||
|
||
# 保存结果到日志文件
|
||
with open(output_log, 'w', encoding='utf-8') as f:
|
||
f.write(f"=== DMS合规性测试结果 ===\n")
|
||
f.write(f"执行时间: {duration:.2f}秒\n")
|
||
f.write(f"HTTP状态码: {response.status_code}\n")
|
||
f.write(f"响应头: {dict(response.headers)}\n\n")
|
||
|
||
if response.status_code == 200:
|
||
f.write("=== 测试成功 ===\n")
|
||
try:
|
||
result_json = response.json()
|
||
f.write(json.dumps(result_json, indent=2, ensure_ascii=False))
|
||
except:
|
||
f.write("响应内容（非JSON格式）:\n")
|
||
f.write(response.text)
|
||
else:
|
||
f.write("=== 测试失败 ===\n")
|
||
f.write(f"错误信息: {response.text}\n")
|
||
|
||
# 显示结果
|
||
print(f"[信息] 请求完成，耗时: {duration:.2f}秒")
|
||
print(f"[信息] HTTP状态码: {response.status_code}")
|
||
|
||
if response.status_code == 200:
|
||
print(f"[成功] 测试执行成功")
|
||
try:
|
||
result = response.json()
|
||
print(f"[信息] 测试状态: {result.get('status', '未知')}")
|
||
if 'summary' in result:
|
||
summary = result['summary']
|
||
print(f"[信息] 测试摘要:")
|
||
print(f" - 总测试数: {summary.get('total_tests', 0)}")
|
||
print(f" - 通过数: {summary.get('passed_tests', 0)}")
|
||
print(f" - 失败数: {summary.get('failed_tests', 0)}")
|
||
print(f" - 跳过数: {summary.get('skipped_tests', 0)}")
|
||
|
||
if 'output_files' in result:
|
||
print(f"[信息] 生成的文件:")
|
||
for file_info in result['output_files']:
|
||
print(f" - {file_info}")
|
||
|
||
return True
|
||
except:
|
||
print(f"[信息] 响应内容已保存到: {output_log}")
|
||
return True
|
||
else:
|
||
print(f"[失败] 测试执行失败")
|
||
print(f"[错误] 错误信息: {response.text[:200]}...")
|
||
return False
|
||
|
||
except requests.exceptions.Timeout:
|
||
print(f"[错误] 请求超时（2分钟）")
|
||
return False
|
||
except Exception as e:
|
||
print(f"[错误] 请求异常: {e}")
|
||
return False
|
||
|
||
def main():
|
||
print("=== DMS合规性测试工具 - 完整功能测试 ===")
|
||
|
||
api_server = "http://localhost:5050"
|
||
history_server = "http://localhost:5051"
|
||
target_api = "http://host.docker.internal:5001/" # 使用Docker内部主机访问
|
||
|
||
# 检查mock脚本是否存在
|
||
mock_script = "mock_dms_server.py"
|
||
if not Path(mock_script).exists():
|
||
print(f"[警告] Mock脚本不存在: {mock_script}")
|
||
print(f"[信息] 请手动启动mock服务器: python mock_dms_server.py")
|
||
mock_manager = None
|
||
else:
|
||
mock_manager = MockServerManager(mock_script, 5001)
|
||
|
||
try:
|
||
# 1. 启动mock服务器（如果存在）
|
||
if mock_manager:
|
||
if not mock_manager.start():
|
||
print(f"[错误] Mock服务器启动失败")
|
||
print(f"[建议] 请手动启动: python {mock_script}")
|
||
return 1
|
||
|
||
# 2. 测试API服务器健康检查
|
||
print(f"\n[测试] API服务器健康检查...")
|
||
health_response = requests.get(f"{api_server}/", timeout=10)
|
||
if health_response.status_code == 200:
|
||
print(f"[成功] API服务器运行正常")
|
||
else:
|
||
print(f"[错误] API服务器不可用: {health_response.status_code}")
|
||
return 1
|
||
|
||
# 3. 测试历史查看器
|
||
print(f"\n[测试] 历史查看器...")
|
||
history_response = requests.get(f"{history_server}/", timeout=10)
|
||
if history_response.status_code == 200:
|
||
print(f"[成功] 历史查看器运行正常")
|
||
else:
|
||
print(f"[警告] 历史查看器异常: {history_response.status_code}")
|
||
|
||
# 4. 执行完整的API测试
|
||
success = test_api_server(api_server, target_api)
|
||
|
||
if success:
|
||
print(f"\n[成功] 所有测试完成！")
|
||
print(f"[信息] 查看详细日志: cat log_dms_docker.txt")
|
||
print(f"[信息] 查看测试报告: ls -la test_reports/")
|
||
return 0
|
||
else:
|
||
print(f"\n[失败] 测试执行失败")
|
||
return 1
|
||
|
||
except KeyboardInterrupt:
|
||
print(f"\n[信息] 用户中断测试")
|
||
return 1
|
||
except Exception as e:
|
||
print(f"\n[错误] 测试异常: {e}")
|
||
return 1
|
||
finally:
|
||
# 清理mock服务器
|
||
if mock_manager:
|
||
mock_manager.stop()
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|