"""
执行结果展示和WebSocket功能测试脚本
"""
import asyncio
import sys
import os
import json
import time

# 添加项目路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.core.database import SessionLocal
from app.models.workflow import Workflow
from app.models.execution import Execution
from app.models.user import User
from app.services.workflow_engine import WorkflowEngine
import uuid


def test_execution_api():
    """测试执行API"""
    print("=" * 60)
    print("测试1: 执行API功能")
    print("=" * 60)
    
    db = SessionLocal()
    
    try:
        # 获取或创建测试用户
        user = db.query(User).first()
        if not user:
            print("❌ 没有找到用户，请先创建用户")
            return False
        
        # 获取或创建测试工作流
        workflow = db.query(Workflow).filter(Workflow.user_id == user.id).first()
        if not workflow:
            print("❌ 没有找到工作流，请先创建工作流")
            return False
        
        print(f"✅ 使用工作流: {workflow.id} ({workflow.name})")
        
        # 创建执行记录
        execution = Execution(
            workflow_id=str(workflow.id),
            input_data={"input": "测试输入数据"},
            status="pending"
        )
        db.add(execution)
        db.commit()
        db.refresh(execution)
        
        print(f"✅ 创建执行记录: {execution.id}")
        print(f"   状态: {execution.status}")
        print(f"   工作流ID: {execution.workflow_id}")
        print(f"   输入数据: {execution.input_data}")
        
        # 测试获取执行列表
        executions = db.query(Execution).filter(
            Execution.workflow_id == workflow.id
        ).limit(5).all()
        
        print(f"\n✅ 获取执行列表: 共 {len(executions)} 条记录")
        for i, exec in enumerate(executions[:3], 1):
            print(f"   {i}. ID: {exec.id[:8]}..., 状态: {exec.status}, 创建时间: {exec.created_at}")
        
        # 测试获取执行详情
        execution_detail = db.query(Execution).filter(
            Execution.id == execution.id
        ).first()
        
        if execution_detail:
            print(f"\n✅ 获取执行详情成功")
            print(f"   ID: {execution_detail.id}")
            print(f"   状态: {execution_detail.status}")
            print(f"   输入数据: {execution_detail.input_data}")
            print(f"   输出数据: {execution_detail.output_data or '暂无'}")
        
        return True
        
    except Exception as e:
        print(f"❌ 测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        db.close()


async def test_workflow_execution():
    """测试工作流执行"""
    print("\n" + "=" * 60)
    print("测试2: 工作流执行")
    print("=" * 60)
    
    db = SessionLocal()
    
    try:
        # 获取测试工作流
        workflow = db.query(Workflow).first()
        if not workflow:
            print("❌ 没有找到工作流")
            return False
        
        print(f"✅ 使用工作流: {workflow.id}")
        print(f"   节点数: {len(workflow.nodes)}")
        print(f"   边数: {len(workflow.edges)}")
        
        # 创建执行记录
        execution = Execution(
            workflow_id=str(workflow.id),
            input_data={"input": "测试执行"},
            status="pending"
        )
        db.add(execution)
        db.commit()
        db.refresh(execution)
        
        print(f"✅ 创建执行记录: {execution.id}")
        
        # 执行工作流
        workflow_data = {
            'nodes': workflow.nodes,
            'edges': workflow.edges
        }
        
        print("\n🔄 开始执行工作流...")
        start_time = time.time()
        
        engine = WorkflowEngine(str(workflow.id), workflow_data)
        result = await engine.execute(execution.input_data)
        
        execution_time = int((time.time() - start_time) * 1000)
        
        # 更新执行记录
        execution.status = "completed"
        execution.output_data = result
        execution.execution_time = execution_time
        db.commit()
        db.refresh(execution)
        
        print(f"✅ 工作流执行完成")
        print(f"   执行时间: {execution_time}ms")
        print(f"   状态: {execution.status}")
        print(f"   结果: {json.dumps(result, ensure_ascii=False, indent=2)[:200]}...")
        
        return True
        
    except Exception as e:
        print(f"❌ 执行失败: {str(e)}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        db.close()


def test_execution_status_api():
    """测试执行状态API"""
    print("\n" + "=" * 60)
    print("测试3: 执行状态API")
    print("=" * 60)
    
    db = SessionLocal()
    
    try:
        # 获取执行记录
        execution = db.query(Execution).order_by(Execution.created_at.desc()).first()
        if not execution:
            print("❌ 没有找到执行记录")
            return False
        
        print(f"✅ 测试执行记录: {execution.id}")
        print(f"   当前状态: {execution.status}")
        print(f"   执行时间: {execution.execution_time or 'N/A'}ms")
        print(f"   输出数据: {'有' if execution.output_data else '无'}")
        print(f"   错误信息: {execution.error_message or '无'}")
        
        # 测试不同状态
        statuses = ['pending', 'running', 'completed', 'failed']
        print(f"\n✅ 支持的状态: {', '.join(statuses)}")
        
        return True
        
    except Exception as e:
        print(f"❌ 测试失败: {str(e)}")
        return False
    finally:
        db.close()


def test_execution_response_format():
    """测试执行响应格式"""
    print("\n" + "=" * 60)
    print("测试4: 执行响应格式")
    print("=" * 60)
    
    db = SessionLocal()
    
    try:
        execution = db.query(Execution).order_by(Execution.created_at.desc()).first()
        if not execution:
            print("❌ 没有找到执行记录")
            return False
        
        # 检查响应字段
        required_fields = [
            'id', 'workflow_id', 'status', 'input_data', 
            'output_data', 'created_at'
        ]
        
        print("✅ 检查响应字段:")
        for field in required_fields:
            has_field = hasattr(execution, field)
            value = getattr(execution, field, None)
            status = "✅" if has_field else "❌"
            print(f"   {status} {field}: {type(value).__name__}")
        
        # 检查created_at类型
        if execution.created_at:
            print(f"\n✅ created_at类型: {type(execution.created_at).__name__}")
            print(f"   created_at值: {execution.created_at}")
        
        return True
        
    except Exception as e:
        print(f"❌ 测试失败: {str(e)}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        db.close()


async def main():
    """主测试函数"""
    print("\n" + "🚀 开始执行结果展示功能测试" + "\n")
    
    results = []
    
    # 运行测试
    results.append(test_execution_api())
    results.append(await test_workflow_execution())
    results.append(test_execution_status_api())
    results.append(test_execution_response_format())
    
    # 汇总结果
    print("\n" + "=" * 60)
    print("测试结果汇总")
    print("=" * 60)
    passed = sum(results)
    total = len(results)
    print(f"通过: {passed}/{total}")
    print(f"失败: {total - passed}/{total}")
    
    if passed == total:
        print("\n✅ 所有测试通过！执行结果展示功能正常！")
    else:
        print(f"\n⚠️  有 {total - passed} 个测试失败")


if __name__ == "__main__":
    asyncio.run(main())