LEDMatrix/test/test_soccer_logo_permission_fix.py

#!/usr/bin/env python3
"""
Test script to verify the soccer logo permission fix.
This script tests the _load_and_resize_logo method to ensure it can handle permission errors
gracefully and provide helpful error messages.
"""

import os
import sys
import tempfile
import shutil
from PIL import Image, ImageDraw, ImageFont
import random

# Add the src directory to the path so we can import the modules
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

try:
    from cache_manager import CacheManager
    from soccer_managers import BaseSoccerManager
    from display_manager import DisplayManager
except ImportError as e:
    print(f"Import error: {e}")
    print("Make sure you're running this from the LEDMatrix root directory")
    sys.exit(1)

def test_soccer_logo_permission_handling():
    """Test that soccer logo permission errors are handled gracefully."""
    
    print("Testing soccer logo permission handling...")
    
    # Create a temporary directory for testing
    test_dir = tempfile.mkdtemp(prefix="ledmatrix_test_")
    print(f"Using test directory: {test_dir}")
    
    try:
        # Create a minimal config
        config = {
            "soccer_scoreboard": {
                "enabled": True,
                "logo_dir": "assets/sports/soccer_logos",
                "update_interval_seconds": 60,
                "target_leagues": ["mls", "epl", "bundesliga"]
            },
            "display": {
                "width": 64,
                "height": 32
            }
        }
        
        # Create cache manager with test directory
        cache_manager = CacheManager()
        # Override cache directory for testing
        cache_manager.cache_dir = test_dir
        
        # Create a mock display manager
        class MockDisplayManager:
            def __init__(self):
                self.width = 64
                self.height = 32
                self.image = Image.new('RGB', (64, 32), (0, 0, 0))
        
        display_manager = MockDisplayManager()
        
        # Create soccer manager
        soccer_manager = BaseSoccerManager(config, display_manager, cache_manager)
        
        # Test teams that might not have logos
        test_teams = ["ATX", "STL", "SD", "CLT", "TEST1", "TEST2"]
        
        print("\nTesting logo creation for missing teams:")
        for team in test_teams:
            print(f"  Testing {team}...")
            try:
                logo = soccer_manager._load_and_resize_logo(team)
                if logo:
                    print(f"    ✓ Successfully created logo for {team} (size: {logo.size})")
                else:
                    print(f"    ✗ Failed to create logo for {team}")
            except Exception as e:
                print(f"    ✗ Error creating logo for {team}: {e}")
        
        # Check if placeholder logos were created in cache
        placeholder_dir = os.path.join(test_dir, 'placeholder_logos')
        if os.path.exists(placeholder_dir):
            placeholder_files = os.listdir(placeholder_dir)
            print(f"\nPlaceholder logos created in cache: {len(placeholder_files)} files")
            for file in placeholder_files:
                print(f"  - {file}")
        else:
            print("\nNo placeholder logos directory created (using in-memory placeholders)")
        
        print("\n✓ Soccer logo permission test completed successfully!")
        
    except Exception as e:
        print(f"\n✗ Test failed with error: {e}")
        import traceback
        traceback.print_exc()
        return False
    
    finally:
        # Clean up test directory
        try:
            shutil.rmtree(test_dir)
            print(f"Cleaned up test directory: {test_dir}")
        except Exception as e:
            print(f"Warning: Could not clean up test directory: {e}")
    
    return True

def test_permission_error_messages():
    """Test that permission error messages include helpful instructions."""
    
    print("\nTesting permission error message format...")
    
    # This test verifies that the error messages include the fix script instruction
    # We can't easily simulate permission errors in a test environment,
    # but we can verify the code structure is correct
    
    try:
        from soccer_managers import BaseSoccerManager
        import inspect
        
        # Get the source code of the _load_and_resize_logo method
        source = inspect.getsource(BaseSoccerManager._load_and_resize_logo)
        
        # Check that the method includes permission error handling
        if "Permission denied" in source and "fix_assets_permissions.sh" in source:
            print("✓ Permission error handling with helpful messages is implemented")
            return True
        else:
            print("✗ Permission error handling is missing or incomplete")
            return False
            
    except Exception as e:
        print(f"✗ Error checking permission error handling: {e}")
        return False

if __name__ == "__main__":
    print("LEDMatrix Soccer Logo Permission Fix Test")
    print("=" * 50)
    
    success1 = test_soccer_logo_permission_handling()
    success2 = test_permission_error_messages()
    
    if success1 and success2:
        print("\n🎉 All tests passed! The soccer logo permission fix is working correctly.")
        print("\nTo apply this fix on your Raspberry Pi:")
        print("1. Transfer the updated files to your Pi")
        print("2. Run: chmod +x fix_assets_permissions.sh")
        print("3. Run: sudo ./fix_assets_permissions.sh")
        print("4. Restart your LEDMatrix application")
    else:
        print("\n❌ Tests failed. Please check the error messages above.")
        sys.exit(1)