responses.py

"""Mock API responses for testing."""

from __future__ import annotations

# Planetary Positions Response
MOCK_POSITIONS_RESPONSE = {
    "data": {
        "positions": [
            {
                "name": "Sun",
                "longitude": 283.45,
                "latitude": 0.0,
                "speed": 0.958,
                "sign": "Capricorn",
                "sign_longitude": 13.45,
                "house": 10,
                "retrograde": False,
            },
            {
                "name": "Moon",
                "longitude": 145.23,
                "latitude": 5.12,
                "speed": 13.176,
                "sign": "Leo",
                "sign_longitude": 25.23,
                "house": 5,
                "retrograde": False,
            },
            {
                "name": "Mercury",
                "longitude": 275.89,
                "latitude": -2.45,
                "speed": 1.234,
                "sign": "Capricorn",
                "sign_longitude": 5.89,
                "house": 9,
                "retrograde": False,
            },
        ]
    }
}

# Natal Chart Response
MOCK_NATAL_CHART_RESPONSE = {
    "result": {
        "chart": {
            "planetary_positions": [
                {
                    "name": "Sun",
                    "longitude": 283.45,
                    "latitude": 0.0,
                    "speed": 0.958,
                    "sign": "Capricorn",
                    "house": 10,
                },
                {
                    "name": "Moon",
                    "longitude": 145.23,
                    "latitude": 5.12,
                    "speed": 13.176,
                    "sign": "Leo",
                    "house": 5,
                },
            ],
            "house_cusps": [
                {"house": 1, "longitude": 90.0, "sign": "Cancer"},
                {"house": 2, "longitude": 120.0, "sign": "Leo"},
                {"house": 3, "longitude": 150.0, "sign": "Virgo"},
            ],
            "aspects": [
                {
                    "point1": "Sun",
                    "point2": "Moon",
                    "type": "Trine",
                    "angle": 120.0,
                    "orb": 2.22,
                    "applying": True,
                }
            ],
        }
    }
}

# Synastry Chart Response
MOCK_SYNASTRY_RESPONSE = {
    "data": {
        "subject1_chart": {
            "planetary_positions": [{"name": "Sun", "longitude": 283.45, "sign": "Capricorn"}]
        },
        "subject2_chart": {
            "planetary_positions": [{"name": "Sun", "longitude": 150.0, "sign": "Virgo"}]
        },
        "inter_aspects": [
            {
                "point1": "Sun",
                "point2": "Moon",
                "type": "Sextile",
                "angle": 60.0,
                "orb": 1.5,
            }
        ],
    }
}

# Horoscope Response
MOCK_HOROSCOPE_RESPONSE = {
    "data": {
        "content": "Today is a great day for new beginnings. The planetary alignments suggest...",
        "date": "2024-06-15",
        "period": "daily",
        "transits": [
            {"planet": "Mars", "house": 7, "aspect": "opposition", "native_planet": "Venus"}
        ],
    }
}

# BaZi Response
MOCK_BAZI_RESPONSE = {
    "result": {
        "year_pillar": {
            "heavenly_stem": "Geng",
            "earthly_branch": "Wu",
            "element": "Metal",
            "animal": "Horse",
            "yin_yang": "Yang",
        },
        "month_pillar": {
            "heavenly_stem": "Ren",
            "earthly_branch": "Wu",
            "element": "Water",
            "animal": "Horse",
            "yin_yang": "Yang",
        },
        "day_pillar": {
            "heavenly_stem": "Ji",
            "earthly_branch": "Wei",
            "element": "Earth",
            "animal": "Goat",
            "yin_yang": "Yin",
        },
        "hour_pillar": {
            "heavenly_stem": "Geng",
            "earthly_branch": "Wu",
            "element": "Metal",
            "animal": "Horse",
            "yin_yang": "Yang",
        },
        "day_master": {
            "element": "Earth",
            "chinese": "己",
            "strength_score": 88,
            "classification": "Very Strong",
        },
        "element_balance": {
            "Wood": 10,
            "Fire": 30,
            "Earth": 40,
            "Metal": 15,
            "Water": 5,
        },
    }
}

# Glossary Response
MOCK_GLOSSARY_CITIES_RESPONSE = {
    "data": {
        "items": [
            {
                "name": "New York",
                "country": "United States",
                "country_code": "US",
                "latitude": 40.7128,
                "longitude": -74.0060,
                "timezone": "America/New_York",
                "population": 8336817,
            },
            {
                "name": "London",
                "country": "United Kingdom",
                "country_code": "GB",
                "latitude": 51.5074,
                "longitude": -0.1278,
                "timezone": "Europe/London",
                "population": 8982000,
            },
        ],
        "count": 2,
    }
}

# Generic Error Response
MOCK_ERROR_RESPONSE = {
    "message": "Invalid API key",
    "code": "UNAUTHORIZED",
    "details": {"reason": "The provided API key is not valid"},
}

# Validation Error Response
MOCK_VALIDATION_ERROR = {
    "detail": [
        {
            "type": "missing",
            "loc": ["body", "subject"],
            "msg": "Field required",
            "input": {"year": 1990},
        }
    ]
}