#!/usr/bin/env python3
"""
Test script to verify all 5 pivot levels are being calculated
"""

import sys
import os
import logging

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

from core.data_provider import DataProvider

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

def test_pivot_levels():
    """Test all 5 pivot levels calculation"""
    try:
        logger.info("Initializing DataProvider...")
        data_provider = DataProvider()

        # Wait for initial data to load
        import time
        time.sleep(3)

        # Test pivot levels for ETH/USDT
        symbol = 'ETH/USDT'
        logger.info(f"\nTesting Williams pivot levels for {symbol}:")

        # Get pivot levels
        pivot_levels = data_provider.get_williams_pivot_levels(symbol, base_timeframe='1m', limit=5000)

        if not pivot_levels:
            logger.error(f"❌ NO PIVOT LEVELS for {symbol}")
            return False

        logger.info(f"✅ Found {len(pivot_levels)} pivot levels")

        for level_num in sorted(pivot_levels.keys()):
            trend_level = pivot_levels[level_num]
            pivot_count = len(getattr(trend_level, 'pivot_points', []))
            direction = getattr(trend_level, 'trend_direction', 'unknown')
            strength = getattr(trend_level, 'trend_strength', 0.0)

            logger.info(f"  Level {level_num}: {pivot_count} pivots, {direction} ({strength:.1%})")

            if pivot_count > 0:
                # Show sample of pivot types
                high_count = sum(1 for p in trend_level.pivot_points if getattr(p, 'pivot_type', '') == 'high')
                low_count = sum(1 for p in trend_level.pivot_points if getattr(p, 'pivot_type', '') == 'low')
                logger.info(f"    High pivots: {high_count}, Low pivots: {low_count}")

        # Check if we have all levels
        expected_levels = {1, 2, 3, 4, 5}
        actual_levels = set(pivot_levels.keys())

        if expected_levels.issubset(actual_levels):
            logger.info("✅ ALL 5 PIVOT LEVELS PRESENT!")
        else:
            missing = expected_levels - actual_levels
            logger.warning(f"❌ MISSING LEVELS: {missing}")

        return True

    except Exception as e:
        logger.error(f"Test failed with error: {e}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    success = test_pivot_levels()
    if success:
        print("\n🎉 Pivot levels test completed!")
    else:
        print("\n❌ Pivot levels test failed!")
        sys.exit(1)