#!/usr/bin/env python3
"""
DMS合规性测试工具 - Python API测试脚本
对应原始命令：python run_api_tests.py --base-url http://127.0.0.1:5001/ --dms ./assets/doc/dms/domain.json --stages-dir ./custom_stages --custom-test-cases-dir ./custom_testcases -v -o ./test_reports/
"""

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

def main():
    print("=== DMS合规性测试工具 - Python API测试 ===")
    
    # 配置变量
    api_server = "http://localhost:5050"
    target_api = "http://127.0.0.1:5001/"
    output_log = "log_dms_docker.txt"
    
    print(f"[信息] 目标API: {target_api}")
    print(f"[信息] API服务器: {api_server}")
    print(f"[信息] 输出日志: {output_log}")
    
    # 测试配置（对应原始命令行参数）
    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("\n[信息] 测试配置:")
    print(json.dumps(test_config, indent=2, ensure_ascii=False))
    
    try:
        # 首先检查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}")
        
        # 执行测试
        print(f"\n[信息] 开始执行测试...")
        print(f"[信息] 这可能需要几分钟时间...")
        
        start_time = time.time()
        
        response = requests.post(
            f"{api_server}/run",
            json=test_config,
            headers={"Content-Type": "application/json"},
            timeout=300  # 5分钟超时
        )
        
        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")
        
        # 显示结果
        if response.status_code == 200:
            print(f"[成功] 测试完成！耗时: {duration:.2f}秒")
            
            try:
                result = response.json()
                print(f"\n=== 测试结果摘要 ===")
                print(f"状态: {result.get('status', '未知')}")
                if 'summary' in result:
                    summary = result['summary']
                    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"\n=== 生成的文件 ===")
                    for file_info in result['output_files']:
                        print(f"- {file_info}")
                        
            except Exception as e:
                print(f"[警告] 解析JSON响应失败: {e}")
                print(f"[信息] 原始响应已保存到: {output_log}")
        else:
            print(f"[错误] 测试执行失败")
            print(f"[信息] HTTP状态码: {response.status_code}")
            print(f"[信息] 错误信息: {response.text[:200]}...")
        
        print(f"\n[信息] 详细日志已保存到: {output_log}")
        print(f"[信息] 查看测试报告: ls -la test_reports/")
        
    except requests.exceptions.Timeout:
        print(f"[错误] 请求超时（5分钟）")
        print(f"[建议] 测试可能仍在后台运行，请稍后查看test_reports/目录")
    except requests.exceptions.ConnectionError:
        print(f"[错误] 无法连接到API服务器: {api_server}")
        print(f"[建议] 请确认Docker容器正在运行: docker-compose ps")
    except Exception as e:
        print(f"[错误] 执行失败: {e}")
        return 1
    
    print(f"\n=== 管理命令 ===")
    print(f"- 查看服务状态: docker-compose ps")
    print(f"- 查看服务日志: docker-compose logs")
    print(f"- 查看测试日志: cat {output_log}")
    print(f"- 查看测试报告: ls -la test_reports/")
    
    return 0

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