test_api.py

#!/usr/bin/env python3.11
"""Quick test of the Astrology API client with real API calls."""

import sys
sys.path.insert(0, 'src')

from astroapi import AstrologyClient, BirthData, Subject
from astroapi.types import (
    AstrologyClientConfig,
    PlanetaryPositionsRequest,
    NatalChartRequest,
    PersonalizedHoroscopeRequest,
)

# API key
API_KEY = "ask_edacc7f61f5987c591f4848af7a62df54117bc1f4a0b9d2281179f01c0eed419"

def test_1_synastry():
    """Test 1: Synastry chart (compatibility)."""
    print("\n" + "="*60)
    print("ТЕСТ 1: Синастрия (совместимость)")
    print("="*60)

    config = AstrologyClientConfig(api_key=API_KEY, debug=True)
    client = AstrologyClient(config)

    from astroapi.types import SynastryChartRequest

    person1 = Subject(
        name="Person A",
        birth_data=BirthData(
            year=1985,
            month=3,
            day=15,
            hour=10,
            minute=30,
            latitude=40.7128,
            longitude=-74.0060
        )
    )

    person2 = Subject(
        name="Person B",
        birth_data=BirthData(
            year=1987,
            month=7,
            day=22,
            hour=14,
            minute=45,
            latitude=40.7128,
            longitude=-74.0060
        )
    )

    request = SynastryChartRequest(subject1=person1, subject2=person2)

    try:
        synastry = client.charts.get_synastry_chart(request)
        print(f"✅ Синастрическая карта создана")
        if synastry.inter_aspects:
            print(f"  Межличностные аспекты: {len(synastry.inter_aspects)}")

        return True
    except Exception as e:
        print(f"❌ Ошибка: {e}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        client.close()

def test_2_natal_chart():
    """Test 2: Generate natal chart."""
    print("\n" + "="*60)
    print("ТЕСТ 2: Натальная карта")
    print("="*60)

    config = AstrologyClientConfig(api_key=API_KEY)
    client = AstrologyClient(config)

    subject = Subject(
        name="Test Person",
        birth_data=BirthData(
            year=1990,
            month=6,
            day=15,
            hour=14,
            minute=30,
            latitude=40.7128,
            longitude=-74.0060
        )
    )

    request = NatalChartRequest(subject=subject)

    try:
        chart = client.charts.get_natal_chart(request)
        print(f"✅ Натальная карта создана успешно")

        if chart.chart:
            if chart.chart.planetary_positions:
                print(f"  Планеты: {len(chart.chart.planetary_positions)}")
            if chart.chart.house_cusps:
                print(f"  Дома: {len(chart.chart.house_cusps)}")
            if chart.chart.aspects:
                print(f"  Аспекты: {len(chart.chart.aspects)}")

        return True
    except Exception as e:
        print(f"❌ Ошибка: {e}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        client.close()

def test_3_chinese_bazi():
    """Test 3: Chinese BaZi (Four Pillars)."""
    print("\n" + "="*60)
    print("ТЕСТ 3: Китайская астрология (BaZi)")
    print("="*60)

    config = AstrologyClientConfig(api_key=API_KEY)
    client = AstrologyClient(config)

    from astroapi.types import BaZiRequest

    subject = Subject(
        birth_data=BirthData(
            year=1990,
            month=6,
            day=15,
            hour=12,
            minute=0,
            latitude=40.7128,
            longitude=-74.0060,
            timezone="America/New_York"
        )
    )

    request = BaZiRequest(subject=subject, language="en")

    try:
        bazi = client.chinese.get_bazi(request)
        print(f"✅ BaZi анализ получен")
        if bazi.day_master:
            print(f"  День Мастер: {bazi.day_master}")
        if bazi.year_pillar:
            print(f"  Год: {bazi.year_pillar.heavenly_stem} {bazi.year_pillar.earthly_branch}")

        return True
    except Exception as e:
        print(f"❌ Ошибка: {e}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        client.close()

def test_4_glossary():
    """Test 4: Search glossary."""
    print("\n" + "="*60)
    print("ТЕСТ 4: Глоссарий (поиск городов)")
    print("="*60)

    config = AstrologyClientConfig(api_key=API_KEY)
    client = AstrologyClient(config)

    try:
        cities = client.glossary.get_cities(query="New York", limit=5)
        print(f"✅ Найдено результатов: {cities.count if cities.count else 'N/A'}")

        if cities.items:
            for i, city in enumerate(cities.items[:3], 1):
                print(f"  {i}. {city}")

        return True
    except Exception as e:
        print(f"❌ Ошибка: {e}")
        import traceback
        traceback.print_exc()
        return False
    finally:
        client.close()

def main():
    """Run all tests."""
    print("\n" + "="*60)
    print("ТЕСТИРОВАНИЕ ASTROLOGY API CLIENT")
    print("="*60)

    results = []

    # Run tests
    results.append(("Синастрия", test_1_synastry()))
    results.append(("Натальная карта", test_2_natal_chart()))
    results.append(("Китайская астрология", test_3_chinese_bazi()))
    results.append(("Глоссарий", test_4_glossary()))

    # Summary
    print("\n" + "="*60)
    print("РЕЗУЛЬТАТЫ ТЕСТИРОВАНИЯ")
    print("="*60)

    passed = sum(1 for _, result in results if result)
    total = len(results)

    for name, result in results:
        status = "✅ ПРОШЕЛ" if result else "❌ НЕ ПРОШЕЛ"
        print(f"{status}: {name}")

    print(f"\nИтого: {passed}/{total} тестов прошло успешно")

    if passed == total:
        print("\n🎉 ВСЕ ТЕСТЫ ПРОШЛИ УСПЕШНО!")
        return 0
    else:
        print(f"\n⚠️ {total - passed} тест(ов) не прошли")
        return 1

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