#!/usr/bin/env python3
"""
DMS合规性测试工具 - API服务器功能测试脚本
测试Docker化的API服务器是否正常工作
"""

import requests
import json
import time
import sys
from pathlib import Path

def test_health_check(api_server):
    """测试健康检查端点"""
    print(f"[测试] 健康检查端点...")
    try:
        response = requests.get(f"{api_server}/", timeout=10)
        if response.status_code == 200:
            result = response.json()
            print(f"[成功] 健康检查通过: {result}")
            return True
        else:
            print(f"[失败] 健康检查失败: {response.status_code}")
            return False
    except Exception as e:
        print(f"[错误] 健康检查异常: {e}")
        return False

def test_with_mock_api(api_server):
    """使用模拟API进行测试"""
    print(f"[测试] 使用模拟API配置进行测试...")
    
    # 使用一个简单的测试配置，不依赖外部API
    test_config = {
        "base-url": "https://httpbin.org/",  # 使用httpbin作为测试目标
        "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": False,  # 暂时不生成PDF
        "strictness-level": "CRITICAL",
        "ignore-ssl": True
    }
    
    print(f"[信息] 测试配置:")
    print(json.dumps(test_config, indent=2, ensure_ascii=False))
    
    try:
        print(f"[信息] 发送测试请求...")
        start_time = time.time()
        
        response = requests.post(
            f"{api_server}/run",
            json=test_config,
            headers={"Content-Type": "application/json"},
            timeout=60  # 1分钟超时
        )
        
        end_time = time.time()
        duration = end_time - start_time
        
        print(f"[信息] 请求完成，耗时: {duration:.2f}秒")
        print(f"[信息] HTTP状态码: {response.status_code}")
        
        if response.status_code == 200:
            print(f"[成功] API服务器响应正常")
            try:
                result = response.json()
                print(f"[信息] 响应状态: {result.get('status', '未知')}")
                if 'message' in result:
                    print(f"[信息] 响应消息: {result['message']}")
                return True
            except:
                print(f"[信息] 响应内容（非JSON）: {response.text[:200]}...")
                return True
        else:
            print(f"[失败] API服务器响应异常")
            print(f"[错误] 响应内容: {response.text[:200]}...")
            return False
            
    except requests.exceptions.Timeout:
        print(f"[错误] 请求超时（1分钟）")
        return False
    except Exception as e:
        print(f"[错误] 请求异常: {e}")
        return False

def test_history_viewer(history_server):
    """测试历史查看器"""
    print(f"[测试] 历史查看器端点...")
    try:
        response = requests.get(f"{history_server}/", timeout=10)
        if response.status_code == 200:
            print(f"[成功] 历史查看器响应正常")
            return True
        else:
            print(f"[失败] 历史查看器响应异常: {response.status_code}")
            return False
    except Exception as e:
        print(f"[错误] 历史查看器异常: {e}")
        return False

def main():
    print("=== DMS合规性测试工具 - API服务器功能测试 ===")
    
    api_server = "http://localhost:5050"
    history_server = "http://localhost:5051"
    
    print(f"[信息] API服务器: {api_server}")
    print(f"[信息] 历史查看器: {history_server}")
    
    # 测试结果
    results = []
    
    # 1. 测试API服务器健康检查
    results.append(("API服务器健康检查", test_health_check(api_server)))
    
    # 2. 测试历史查看器
    results.append(("历史查看器", test_history_viewer(history_server)))
    
    # 3. 测试API服务器功能（使用模拟API）
    results.append(("API服务器功能测试", test_with_mock_api(api_server)))
    
    # 显示测试结果摘要
    print(f"\n=== 测试结果摘要 ===")
    passed = 0
    total = len(results)
    
    for test_name, result in results:
        status = "✅ 通过" if result else "❌ 失败"
        print(f"{status} {test_name}")
        if result:
            passed += 1
    
    print(f"\n总计: {passed}/{total} 测试通过")
    
    if passed == total:
        print(f"[成功] 所有测试通过！Docker化的API服务器工作正常！")
        return 0
    else:
        print(f"[警告] 部分测试失败，请检查服务状态")
        return 1

if __name__ == "__main__":
    sys.exit(main())