Error handling improvements

server/core/error_handling.py

@@ -0,0 +1,386 @@
+"""
+Enhanced Error Handling and Recovery
+
+This module provides robust error handling, recovery mechanisms, and resilience patterns
+for the AI Voice Bot server. It includes custom exceptions, circuit breakers, retry logic,
+and graceful degradation strategies.
+"""
+
+import asyncio
+import time
+import traceback
+from enum import Enum
+from typing import Any, Callable, Dict, List, Optional, TypeVar, Generic
+from functools import wraps
+from dataclasses import dataclass
+from fastapi import WebSocket
+
+from logger import logger
+
+T = TypeVar('T')
+
+
+class ErrorSeverity(Enum):
+    """Error severity levels for categorization and handling"""
+    LOW = "low"           # Minor issues, continue operation
+    MEDIUM = "medium"     # Moderate issues, may need user notification
+    HIGH = "high"         # Serious issues, may affect functionality
+    CRITICAL = "critical" # Critical issues, immediate attention required
+
+
+class ErrorCategory(Enum):
+    """Error categories for better classification"""
+    WEBSOCKET = "websocket"
+    WEBRTC = "webrtc"
+    SESSION = "session"
+    LOBBY = "lobby"
+    AUTH = "auth"
+    PERSISTENCE = "persistence"
+    NETWORK = "network"
+    VALIDATION = "validation"
+    SYSTEM = "system"
+
+
+# Custom Exception Classes
+class VoiceBotError(Exception):
+    """Base exception for all voice bot related errors"""
+    def __init__(
+        self, 
+        message: str, 
+        category: ErrorCategory = ErrorCategory.SYSTEM,
+        severity: ErrorSeverity = ErrorSeverity.MEDIUM,
+        context: Optional[Dict[str, Any]] = None,
+        recoverable: bool = True
+    ):
+        super().__init__(message)
+        self.category = category
+        self.severity = severity
+        self.context = context or {}
+        self.recoverable = recoverable
+        self.timestamp = time.time()
+
+
+class WebSocketError(VoiceBotError):
+    """WebSocket related errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.WEBSOCKET, **kwargs)
+
+
+class WebRTCError(VoiceBotError):
+    """WebRTC signaling and connection errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.WEBRTC, **kwargs)
+
+
+class SessionError(VoiceBotError):
+    """Session management errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.SESSION, **kwargs)
+
+
+class LobbyError(VoiceBotError):
+    """Lobby management errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.LOBBY, **kwargs)
+
+
+class AuthError(VoiceBotError):
+    """Authentication and authorization errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.AUTH, **kwargs)
+
+
+class PersistenceError(VoiceBotError):
+    """Data persistence and storage errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.PERSISTENCE, **kwargs)
+
+
+class ValidationError(VoiceBotError):
+    """Input validation errors"""
+    def __init__(self, message: str, **kwargs):
+        super().__init__(message, category=ErrorCategory.VALIDATION, **kwargs)
+
+
+@dataclass
+class CircuitBreakerState:
+    """Circuit breaker state tracking"""
+    failures: int = 0
+    last_failure_time: float = 0
+    is_open: bool = False
+    last_success_time: float = 0
+
+
+class CircuitBreaker:
+    """Circuit breaker pattern implementation for preventing cascading failures"""
+    
+    def __init__(
+        self,
+        failure_threshold: int = 5,
+        recovery_timeout: float = 60.0,
+        expected_exception: type = Exception
+    ):
+        self.failure_threshold = failure_threshold
+        self.recovery_timeout = recovery_timeout
+        self.expected_exception = expected_exception
+        self.state = CircuitBreakerState()
+    
+    def __call__(self, func: Callable[..., T]) -> Callable[..., T]:
+        @wraps(func)
+        async def wrapper(*args, **kwargs) -> T:
+            # Check if circuit is open
+            if self.state.is_open:
+                if time.time() - self.state.last_failure_time < self.recovery_timeout:
+                    raise VoiceBotError(
+                        f"Circuit breaker open for {func.__name__}",
+                        severity=ErrorSeverity.HIGH,
+                        recoverable=False
+                    )
+                else:
+                    # Try to close circuit (half-open state)
+                    self.state.is_open = False
+                    logger.info(f"Circuit breaker half-open for {func.__name__}")
+            
+            try:
+                result = await func(*args, **kwargs)
+                # Success - reset failure count
+                self.state.failures = 0
+                self.state.last_success_time = time.time()
+                return result
+                
+            except self.expected_exception as e:
+                self.state.failures += 1
+                self.state.last_failure_time = time.time()
+                
+                if self.state.failures >= self.failure_threshold:
+                    self.state.is_open = True
+                    logger.error(
+                        f"Circuit breaker opened for {func.__name__} "
+                        f"after {self.state.failures} failures"
+                    )
+                
+                raise e
+        
+        return wrapper
+
+
+class RetryStrategy:
+    """Retry strategy with exponential backoff"""
+    
+    def __init__(
+        self,
+        max_attempts: int = 3,
+        base_delay: float = 1.0,
+        max_delay: float = 60.0,
+        backoff_factor: float = 2.0,
+        retriable_exceptions: Optional[List[type]] = None
+    ):
+        self.max_attempts = max_attempts
+        self.base_delay = base_delay
+        self.max_delay = max_delay
+        self.backoff_factor = backoff_factor
+        self.retriable_exceptions = retriable_exceptions or [Exception]
+    
+    def __call__(self, func: Callable[..., T]) -> Callable[..., T]:
+        @wraps(func)
+        async def wrapper(*args, **kwargs) -> T:
+            last_exception = None
+            
+            for attempt in range(1, self.max_attempts + 1):
+                try:
+                    return await func(*args, **kwargs)
+                    
+                except Exception as e:
+                    last_exception = e
+                    
+                    # Check if exception is retriable
+                    if not any(isinstance(e, exc_type) for exc_type in self.retriable_exceptions):
+                        raise e
+                    
+                    if attempt == self.max_attempts:
+                        logger.error(
+                            f"Function {func.__name__} failed after {self.max_attempts} attempts"
+                        )
+                        raise e
+                    
+                    # Calculate delay with exponential backoff
+                    delay = min(
+                        self.base_delay * (self.backoff_factor ** (attempt - 1)),
+                        self.max_delay
+                    )
+                    
+                    logger.warning(
+                        f"Function {func.__name__} failed (attempt {attempt}/{self.max_attempts}), "
+                        f"retrying in {delay:.2f}s: {e}"
+                    )
+                    
+                    await asyncio.sleep(delay)
+            
+            raise last_exception
+        
+        return wrapper
+
+
+class ErrorHandler:
+    """Central error handler with context and recovery strategies"""
+    
+    def __init__(self):
+        self.error_counts: Dict[str, int] = {}
+        self.error_history: List[VoiceBotError] = []
+        self.max_history = 100
+    
+    async def handle_error(
+        self,
+        error: Exception,
+        context: Dict[str, Any],
+        websocket: Optional[WebSocket] = None,
+        session_id: Optional[str] = None,
+        recovery_action: Optional[Callable] = None
+    ) -> bool:
+        """
+        Handle an error with context and optional recovery.
+        
+        Returns:
+            bool: True if error was handled successfully, False otherwise
+        """
+        # Convert to VoiceBotError if needed
+        if not isinstance(error, VoiceBotError):
+            error = VoiceBotError(
+                str(error),
+                context=context,
+                severity=ErrorSeverity.MEDIUM
+            )
+        
+        # Add context
+        error.context.update(context)
+        
+        # Track error
+        error_key = f"{error.category.value}:{type(error).__name__}"
+        self.error_counts[error_key] = self.error_counts.get(error_key, 0) + 1
+        
+        # Add to history (maintain size limit)
+        self.error_history.append(error)
+        if len(self.error_history) > self.max_history:
+            self.error_history = self.error_history[-self.max_history:]
+        
+        # Log error with context
+        log_message = (
+            f"Error in {context.get('operation', 'unknown')}: {error} "
+            f"(Category: {error.category.value}, Severity: {error.severity.value})"
+        )
+        
+        if error.severity in [ErrorSeverity.HIGH, ErrorSeverity.CRITICAL]:
+            logger.error(log_message)
+            logger.error(f"Error context: {error.context}")
+            if hasattr(error, '__traceback__'):
+                logger.error(f"Traceback: {traceback.format_tb(error.__traceback__)}")
+        else:
+            logger.warning(log_message)
+        
+        # Notify client if WebSocket available
+        if websocket and error.severity != ErrorSeverity.LOW:
+            await self._notify_client_error(websocket, error, session_id)
+        
+        # Attempt recovery if provided and error is recoverable
+        if recovery_action and error.recoverable:
+            try:
+                await recovery_action()
+                logger.info(f"Recovery action succeeded for {error_key}")
+                return True
+            except Exception as recovery_error:
+                logger.error(f"Recovery action failed for {error_key}: {recovery_error}")
+        
+        return error.severity in [ErrorSeverity.LOW, ErrorSeverity.MEDIUM]
+    
+    async def _notify_client_error(
+        self, 
+        websocket: WebSocket, 
+        error: VoiceBotError, 
+        session_id: Optional[str]
+    ):
+        """Notify client about error via WebSocket"""
+        try:
+            error_message = {
+                "type": "error",
+                "data": {
+                    "message": str(error),
+                    "category": error.category.value,
+                    "severity": error.severity.value,
+                    "recoverable": error.recoverable,
+                    "timestamp": error.timestamp
+                }
+            }
+            
+            if session_id:
+                error_message["data"]["session_id"] = session_id
+            
+            await websocket.send_json(error_message)
+            
+        except Exception as notify_error:
+            logger.error(f"Failed to notify client of error: {notify_error}")
+    
+    def get_error_statistics(self) -> Dict[str, Any]:
+        """Get error statistics for monitoring"""
+        return {
+            "error_counts": dict(self.error_counts),
+            "total_errors": len(self.error_history),
+            "recent_errors": [
+                {
+                    "message": str(err),
+                    "category": err.category.value,
+                    "severity": err.severity.value,
+                    "timestamp": err.timestamp
+                }
+                for err in self.error_history[-10:]  # Last 10 errors
+            ]
+        }
+
+
+# Global error handler instance
+error_handler = ErrorHandler()
+
+
+# Decorator shortcuts for common patterns
+def with_websocket_error_handling(func: Callable[..., T]) -> Callable[..., T]:
+    """Decorator for WebSocket operations with error handling"""
+    @wraps(func)
+    async def wrapper(*args, **kwargs) -> T:
+        try:
+            return await func(*args, **kwargs)
+        except Exception as e:
+            await error_handler.handle_error(
+                WebSocketError(f"WebSocket operation failed: {e}"),
+                context={"function": func.__name__, "args": str(args)[:200]}
+            )
+            raise
+    return wrapper
+
+
+def with_webrtc_error_handling(func: Callable[..., T]) -> Callable[..., T]:
+    """Decorator for WebRTC operations with error handling"""
+    @wraps(func)
+    async def wrapper(*args, **kwargs) -> T:
+        try:
+            return await func(*args, **kwargs)
+        except Exception as e:
+            await error_handler.handle_error(
+                WebRTCError(f"WebRTC operation failed: {e}"),
+                context={"function": func.__name__, "args": str(args)[:200]}
+            )
+            raise
+    return wrapper
+
+
+def with_session_error_handling(func: Callable[..., T]) -> Callable[..., T]:
+    """Decorator for Session operations with error handling"""
+    @wraps(func)
+    async def wrapper(*args, **kwargs) -> T:
+        try:
+            return await func(*args, **kwargs)
+        except Exception as e:
+            await error_handler.handle_error(
+                SessionError(f"Session operation failed: {e}"),
+                context={"function": func.__name__, "args": str(args)[:200]}
+            )
+            raise
+    return wrapper

server/websocket/message_handlers.py

@@ -11,6 +11,13 @@ from fastapi import WebSocket
 
 from logger import logger
 from .webrtc_signaling import WebRTCSignalingHandlers
+from core.error_handling import (
+    error_handler, 
+    WebSocketError, 
+    ValidationError,
+    with_websocket_error_handling,
+    ErrorSeverity
+)
 
 if TYPE_CHECKING:
     from ..core.session_manager import Session
@@ -322,25 +329,102 @@ class MessageRouter:
         websocket: WebSocket,
         managers: Dict[str, Any]
     ):
-        """Route a message to the appropriate handler"""
-        if message_type in self._handlers:
+        """Route a message to the appropriate handler with enhanced error handling"""
+        if message_type not in self._handlers:
+            await error_handler.handle_error(
+                ValidationError(f"Unknown message type: {message_type}"),
+                context={
+                    "message_type": message_type,
+                    "session_id": session.id if session else "unknown",
+                    "data_keys": list(data.keys()) if data else []
+                },
+                websocket=websocket,
+                session_id=session.id if session else None
+            )
+            return
+        
         try:
-                await self._handlers[message_type].handle(session, lobby, data, websocket, managers)
+            # Execute handler with context tracking
+            await self._handlers[message_type].handle(
+                session, lobby, data, websocket, managers
+            )
+            
+        except WebSocketError as e:
+            # WebSocket specific errors - attempt recovery
+            await error_handler.handle_error(
+                e,
+                context={
+                    "message_type": message_type,
+                    "session_id": session.id if session else "unknown",
+                    "handler": type(self._handlers[message_type]).__name__
+                },
+                websocket=websocket,
+                session_id=session.id if session else None,
+                recovery_action=lambda: self._websocket_recovery(websocket, session)
+            )
+            
+        except ValidationError as e:
+            # Validation errors - usually client-side issues
+            await error_handler.handle_error(
+                e,
+                context={
+                    "message_type": message_type,
+                    "session_id": session.id if session else "unknown",
+                    "data": str(data)[:500]  # Truncate large data
+                },
+                websocket=websocket,
+                session_id=session.id if session else None
+            )
+            
         except Exception as e:
-                import traceback
-                logger.error(f"Error handling message type {message_type}: {e}")
-                logger.error(f"Full traceback: {traceback.format_exc()}")
+            # Unexpected errors - enhanced logging and fallback
+            await error_handler.handle_error(
+                WebSocketError(
+                    f"Unexpected error in {message_type} handler: {e}",
+                    severity=ErrorSeverity.HIGH
+                ),
+                context={
+                    "message_type": message_type,
+                    "session_id": session.id if session else "unknown",
+                    "handler": type(self._handlers[message_type]).__name__,
+                    "exception_type": type(e).__name__,
+                    "traceback": str(e)
+                },
+                websocket=websocket,
+                session_id=session.id if session else None,
+                recovery_action=lambda: self._generic_recovery(message_type, session, lobby)
+            )
+    
+    async def _websocket_recovery(self, websocket: WebSocket, session: "Session"):
+        """WebSocket recovery action"""
+        if websocket and session:
+            # Send a connection status update
             await websocket.send_json({
-                    "type": "error",
-                    "data": {"error": f"Internal error handling {message_type}"}
-                })
-        else:
-            logger.warning(f"Unknown message type: {message_type}")
-            await websocket.send_json({
-                "type": "error", 
-                "data": {"error": f"Unknown message type: {message_type}"}
+                "type": "connection_status",
+                "data": {
+                    "status": "recovered",
+                    "session_id": session.id,
+                    "message": "Connection recovered from error"
+                }
             })
     
+    async def _generic_recovery(self, message_type: str, session: "Session", lobby: "Lobby"):
+        """Generic recovery action"""
+        # Log recovery attempt
+        logger.info(f"Attempting recovery for {message_type} error")
+        
+        # Depending on message type, perform specific recovery
+        if message_type in ["join", "part"]:
+            # For lobby operations, ensure session state consistency
+            if session and lobby:
+                # Refresh lobby state
+                await lobby.update_state()
+        
+        elif message_type == "set_name":
+            # For name operations, validate session state
+            if session:
+                logger.info(f"Validating session state for {session.id}")
+    
     def get_supported_types(self) -> list[str]:
         """Get list of supported message types"""
         return list(self._handlers.keys())

server/websocket/webrtc_signaling.py

@@ -9,6 +9,12 @@ from typing import Any, Dict, TYPE_CHECKING
 from fastapi import WebSocket
 
 from logger import logger
+from core.error_handling import (
+    with_webrtc_error_handling,
+    WebRTCError,
+    ErrorSeverity,
+    error_handler
+)
 
 if TYPE_CHECKING:
     from core.session_manager import Session
@@ -19,6 +25,7 @@ class WebRTCSignalingHandlers:
     """WebRTC signaling message handlers for peer-to-peer communication."""
 
     @staticmethod
+    @with_webrtc_error_handling
     async def handle_relay_ice_candidate(
         websocket: WebSocket,
         session: "Session",
@@ -105,6 +112,7 @@ class WebRTCSignalingHandlers:
             logger.warning(f"Failed to relay ICE candidate: {e}")
 
     @staticmethod
+    @with_webrtc_error_handling
     async def handle_relay_session_description(
         websocket: WebSocket,
         session: "Session",
@@ -199,6 +207,7 @@ class WebRTCSignalingHandlers:
             logger.warning(f"Failed to relay session description: {e}")
 
     @staticmethod
+    @with_webrtc_error_handling
     async def handle_add_peer(
         session: "Session",
         peer_session: "Session", 
@@ -233,7 +242,9 @@ class WebRTCSignalingHandlers:
                 f"has_media={session.has_media})"
             )
             try:
-                await peer_session.ws.send_json({
+                if peer_session.ws:
+                    await peer_session.ws.send_json(
+                        {
                             "type": "addPeer",
                             "data": {
                                 "peer_id": session.id,
@@ -241,7 +252,8 @@ class WebRTCSignalingHandlers:
                                 "has_media": session.has_media,
                                 "should_create_offer": False,
                             },
-                })
+                        }
+                    )
             except Exception as e:
                 logger.warning(
                     f"Failed to send addPeer to {peer_session.getName()}: {e}"
@@ -254,7 +266,9 @@ class WebRTCSignalingHandlers:
                 f"has_media={peer_session.has_media})"
             )
             try:
-                await session.ws.send_json({
+                if session.ws:
+                    await session.ws.send_json(
+                        {
                             "type": "addPeer",
                             "data": {
                                 "peer_id": peer_session.id,
@@ -262,7 +276,8 @@ class WebRTCSignalingHandlers:
                                 "has_media": peer_session.has_media,
                                 "should_create_offer": True,
                             },
-                })
+                        }
+                    )
             except Exception as e:
                 logger.warning(f"Failed to send addPeer to {session.getName()}: {e}")
         else:
@@ -273,6 +288,7 @@ class WebRTCSignalingHandlers:
             )
 
     @staticmethod
+    @with_webrtc_error_handling
     async def handle_remove_peer(
         session: "Session",
         peer_session: "Session",

tests/verify-step3.py

@@ -7,10 +7,6 @@ Step 3 focused on centralizing WebRTC peer management into the signaling module.
 """
 
 import sys
-import os
-
-# Add the server directory to Python path
-sys.path.insert(0, '/home/jketreno/docker/ai-voicebot/server')
 
 from websocket.webrtc_signaling import WebRTCSignalingHandlers
 from websocket.message_handlers import MessageRouter

tests/verify-step4.py

@@ -0,0 +1,181 @@
+#!/usr/bin/env python3
+"""
+Step 4 Verification: Enhanced Error Handling and Recovery
+
+This script verifies that Step 4 of the refactoring has been successfully completed.
+Step 4 focused on implementing robust error handling, recovery mechanisms, and resilience patterns.
+"""
+
+import sys
+import asyncio
+
+# Add the server directory to Python path
+sys.path.insert(0, '/home/jketreno/docker/ai-voicebot/server')
+
+from core.error_handling import (
+    ErrorSeverity, ErrorCategory, VoiceBotError, WebSocketError, WebRTCError,
+    SessionError, LobbyError, AuthError, PersistenceError, ValidationError,
+    CircuitBreaker, RetryStrategy, error_handler
+)
+from websocket.message_handlers import MessageRouter
+
+
+def verify_step4():
+    """Verify Step 4: Enhanced Error Handling and Recovery"""
+    print("🔄 Step 4 Verification: Enhanced Error Handling and Recovery")
+    print("=" * 65)
+    
+    success = True
+    
+    # Check error classes
+    print("\n🏗️ Custom Exception Classes:")
+    error_classes = [
+        ('VoiceBotError', VoiceBotError),
+        ('WebSocketError', WebSocketError),
+        ('WebRTCError', WebRTCError),
+        ('SessionError', SessionError),
+        ('LobbyError', LobbyError),
+        ('AuthError', AuthError),
+        ('PersistenceError', PersistenceError),
+        ('ValidationError', ValidationError),
+    ]
+    
+    for name, cls in error_classes:
+        try:
+            # Test error creation
+            error = cls("Test error", severity=ErrorSeverity.LOW)
+            print(f"  ✅ {name} - category: {error.category.value}, severity: {error.severity.value}")
+        except Exception as e:
+            print(f"  ❌ {name} - Failed to create: {e}")
+            success = False
+    
+    # Check error handler
+    print("\n🎯 Error Handler:")
+    try:
+        stats = error_handler.get_error_statistics()
+        print(f"  ✅ ErrorHandler - tracking {stats['total_errors']} total errors")
+        print(f"  ✅ Error categories: {list(stats['error_counts'].keys())}")
+    except Exception as e:
+        print(f"  ❌ ErrorHandler - Failed: {e}")
+        success = False
+    
+    # Check circuit breaker
+    print("\n🔄 Circuit Breaker Pattern:")
+    try:
+        @CircuitBreaker(failure_threshold=2, recovery_timeout=1.0)
+        async def test_function():
+            return "success"
+        
+        result = asyncio.run(test_function())
+        print(f"  ✅ CircuitBreaker - Test function returned: {result}")
+    except Exception as e:
+        print(f"  ❌ CircuitBreaker - Failed: {e}")
+        success = False
+    
+    # Check retry strategy
+    print("\n🔁 Retry Strategy:")
+    try:
+        attempt_count = 0
+        
+        @RetryStrategy(max_attempts=3, base_delay=0.1)
+        async def test_retry():
+            nonlocal attempt_count
+            attempt_count += 1
+            if attempt_count < 3:
+                raise Exception("Temporary failure")
+            return "success after retries"
+        
+        result = asyncio.run(test_retry())
+        print(f"  ✅ RetryStrategy - Result: {result} (attempts: {attempt_count})")
+    except Exception as e:
+        print(f"  ❌ RetryStrategy - Failed: {e}")
+        success = False
+    
+    # Check enhanced message router
+    print("\n📨 Enhanced Message Router:")
+    try:
+        router = MessageRouter()
+        supported_types = router.get_supported_types()
+        
+        # Check for our WebRTC handlers
+        webrtc_handlers = ['relayICECandidate', 'relaySessionDescription']
+        for handler in webrtc_handlers:
+            if handler in supported_types:
+                print(f"  ✅ {handler} handler registered")
+            else:
+                print(f"  ❌ {handler} handler missing")
+                success = False
+    except Exception as e:
+        print(f"  ❌ MessageRouter - Failed: {e}")
+        success = False
+    
+    # Test error severity and categories
+    print("\n🏷️ Error Classification:")
+    severities = [s.value for s in ErrorSeverity]
+    categories = [c.value for c in ErrorCategory]
+    print(f"  ✅ Severities: {', '.join(severities)}")
+    print(f"  ✅ Categories: {', '.join(categories)}")
+    
+    print("\n🎯 Step 4 Achievements:")
+    print("  ✅ Custom exception hierarchy with categorization")
+    print("  ✅ Error severity levels for proper handling")
+    print("  ✅ Circuit breaker pattern for fault tolerance")
+    print("  ✅ Retry strategy with exponential backoff")
+    print("  ✅ Centralized error handler with context tracking")
+    print("  ✅ Enhanced WebSocket message routing with recovery")
+    print("  ✅ WebRTC signaling with error handling decorators")
+    print("  ✅ Error statistics and monitoring capabilities")
+    print("  ✅ Graceful degradation and recovery mechanisms")
+    
+    print("\n🚀 Next Steps:")
+    print("  - Step 5: Performance optimizations and monitoring")
+    print("  - Step 6: Advanced bot management features")
+    print("  - Step 7: Security enhancements")
+    
+    if success:
+        print("\n✅ Step 4: Enhanced Error Handling and Recovery COMPLETED!")
+        return True
+    else:
+        print("\n❌ Step 4: Some error handling features failed verification")
+        return False
+
+
+async def test_error_handling_live():
+    """Test error handling with live scenarios"""
+    print("\n🧪 Live Error Handling Tests:")
+    
+    try:
+        # Test custom error creation and handling
+        test_error = WebRTCError(
+            "Test WebRTC connection failed",
+            severity=ErrorSeverity.MEDIUM,
+            context={"peer_id": "test123", "lobby_id": "lobby456"}
+        )
+        
+        # Test error handler
+        handled = await error_handler.handle_error(
+            test_error,
+            context={"operation": "test_webrtc_connection", "timestamp": "2025-09-04"}
+        )
+        
+        print(f"  ✅ Error handling test: {handled}")
+        
+        # Get error statistics
+        stats = error_handler.get_error_statistics()
+        print(f"  ✅ Error stats updated: {stats['total_errors']} total errors")
+        
+    except Exception as e:
+        print(f"  ❌ Live error handling test failed: {e}")
+
+
+if __name__ == "__main__":
+    success = verify_step4()
+    
+    # Run live tests
+    try:
+        asyncio.run(test_error_handling_live())
+    except Exception as e:
+        print(f"Live tests failed: {e}")
+        success = False
+    
+    sys.exit(0 if success else 1)