281 lines
10 KiB
Python
281 lines
10 KiB
Python
"""
|
||
文档列表综合测试 - 搜索过滤 + 完整数据展示
|
||
结合 test_search_filters.py 和 test_all_documents.py 的功能
|
||
"""
|
||
import sys
|
||
import io
|
||
import requests
|
||
import json
|
||
import time
|
||
from datetime import datetime, timedelta
|
||
|
||
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
|
||
|
||
# 配置
|
||
BASE_URL = "http://localhost:8073/admin/api/v3"
|
||
ADMIN_TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjo1LCJ1c2VybmFtZSI6ImFkbWluIiwidXNlcl9yb2xlIjoiYWRtaW4iLCJleHAiOjE3NjM1NDAwMTksImlhdCI6MTc2MzUxODQxOSwiaXNzdWVkX3RpbWUiOiIyMDI1LTExLTE5IDEwOjEzOjM5IiwiYXVkIjoiZG9jcmV2aWV3LWZyb250ZW5kIiwic3ViIjoiMDAwIiwibmlja19uYW1lIjoiYWRtaW4iLCJvdV9pZCI6IjAwMCIsIm91X25hbWUiOiJ0ZXN0IiwiaXNfbGVhZGVyIjp0cnVlfQ.Mocax00w0Os4n2WjogOk5kwE-KYkR4AfGTGA33sjgc0"
|
||
|
||
# ANSI颜色
|
||
GREEN = '\033[92m'
|
||
RED = '\033[91m'
|
||
YELLOW = '\033[93m'
|
||
BLUE = '\033[94m'
|
||
CYAN = '\033[96m'
|
||
MAGENTA = '\033[95m'
|
||
RESET = '\033[0m'
|
||
BOLD = '\033[1m'
|
||
|
||
headers = {
|
||
"Authorization": f"Bearer {ADMIN_TOKEN}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
def print_header(text: str):
|
||
print(f"\n{BLUE}{BOLD}{'='*100}{RESET}")
|
||
print(f"{BLUE}{BOLD}{text:^100}{RESET}")
|
||
print(f"{BLUE}{BOLD}{'='*100}{RESET}")
|
||
|
||
def print_subheader(text: str):
|
||
print(f"\n{CYAN}{'-'*100}{RESET}")
|
||
print(f"{CYAN}{text}{RESET}")
|
||
print(f"{CYAN}{'-'*100}{RESET}")
|
||
|
||
def test_scenario(title: str, params: dict, show_full_json: bool = False, show_details: bool = True):
|
||
"""
|
||
测试场景：执行搜索并展示结果
|
||
|
||
Args:
|
||
title: 测试场景标题
|
||
params: 查询参数
|
||
show_full_json: 是否显示完整JSON
|
||
show_details: 是否显示详细信息
|
||
"""
|
||
print_header(f"📋 {title}")
|
||
|
||
# 显示查询参数
|
||
print(f"\n{CYAN}🔍 查询参数:{RESET}")
|
||
for key, value in params.items():
|
||
print(f" • {key}: {value}")
|
||
|
||
# 执行请求
|
||
url = f"{BASE_URL}/versions/documents-list"
|
||
start_time = time.time()
|
||
response = requests.get(url, headers=headers, params=params)
|
||
elapsed = time.time() - start_time
|
||
|
||
# 状态码
|
||
if response.status_code == 200:
|
||
print(f"\n{GREEN}✓ 请求成功{RESET} - 状态码: {response.status_code} - 耗时: {elapsed*1000:.0f}ms")
|
||
else:
|
||
print(f"\n{RED}✗ 请求失败{RESET} - 状态码: {response.status_code}")
|
||
print(f"{RED}{response.text}{RESET}")
|
||
return
|
||
|
||
data = response.json()
|
||
documents = data.get("documents", [])
|
||
total = data.get("total", 0)
|
||
|
||
# 基本统计
|
||
print(f"\n{CYAN}📊 查询结果统计:{RESET}")
|
||
print(f" • 总文档数: {total}")
|
||
print(f" • 当前页: {data.get('page', 0)}/{data.get('total_pages', 0)}")
|
||
print(f" • 每页数量: {data.get('page_size', 0)}")
|
||
print(f" • 返回文档数: {len(documents)}")
|
||
|
||
if len(documents) == 0:
|
||
print(f"\n{YELLOW}⚠ 未找到任何文档{RESET}")
|
||
return
|
||
|
||
# 显示文档列表
|
||
print(f"\n{CYAN}📄 文档列表:{RESET}")
|
||
for idx, doc in enumerate(documents, 1):
|
||
name = doc.get('name', 'N/A')
|
||
status = doc.get('status', 'N/A')
|
||
audit = doc.get('audit_status', 'N/A')
|
||
version = doc.get('version_number', 'N/A')
|
||
total_ver = doc.get('total_versions', 'N/A')
|
||
|
||
# 审核状态颜色
|
||
audit_color = RESET
|
||
if audit == 1:
|
||
audit_color = GREEN
|
||
elif audit == 0:
|
||
audit_color = YELLOW
|
||
elif audit in [-1, -2]:
|
||
audit_color = RED
|
||
|
||
print(f" {idx}. {name}")
|
||
print(f" └─ 状态:{status} | 审核:{audit_color}{audit}{RESET} | 版本:v{version}/{total_ver}")
|
||
|
||
# 详细信息
|
||
if show_details:
|
||
print_subheader("📖 文档详细信息")
|
||
for idx, doc in enumerate(documents, 1):
|
||
print(f"\n{MAGENTA}{BOLD}▼ 文档 {idx}: {doc.get('name')}{RESET}")
|
||
|
||
# 基本信息
|
||
print(f"\n {CYAN}📄 基本信息:{RESET}")
|
||
print(f" • ID: {doc.get('id')}")
|
||
print(f" • 文件大小: {doc.get('file_size', 0):,} bytes ({doc.get('file_size', 0)/1024/1024:.2f} MB)")
|
||
print(f" • 创建时间: {doc.get('created_at')}")
|
||
print(f" • 上传时间: {doc.get('upload_time')}")
|
||
print(f" • 文档类型ID: {doc.get('type_id')}")
|
||
|
||
# 评查统计
|
||
print(f"\n {CYAN}📊 评查统计:{RESET}")
|
||
print(f" • 总评查点: {doc.get('total_evaluation_points', 0)}")
|
||
print(f" • {GREEN}通过: {doc.get('pass_count', 0)}{RESET}")
|
||
print(f" • {YELLOW}警告: {doc.get('warning_count', 0)}{RESET}")
|
||
print(f" • {RED}错误: {doc.get('error_count', 0)}{RESET}")
|
||
print(f" • {MAGENTA}需人工: {doc.get('manual_count', 0)}{RESET}")
|
||
print(f" • {CYAN}问题总数: {doc.get('issue_count', 0)}{RESET}")
|
||
|
||
# 版本信息
|
||
history = doc.get('history_versions', [])
|
||
print(f"\n {CYAN}📚 版本信息:{RESET}")
|
||
print(f" • 当前版本: v{doc.get('version_number')}")
|
||
print(f" • 总版本数: {doc.get('total_versions')}")
|
||
print(f" • 历史版本数: {len(history)}")
|
||
|
||
if len(history) > 0:
|
||
for h_idx, h_ver in enumerate(history, 1):
|
||
print(f" └─ 历史 {h_idx}: v{h_ver.get('version_number')} (ID:{h_ver.get('id')})")
|
||
print(f" ├─ 创建: {h_ver.get('created_at')}")
|
||
print(f" ├─ 评查: {GREEN}{h_ver.get('pass_count', 0)}通过{RESET} / {YELLOW}{h_ver.get('warning_count', 0)}警告{RESET} / {RED}{h_ver.get('error_count', 0)}错误{RESET}")
|
||
print(f" └─ 问题: {h_ver.get('issue_count', 0)}个")
|
||
|
||
# 问题消息
|
||
warning_msgs = doc.get('warning_messages', [])
|
||
error_msgs = doc.get('error_messages', [])
|
||
manual_msgs = doc.get('manual_messages', [])
|
||
|
||
if len(warning_msgs) > 0:
|
||
print(f"\n {YELLOW}⚠️ 警告消息 ({len(warning_msgs)}条):{RESET}")
|
||
for i, msg in enumerate(warning_msgs[:3], 1): # 只显示前3条
|
||
print(f" {i}. {msg}")
|
||
if len(warning_msgs) > 3:
|
||
print(f" ... 还有 {len(warning_msgs) - 3} 条警告")
|
||
|
||
if len(error_msgs) > 0:
|
||
print(f"\n {RED}❌ 错误消息 ({len(error_msgs)}条):{RESET}")
|
||
for i, msg in enumerate(error_msgs, 1):
|
||
print(f" {i}. {msg}")
|
||
|
||
if len(manual_msgs) > 0:
|
||
print(f"\n {MAGENTA}👤 人工确认消息 ({len(manual_msgs)}条):{RESET}")
|
||
for i, msg in enumerate(manual_msgs[:3], 1): # 只显示前3条
|
||
print(f" {i}. {msg}")
|
||
if len(manual_msgs) > 3:
|
||
print(f" ... 还有 {len(manual_msgs) - 3} 条消息")
|
||
|
||
# 显示完整JSON
|
||
if show_full_json:
|
||
print_subheader("📝 完整JSON响应")
|
||
print(json.dumps(data, indent=2, ensure_ascii=False))
|
||
|
||
print(f"\n{GREEN}{'─'*100}{RESET}")
|
||
|
||
|
||
def main():
|
||
"""主测试流程"""
|
||
print_header("🔬 文档列表API综合测试套件")
|
||
print(f"{CYAN}测试时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}{RESET}\n")
|
||
|
||
# 定义测试场景
|
||
test_scenarios = [
|
||
{
|
||
"title": "测试1: 获取所有文档",
|
||
"params": {"page": 1, "page_size": 10},
|
||
"show_details": True,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试2: 文档名称搜索（包含'第'）",
|
||
"params": {"page": 1, "page_size": 10, "name": "第"},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试3: 文件状态过滤（已完成）",
|
||
"params": {"page": 1, "page_size": 10, "status": "Processed"},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试4: 审核状态过滤（待审核=0）",
|
||
"params": {"page": 1, "page_size": 10, "audit_status": 0},
|
||
"show_details": True,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试5: 审核状态过滤（通过=1）",
|
||
"params": {"page": 1, "page_size": 10, "audit_status": 1},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试6: 时间范围（今天）",
|
||
"params": {
|
||
"page": 1,
|
||
"page_size": 10,
|
||
"start_time": datetime.now().strftime("%Y-%m-%d"),
|
||
"end_time": datetime.now().strftime("%Y-%m-%d")
|
||
},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试7: 时间范围（最近7天）",
|
||
"params": {
|
||
"page": 1,
|
||
"page_size": 10,
|
||
"start_time": (datetime.now() - timedelta(days=7)).strftime("%Y-%m-%d"),
|
||
"end_time": datetime.now().strftime("%Y-%m-%d")
|
||
},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试8: 文档类型过滤（type_id=3）",
|
||
"params": {"page": 1, "page_size": 10, "type_id": 3},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试9: 组合过滤（名称+状态+审核）",
|
||
"params": {
|
||
"page": 1,
|
||
"page_size": 10,
|
||
"name": "pdf",
|
||
"status": "Processed",
|
||
"audit_status": 2
|
||
},
|
||
"show_details": False,
|
||
"show_full_json": False
|
||
},
|
||
{
|
||
"title": "测试10: 完整JSON输出示例",
|
||
"params": {"page": 1, "page_size": 2},
|
||
"show_details": False,
|
||
"show_full_json": True
|
||
}
|
||
]
|
||
|
||
# 执行所有测试
|
||
for scenario in test_scenarios:
|
||
test_scenario(
|
||
scenario["title"],
|
||
scenario["params"],
|
||
scenario.get("show_full_json", False),
|
||
scenario.get("show_details", True)
|
||
)
|
||
print() # 空行分隔
|
||
|
||
# 汇总
|
||
print_header("✅ 所有测试完成")
|
||
print(f"{GREEN}共执行 {len(test_scenarios)} 个测试场景{RESET}\n")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|