#!/usr/bin/env python3
"""
TSNet Phase 2 - Test Simple Verificación
Test simple para verificar que las correcciones TSNet funcionan
"""

import requests
import json
import time


def test_simple_tsnet():
    """Test simple de funcionalidad TSNet"""
    try:
        # Test básico de conectividad
        response = requests.post(
            "http://localhost:5006",
            json={
                "jsonrpc": "2.0",
                "method": "execute_python",
                "params": {
                    "code": """
# Test simple de TSNet Phase 2
print("=== TSNet Phase 2 Simple Test ===")

try:
    # Test 1: Verificar que el manager existe
    if hasattr(app, 'tsnetSimulationManager'):
        print("✅ TSNetSimulationManager exists")
    else:
        print("❌ TSNetSimulationManager not found")
        
    # Test 2: Test método básico
    app.tsnetSimulationManager.ResetAllCalculatedValues()
    print("✅ ResetAllCalculatedValues works")
    
    # Test 3: Crear objetos hidráulicos y verificar que no hay NullReference
    from CtrEditor.ObjetosSim import osHydTank
    
    tank = osHydTank()
    tank.CheckData()  # Esto debe inicializar el ID
    print(f"✅ Tank created with ID: {tank.Id.Value}")
    
    # Test 4: Registrar objeto sin crash
    app.tsnetSimulationManager.RegisterHydraulicObject(tank)
    print("✅ Object registered without NullReference")
    
    # Test 5: Validar configuraciones
    errors = app.tsnetSimulationManager.ValidateAllConfigurations()
    print(f"✅ Configuration validation completed: {len(errors)} errors")
    
    # Test 6: Test método principal sin crash
    app.TestTSNetIntegrationSync()
    print("✅ TestTSNetIntegrationSync completed")
    
    print("\\n🎉 ALL TESTS PASSED - TSNet Phase 2 corrections working!")
    result = "SUCCESS"
    
except Exception as e:
    print(f"❌ Error: {str(e)}")
    import traceback
    print(f"Stack trace: {traceback.format_exc()}")
    result = f"FAILED: {str(e)}"

print(f"\\nFinal result: {result}")
"""
                },
                "id": 1,
            },
            timeout=30,
        )

        if response.status_code == 200:
            result = response.json()
            if "result" in result:
                print("📋 Test Output:")
                print("-" * 50)
                print(result["result"])
                print("-" * 50)

                # Verificar si el test fue exitoso
                if "SUCCESS" in str(result["result"]):
                    print("\n🎉 TSNet Phase 2 is working correctly!")
                    print("✅ NullReference issues resolved")
                    print("✅ Object registration working")
                    print("✅ Configuration validation working")
                    return True
                else:
                    print("\n❌ Test completed but with errors")
                    return False
            else:
                print(f"❌ MCP Error: {result.get('error', 'Unknown error')}")
                return False
        else:
            print(f"❌ HTTP Error: {response.status_code}")
            return False

    except Exception as e:
        print(f"❌ Connection Error: {e}")
        return False


def main():
    print("🧪 TSNet Phase 2 - Simple Verification Test")
    print("=" * 50)
    print(f"Testing at: {time.strftime('%Y-%m-%d %H:%M:%S')}")
    print()

    # Verificar que CtrEditor esté ejecutándose
    print("🔍 Checking CtrEditor status...")
    try:
        status_response = requests.post(
            "http://localhost:5006",
            json={"jsonrpc": "2.0", "method": "get_ctreditor_status", "id": 0},
            timeout=5,
        )

        if status_response.status_code == 200:
            status = status_response.json()
            if (
                "result" in status
                and status["result"].get("connection_status") == "available"
            ):
                print("✅ CtrEditor is running and MCP server is responding")
            else:
                print("⚠️ CtrEditor is running but MCP may have issues")
        else:
            print("❌ Cannot reach CtrEditor MCP server")
            return False

    except Exception as e:
        print(f"❌ Cannot connect to CtrEditor: {e}")
        return False

    print("\n🚀 Running TSNet verification test...")
    success = test_simple_tsnet()

    print("\n" + "=" * 50)
    if success:
        print("🎉 VERIFICATION SUCCESSFUL!")
        print("   TSNet Phase 2 corrections are working properly")
        print("   The system is ready for production use")
    else:
        print("❌ VERIFICATION FAILED!")
        print("   Please check the output above for details")

    return success


if __name__ == "__main__":
    success = main()
    exit(0 if success else 1)