test_connection.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
AR.Drone 2.0 Connection Test

This script tests the AR.Drone library connection capabilities:
- Mock mode (always works)
- Live mode (requires actual drone)
- Port conflict detection and resolution
"""

import sys
import time
from libardrone import ARDrone2


def test_mock_mode():
    """Test mock mode functionality."""
    print("=" * 50)
    print("Testing Mock Mode")
    print("=" * 50)
    
    try:
        with ARDrone2(mock=True, hd_video=True) as drone:
            print("✓ Mock drone created and connected")
            
            # Test basic functionality
            time.sleep(0.5)
            
            # Test flight controls
            drone.takeoff()
            print("✓ Takeoff command sent")
            
            drone.move(roll=0.1, pitch=0.1, yaw=0.1, gaz=0.1)
            print("✓ Movement command sent")
            
            drone.hover()
            print("✓ Hover command sent")
            
            drone.land()
            print("✓ Land command sent")
            
            # Test data access
            navdata = drone.get_navdata()
            print(f"✓ Navigation data: {navdata is not None}")
            
            image = drone.get_image()
            print(f"✓ Image data: {image is not None}")
            if image is not None:
                print(f"  Image shape: {image.shape}")
            
            battery = drone.get_battery()
            print(f"✓ Battery: {battery}%")
            
            altitude = drone.get_altitude()
            print(f"✓ Altitude: {altitude}mm")
            
            print("✓ Mock mode test completed successfully")
            return True
            
    except Exception as e:
        print(f"✗ Mock mode test failed: {e}")
        return False


def test_live_mode():
    """Test live mode functionality."""
    print("\n" + "=" * 50)
    print("Testing Live Mode (requires actual drone)")
    print("=" * 50)
    
    try:
        drone = ARDrone2(mock=False, hd_video=True)
        print("✓ Live drone object created")
        
        # Try to connect
        print("Attempting to connect to AR.Drone at 192.168.1.1...")
        drone.connect()
        
        print("✓ Connected to AR.Drone successfully!")
        print("✓ Live mode test completed successfully")
        
        # Test basic status
        print(f"✓ Video ready: {drone.is_video_ready()}")
        print(f"✓ Flying: {drone.is_flying()}")
        print(f"✓ Emergency: {drone.is_emergency()}")
        
        drone.disconnect()
        return True
        
    except Exception as e:
        print(f"✗ Live mode test failed: {e}")
        print("  This is expected if no AR.Drone is connected")
        return False


def test_port_cleanup():
    """Test port cleanup functionality."""
    print("\n" + "=" * 50)
    print("Testing Port Management")
    print("=" * 50)
    
    import subprocess
    
    try:
        # Check for existing processes
        result = subprocess.run(['python3', 'cleanup_ports.py'], 
                               capture_output=True, text=True, timeout=10)
        
        if result.returncode == 0:
            print("✓ Port cleanup utility works")
            if "No processes found" in result.stdout:
                print("✓ No port conflicts detected")
            else:
                print("✓ Port cleanup completed")
            return True
        else:
            print(f"✗ Port cleanup failed: {result.stderr}")
            return False
            
    except Exception as e:
        print(f"✗ Port cleanup test failed: {e}")
        return False


def main():
    """Run all tests."""
    print("AR.Drone 2.0 Library Test Suite")
    print("================================")
    print()
    
    results = []
    
    # Test mock mode
    results.append(("Mock Mode", test_mock_mode()))
    
    # Test port management
    results.append(("Port Management", test_port_cleanup()))
    
    # Test live mode (will likely fail without hardware)
    results.append(("Live Mode", test_live_mode()))
    
    # Summary
    print("\n" + "=" * 50)
    print("Test Results Summary")
    print("=" * 50)
    
    passed = 0
    for test_name, result in results:
        status = "PASS" if result else "FAIL"
        print(f"{test_name:<20}: {status}")
        if result:
            passed += 1
    
    print(f"\nPassed: {passed}/{len(results)} tests")
    
    if passed >= 2:  # Mock mode and port management should always pass
        print("\n✓ AR.Drone 2.0 library is working correctly!")
        print("  Live mode failure is expected without hardware.")
    else:
        print("\n✗ Some tests failed. Check the output above.")
        return 1
    
    return 0


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