#!/usr/bin/env python3
"""
Quick test demo showing the working test framework functionality
"""
import asyncio
import sys
import os

# Add project root to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from tests.test_client import test_clients


async def demo_test():
    """Demo of the testing framework functionality"""
    print("🎮 City Builder - Test Framework Demo")
    print("=" * 50)
    
    # Test 1: Single client connection
    print("\n✅ Test 1: Single client connection")
    async with test_clients("demo_player") as [client]:
        print(f"   Connected: {client.connected}")
        print(f"   Player: {client.player_data['nickname']}")
        print(f"   Money: ${client.get_money():,}")
        print(f"   Population: {client.get_population()}")
    
    # Test 2: Building placement
    print("\n✅ Test 2: Building placement")
    async with test_clients("builder") as [client]:
        # Place a road
        await client.place_building("road", 50, 50)
        message = await client.receive_message(timeout=2.0)
        if message and message["type"] == "building_placed":
            print(f"   Successfully placed: {message['building']['type']}")
            print(f"   At coordinates: ({message['building']['x']}, {message['building']['y']})")
        else:
            print(f"   Error or unexpected message: {message}")
    
    # Test 3: Chat functionality  
    print("\n✅ Test 3: Chat functionality")
    async with test_clients("sender", "receiver") as [sender, receiver]:
        # Clear any initial messages
        receiver.clear_messages()
        
        # Send chat message
        test_message = "Hello from test framework!"
        await sender.send_chat(test_message)
        
        # Receive chat message
        message = await receiver.receive_message(timeout=2.0)
        if message and message["type"] == "chat":
            print(f"   Message sent: '{test_message}'")
            print(f"   Message received: '{message['message']}'")
            print(f"   From: {message['nickname']}")
        else:
            print(f"   Error or unexpected message: {message}")
    
    # Test 4: Multiple client interaction
    print("\n✅ Test 4: Multiple client interaction")
    async with test_clients("alice", "bob", "charlie") as [alice, bob, charlie]:
        print(f"   Connected clients: {len([alice, bob, charlie])}")
        print(f"   Alice: ${alice.get_money():,}")
        print(f"   Bob: ${bob.get_money():,}")
        print(f"   Charlie: ${charlie.get_money():,}")
        
        # All clients send cursor movements simultaneously
        await asyncio.gather(
            alice.send_cursor_move(10, 20),
            bob.send_cursor_move(30, 40),  
            charlie.send_cursor_move(50, 60)
        )
        print("   All cursor movements sent successfully")
    
    print("\n🎉 Test framework demo completed!")
    print("\n📊 Summary:")
    print("   ✅ WebSocket connections work")
    print("   ✅ Building placement works")
    print("   ✅ Chat system works")
    print("   ✅ Multiple clients work")
    print("   ✅ Async operations work")


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