#!/usr/bin/env python3
"""
Test script to verify that the NBA leaderboard fix works correctly.
This script simulates the leaderboard manager's data fetching process.
"""
import sys
import os
import logging

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

# Set up logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

def test_nba_standings_data():
    """Test that NBA standings data includes team ID fields."""
    try:
        from leaderboard_manager import LeaderboardManager
        from display_manager import DisplayManager
        from cache_manager import CacheManager
        import json

        # Load config
        with open('config/config.json', 'r') as f:
            config = json.load(f)

        # Create mock display and cache managers
        display_manager = DisplayManager(config)
        cache_manager = CacheManager()

        # Create leaderboard manager
        leaderboard_manager = LeaderboardManager(config, display_manager)

        # Test NBA standings fetching
        logger.info("Testing NBA standings data fetching...")
        nba_config = leaderboard_manager.league_configs['nba']
        nba_config['enabled'] = True  # Enable NBA for testing

        standings = leaderboard_manager._fetch_standings(nba_config)

        if not standings:
            logger.error("No NBA standings data returned!")
            return False

        logger.info(f"Successfully fetched {len(standings)} NBA teams")

        # Check if team ID fields are present
        missing_id_count = 0
        for i, team in enumerate(standings[:5]):  # Check first 5 teams
            team_id = team.get('id')
            team_abbr = team.get('abbreviation', 'Unknown')
            team_name = team.get('name', 'Unknown')

            logger.info(f"Team {i+1}: ID={team_id}, ABBR={team_abbr}, NAME={team_name}")

            if team_id is None:
                logger.error(f"Team {team_abbr} is missing ID field!")
                missing_id_count += 1

        if missing_id_count > 0:
            logger.error(f"{missing_id_count} teams are missing ID fields!")
            return False
        else:
            logger.info("All tested teams have ID fields!")

        # Test that we can create a leaderboard image (without actually displaying)
        logger.info("Testing leaderboard image creation...")
        leaderboard_manager.leaderboard_data = [{
            'league': 'nba',
            'league_config': nba_config,
            'teams': standings[:3]  # Test with first 3 teams
        }]

        try:
            leaderboard_manager._create_leaderboard_image()
            if leaderboard_manager.leaderboard_image:
                logger.info(f"Successfully created leaderboard image: {leaderboard_manager.leaderboard_image.width}x{leaderboard_manager.leaderboard_image.height}")
                return True
            else:
                logger.error("Failed to create leaderboard image!")
                return False
        except Exception as e:
            logger.error(f"Error creating leaderboard image: {e}")
            return False

    except ImportError as e:
        logger.error(f"Import error: {e}")
        logger.info("This script needs to be run from the LEDMatrix project directory")
        return False
    except Exception as e:
        logger.error(f"Unexpected error: {e}")
        return False

def main():
    """Main test function."""
    logger.info("Testing NBA leaderboard fix...")

    success = test_nba_standings_data()

    if success:
        logger.info("✅ NBA leaderboard fix test PASSED!")
        logger.info("The NBA leaderboard should now work correctly with team logos")
    else:
        logger.error("❌ NBA leaderboard fix test FAILED!")
        logger.info("The issue may not be fully resolved")

    return success

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