#!/usr/bin/env python3
"""
Script de test para validar las correcciones TSNet Phase 2
Prueba que los objetos hidráulicos se pueden crear sin errores de NullReference
"""

import requests
import json
import time
import sys

def test_mcp_connection():
    """Probar conectividad básica con MCP"""
    try:
        response = requests.post(
            'http://localhost:5006',
            json={'jsonrpc': '2.0', 'method': 'get_status', 'id': 1},
            timeout=10
        )
        print(f"✅ MCP Connection: Status {response.status_code}")
        return True
    except Exception as e:
        print(f"❌ MCP Connection failed: {e}")
        return False

def test_create_hydraulic_objects():
    """Probar creación de objetos hidráulicos"""
    objects_to_test = [
        {'type': 'osHydTank', 'name': 'Tank Test'},
        {'type': 'osHydPump', 'name': 'Pump Test'},
        {'type': 'osHydPipe', 'name': 'Pipe Test'}
    ]
    
    success_count = 0
    
    for obj in objects_to_test:
        try:
            response = requests.post(
                'http://localhost:5006',
                json={
                    'jsonrpc': '2.0', 
                    'method': 'create_object', 
                    'params': {
                        'type': obj['type'],
                        'x': 1.0 + success_count,
                        'y': 1.0
                    },
                    'id': success_count + 1
                },
                timeout=10
            )
            
            if response.status_code == 200:
                result = response.json()
                if 'error' not in result:
                    print(f"✅ {obj['name']} created successfully")
                    success_count += 1
                else:
                    print(f"❌ {obj['name']} creation failed: {result.get('error', 'Unknown error')}")
            else:
                print(f"❌ {obj['name']} HTTP error: {response.status_code}")
                
        except Exception as e:
            print(f"❌ {obj['name']} exception: {e}")
    
    return success_count

def test_tsnet_simulation():
    """Probar la simulación TSNet"""
    try:
        response = requests.post(
            'http://localhost:5006',
            json={
                'jsonrpc': '2.0',
                'method': 'execute_python',
                'params': {
                    'code': '''
try:
    # Test basic TSNet integration
    app.TestTSNetIntegrationSync()
    print("✅ TSNet Test Integration successful")
    
    # Test full TSNet simulation
    app.RunTSNetSimulationSync()
    print("✅ TSNet Full Simulation successful")
    
    result = "SUCCESS"
except Exception as e:
    print(f"❌ TSNet Error: {str(e)}")
    result = f"ERROR: {str(e)}"
'''
                },
                'id': 10
            },
            timeout=30
        )
        
        if response.status_code == 200:
            result = response.json()
            if 'error' not in result:
                print("✅ TSNet simulation test completed")
                return True
            else:
                print(f"❌ TSNet simulation failed: {result.get('error', 'Unknown error')}")
        else:
            print(f"❌ TSNet simulation HTTP error: {response.status_code}")
            
    except Exception as e:
        print(f"❌ TSNet simulation exception: {e}")
    
    return False

def main():
    print("🔧 TSNet Phase 2 Fix Validation Test")
    print("=" * 50)
    
    # Test 1: MCP Connection
    print("\n1. Testing MCP Connection...")
    if not test_mcp_connection():
        print("❌ Cannot proceed without MCP connection")
        return False
    
    # Wait for stability
    time.sleep(2)
    
    # Test 2: Object Creation
    print("\n2. Testing Hydraulic Object Creation...")
    created_objects = test_create_hydraulic_objects()
    print(f"Created {created_objects}/3 objects successfully")
    
    # Test 3: TSNet Simulation
    print("\n3. Testing TSNet Simulation...")
    simulation_success = test_tsnet_simulation()
    
    # Summary
    print("\n" + "=" * 50)
    print("🎯 TEST SUMMARY:")
    print(f"  • MCP Connection: ✅")
    print(f"  • Objects Created: {created_objects}/3")
    print(f"  • TSNet Simulation: {'✅' if simulation_success else '❌'}")
    
    if created_objects == 3 and simulation_success:
        print("\n🎉 ALL TESTS PASSED - TSNet Phase 2 fixes are working!")
        return True
    else:
        print("\n⚠️  Some tests failed - please check the output above")
        return False

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