#!/usr/bin/env python3
"""Tests for muhurta.py — Panchanga calculation, auspicious/inauspicious periods."""

from __future__ import annotations
import sys, os, pytest

SCRIPTS = os.path.join(os.path.dirname(__file__), '..', 'scripts')
if SCRIPTS not in sys.path:
    sys.path.insert(0, SCRIPTS)

from muhurta import (
    calc_tithi, calc_nakshatra_from_lon, calc_yoga, calc_karana,
    calc_vara, calc_hora, calc_abhijit_muhurta, calc_panchanga,
    check_activity_muhurta, muhurta_full_report, calc_daytime_inauspicious_periods,
    panchanga_range_report, muhurta_range_search, calc_sunrise_sunset_local, calc_panchanga_end_times,
    build_muhurta_sidecar,
    classify_vrata_tags, classify_panchanga_condition_tags, calc_choghadiya_windows, calc_hora_windows,
    TITHI_NAMES, TITHI_QUALITY, NAKSHATRAS, NAKSHATRA_TYPE,
    YOGA_NAMES, YOGA_QUALITY, KARANA_NAMES, KARANA_QUALITY,
    VARA_LORDS, ACTIVITY_RULES,
)


# ── Tithi Tests ─────────────────────────────────────────────────────

class TestTithi:
    def test_new_moon_tithi_30(self):
        # Moon-Sun diff = 0 → tithi 1
        result = calc_tithi(0.0, 0.0)
        assert result['tithi_num'] == 1

    def test_full_moon_tithi_15(self):
        # Moon-Sun diff = 180° → tithi 16 (Krishna 1)
        result = calc_tithi(0.0, 180.0)
        assert result['tithi_num'] == 16  # Krishna Pratipada

    def test_shukla_paksha(self):
        result = calc_tithi(0.0, 60.0)  # 60/12=5+1=6
        assert result['paksha'] == 'Shukla'

    def test_krishna_paksha(self):
        result = calc_tithi(0.0, 210.0)  # 210/12=17+1=18
        assert result['paksha'] == 'Krishna'

    def test_purnima_name(self):
        result = calc_tithi(0.0, 174.0)  # ~15th tithi
        if result['tithi_num'] == 15:
            assert result['name'] == 'Purnima'

    def test_amavasya_name(self):
        result = calc_tithi(0.0, 354.0)  # ~30th tithi
        if result['tithi_num'] == 30:
            assert result['name'] == 'Amavasya'

    @pytest.mark.parametrize("diff,tithi",
        [(0, 1), (12, 2), (24, 3), (168, 15), (180, 16), (348, 30)])
    def test_tithi_from_diff(self, diff, tithi):
        result = calc_tithi(0.0, float(diff))
        assert result['tithi_num'] == tithi

    def test_tithi_quality_present(self):
        result = calc_tithi(0.0, 60.0)
        assert result['quality'] in ('subha', 'asubha', 'mixed')


# ── Nakshatra from Longitude Tests ──────────────────────────────────

class TestNakshatraFromLon:
    def test_ashwini_at_0(self):
        result = calc_nakshatra_from_lon(0.0)
        assert result['nakshatra'] == 'Ashwini'

    def test_revati_near_360(self):
        result = calc_nakshatra_from_lon(355.0)
        assert result['nakshatra'] == 'Revati'

    def test_pada_range(self):
        for lon in [1.0, 5.0, 8.0, 11.0]:
            result = calc_nakshatra_from_lon(lon)
            assert 1 <= result['pada'] <= 4

    def test_27_nakshatras(self):
        assert len(NAKSHATRAS) == 27

    def test_nakshatra_type(self):
        result = calc_nakshatra_from_lon(0.0)  # Ashwini
        assert result['type'] == 'laghu'


# ── Yoga Tests ──────────────────────────────────────────────────────

class TestYoga:
    def test_vishkambha_at_low_sum(self):
        result = calc_yoga(0.0, 0.0)  # sum=0 → idx 0
        assert result['yoga'] == 'Vishkambha'

    def test_yoga_idx_range(self):
        result = calc_yoga(180.0, 180.0)
        assert 0 <= result['yoga_idx'] < 27

    def test_yoga_quality_present(self):
        result = calc_yoga(45.0, 120.0)
        assert result['quality'] in ('subha', 'asubha', 'mixed')

    def test_27_yogas(self):
        assert len(YOGA_NAMES) == 27

    def test_vaidhriti_asubha(self):
        assert YOGA_QUALITY['Vaidhriti'] == 'asubha'

    def test_priti_subha(self):
        assert YOGA_QUALITY['Priti'] == 'subha'


# ── Karana Tests ────────────────────────────────────────────────────

class TestKarana:
    def test_kimstughna_first_half(self):
        # Moon-Sun diff near 0 → k_num=0
        result = calc_karana(0.0, 1.0)
        assert result['karana'] == 'Kimstughna'

    def test_vishti_asubha(self):
        assert KARANA_QUALITY['Vishti'] == 'asubha'

    def test_karana_quality_present(self):
        result = calc_karana(0.0, 60.0)
        assert result['quality'] in ('subha', 'asubha', 'mixed')

    def test_11_karana_names(self):
        assert len(KARANA_NAMES) == 11

    def test_is_vishti_flag(self):
        # Test when karana is Vishti
        result = calc_karana(0.0, 60.0)
        assert isinstance(result['is_vishti'], bool)


# ── Vara Tests ──────────────────────────────────────────────────────

class TestVara:
    def test_sunday(self):
        result = calc_vara(0)
        assert result['vara'] == 'Sunday'
        assert result['vara_lord'] == 'Sun'
        assert result['quality'] == 'asubha'

    def test_monday(self):
        result = calc_vara(1)
        assert result['vara_lord'] == 'Moon'
        assert result['quality'] == 'subha'

    def test_thursday_jupiter(self):
        result = calc_vara(4)
        assert result['vara_lord'] == 'Jupiter'

    def test_7_days(self):
        assert len(VARA_LORDS) == 7

    @pytest.mark.parametrize("idx,expected_lord",
        [(0, 'Sun'), (1, 'Moon'), (2, 'Mars'), (3, 'Mercury'),
         (4, 'Jupiter'), (5, 'Venus'), (6, 'Saturn')])
    def test_all_vara_lords(self, idx, expected_lord):
        result = calc_vara(idx)
        assert result['vara_lord'] == expected_lord


# ── Hora Tests ──────────────────────────────────────────────────────

class TestHora:
    def test_hora_lord_calculated(self):
        result = calc_hora(0, 6.0)  # Sunday, 6h from sunrise
        assert result['hora_lord'] in ('Sun', 'Moon', 'Mars', 'Mercury',
                                        'Jupiter', 'Venus', 'Saturn')

    def test_hora_quality_present(self):
        result = calc_hora(0, 6.0)
        assert result['quality'] in ('subha', 'asubha', 'mixed')

    def test_hora_num_range(self):
        result = calc_hora(0, 6.0)
        assert 1 <= result['hora_num'] <= 24


# ── Abhijit Muhurta Tests ──────────────────────────────────────────

class TestAbhijitMuhurta:
    def test_basic_structure(self):
        result = calc_abhijit_muhurta()
        assert 'duration_minutes' in result
        assert result['duration_minutes'] == 48

    def test_wednesday_warning(self):
        result = calc_abhijit_muhurta()
        assert 'Wednesday' in result['warning']

    def test_with_jd_values(self):
        result = calc_abhijit_muhurta(2460000.0, 2460000.5)
        assert 'abhijit_start_jd' in result
        assert 'abhijit_end_jd' in result


class TestDaytimeInauspiciousPeriods:
    def test_monday_rahu_kala_uses_second_segment(self):
        result = calc_daytime_inauspicious_periods(1, '06:00', '18:00')
        assert result['rahu_kala']['segment'] == 2
        assert result['rahu_kala']['start'] == '07:30'
        assert result['rahu_kala']['end'] == '09:00'

    def test_gulika_and_yamaganda_present(self):
        result = calc_daytime_inauspicious_periods(4, '06:00', '18:00')
        assert result['policy'] == 'daytime_8_segments_local_clock'
        assert 'yamaganda' in result
        assert 'gulika' in result

    def test_location_aware_sunrise_sunset(self):
        result = calc_sunrise_sunset_local(2026, 6, 22, 28.6, 77.2, 5.5)
        assert result['policy'] in {'swisseph_rise_trans', 'location_aware_solar_event_approximation'}
        assert ':' in result['sunrise']
        assert ':' in result['sunset']


# ── Panchanga Complete Tests ────────────────────────────────────────

class TestPanchangaComplete:
    def test_five_elements_present(self):
        result = calc_panchanga(45.0, 120.0, 4, 6.0)
        assert 'tithi' in result
        assert 'nakshatra' in result
        assert 'yoga' in result
        assert 'karana' in result
        assert 'vara' in result

    def test_overall_score_range(self):
        result = calc_panchanga(45.0, 120.0, 4, 6.0)
        assert 0 <= result['overall_score'] <= 1

    def test_auspicious_count(self):
        result = calc_panchanga(45.0, 120.0, 4, 6.0)
        assert 0 <= result['auspicious_count'] <= 6

    def test_warnings_list(self):
        result = calc_panchanga(45.0, 120.0, 4, 6.0)
        assert isinstance(result['warnings'], list)

    def test_panchanga_range_report_includes_inauspicious_periods(self):
        result = panchanga_range_report('2026-06-22', '2026-06-24')
        assert result['day_count'] == 3
        assert result['days'][0]['query_date'] == '2026-06-22'
        assert result['days'][0]['inauspicious_periods']['rahu_kala']['label'] == 'Rahu Kala'
        assert result['calculation_policy']['inauspicious_periods']

    def test_panchanga_range_report_can_use_location_aware_solar_times(self):
        result = panchanga_range_report('2026-06-22', '2026-06-22', lat=28.6, lon=77.2, tz=5.5)
        assert result['location']['lat'] == 28.6
        assert result['calculation_policy']['sunrise_sunset'] in {
            'SwissEph rise_trans',
            'location-aware solar approximation',
            'mixed SwissEph rise_trans with approximation fallback',
        }
        assert 'solar_times' in result['days'][0]
        assert 'end_times' in result['days'][0]
        assert 'vrata_tags' in result['days'][0]
        assert 'condition_tags' in result['days'][0]
        assert 'festival_details' in result['days'][0]
        assert 'search_summary' in result
        assert 'condition_counts' in result['search_summary']
        assert 'choghadiya' in result['days'][0]
        assert 'hora_windows' in result['days'][0]

    def test_panchanga_range_report_includes_month_grid_for_single_month(self):
        result = panchanga_range_report('2026-06-01', '2026-06-30', activity='business', lat=28.6, lon=77.2, tz=5.5)
        assert result['month_grid']
        cells = [cell for week in result['month_grid'] for cell in week if cell]
        assert len(cells) == 30
        assert cells[0]['date'] == '2026-06-01'
        assert 'best_activities' in cells[0]
        assert 'vrata_tags' in cells[0]
        assert 'condition_tags' in cells[0]

    def test_festival_details_explain_candidate_basis(self):
        result = panchanga_range_report('2026-06-01', '2026-06-30', lat=28.6, lon=77.2, tz=5.5)
        details = [detail for day in result['days'] for detail in day.get('festival_details', [])]
        assert details
        assert all('basis' in detail for detail in details)
        assert any('confirmation_note' in detail for detail in details)
        assert result['search_summary']['query_examples']

    def test_panchanga_end_times_are_available_with_swisseph(self):
        result = calc_panchanga_end_times(2026, 6, 22, 5.5, 11.4)
        assert result is not None
        assert result['policy'] == 'swisseph_boundary_bisection'
        assert 'tithi' in result
        assert 'ends_at' in result['tithi']

    def test_basic_vrata_tags(self):
        assert classify_vrata_tags({'tithi_num': 11, 'paksha': 'Shukla', 'name': 'Ekadashi'})[0]['key'] == 'ekadashi'
        assert classify_vrata_tags({'tithi_num': 28, 'paksha': 'Krishna', 'name': 'Trayodashi'})[0]['key'] == 'pradosham'
        assert classify_vrata_tags({'tithi_num': 30, 'paksha': 'Krishna', 'name': 'Amavasya'})[0]['key'] == 'amavasya'

    def test_richer_vrata_tags_use_nakshatra_and_vara(self):
        tags = classify_vrata_tags(
            {'tithi_num': 13, 'paksha': 'Shukla', 'name': 'Trayodashi'},
            {'nakshatra': 'Pushya'},
            {'vara': 'Thursday'},
        )
        keys = {tag['key'] for tag in tags}
        assert {'pradosham', 'pushya_nakshatra', 'guru_pushya'} <= keys

    def test_vrata_tags_mark_masa_dependent_festivals_as_candidates(self):
        tags = classify_vrata_tags({'tithi_num': 3, 'paksha': 'Shukla', 'name': 'Tritiya'})
        assert any(tag['key'] == 'akshaya_tritiya_candidate' and tag['type'] == 'festival_candidate' for tag in tags)

    def test_panchanga_condition_tags_are_searchable(self):
        report = {
            'panchanga': {'warnings': ['avoid']},
            'summary': {'best_activities': ['business'], 'warnings': ['avoid']},
            'vrata_tags': [{'key': 'ekadashi', 'type': 'fasting'}],
            'activity_checks': {'business': {'verdict': '大吉（Excellent）'}},
            'choghadiya': {'day': [{'quality': 'auspicious'}], 'night': []},
        }
        keys = {tag['key'] for tag in classify_panchanga_condition_tags(report, activity='business')}
        assert {'has_vrata', 'spiritual_practice', 'auspicious_activity', 'avoid_new_start', 'good_choghadiya', 'selected_activity_good'} <= keys

    def test_choghadiya_and_hora_windows(self):
        choghadiya = calc_choghadiya_windows(1, '06:00', '18:00')
        assert choghadiya['day'][0]['name'] == 'Amrit'
        assert len(choghadiya['day']) == 8
        assert len(choghadiya['night']) == 8
        hora = calc_hora_windows(1, '06:00', '18:00')
        assert hora['day'][0]['lord'] == 'Moon'
        assert len(hora['day']) == 12
        assert len(hora['night']) == 12


# ── Activity Muhurta Tests ─────────────────────────────────────────

class TestActivityMuhurta:
    def test_marriage_activity(self):
        panchanga = calc_panchanga(45.0, 120.0, 4, 6.0)
        result = check_activity_muhurta(panchanga, 'marriage')
        assert result['verdict'] is not None

    def test_all_5_activities(self):
        panchanga = calc_panchanga(45.0, 120.0, 4, 6.0)
        for act in ['marriage', 'business', 'travel', 'medical', 'education']:
            result = check_activity_muhurta(panchanga, act)
            assert 'verdict' in result

    def test_unknown_activity_error(self):
        panchanga = calc_panchanga(45.0, 120.0, 4, 6.0)
        result = check_activity_muhurta(panchanga, 'unknown')
        assert 'error' in result

    def test_good_tithi_for_marriage(self):
        # Create a panchanga with tithi 2 (good for marriage)
        panchanga = calc_panchanga(0.0, 24.0, 4, 6.0)  # tithi 3
        result = check_activity_muhurta(panchanga, 'marriage')
        assert result['tithi_eval'] in ('good', 'neutral', 'bad')

    def test_pushya_nakshatra_good_for_business(self):
        # Pushya nakshatra center ≈ 104°
        panchanga = calc_panchanga(80.0, 104.0, 4, 6.0)
        result = check_activity_muhurta(panchanga, 'business')
        if panchanga['nakshatra']['nakshatra'] == 'Pushya':
            assert result['nakshatra_eval'] == 'good'


# ── Full Report Tests ───────────────────────────────────────────────

class TestMuhurtaFullReport:
    def test_report_structure(self):
        result = muhurta_full_report(45.0, 120.0, 4, 6.0, '2026-06-04')
        assert 'panchanga' in result
        assert 'abhijit_muhurta' in result
        assert 'activity_checks' in result
        assert 'summary' in result

    def test_best_activities_list(self):
        result = muhurta_full_report(45.0, 120.0, 4, 6.0)
        assert isinstance(result['summary']['best_activities'], list)

    def test_avoid_activities_list(self):
        result = muhurta_full_report(45.0, 120.0, 4, 6.0)
        assert isinstance(result['summary']['avoid_activities'], list)

    def test_query_date_preserved(self):
        result = muhurta_full_report(45.0, 120.0, 4, 6.0, '2026-06-04')
        assert result['query_date'] == '2026-06-04'


class TestMuhurtaRangeSearch:
    def test_returns_ranked_activity_windows_for_date_range(self):
        result = muhurta_range_search(
            '2026-06-22',
            '2026-06-28',
            activity='business',
            limit=3,
            sunrise='06:00',
            sunset='18:00',
        )

        assert result['mode'] == 'muhurta_date_range_solver'
        assert result['activity'] == 'business'
        assert result['date_range'] == {'start': '2026-06-22', 'end': '2026-06-28'}
        assert result['candidate_count'] <= 3
        assert result['best_windows']
        assert result['rejected_dates']
        assert result['constraints']['avoid_inauspicious_periods'] is True
        first = result['best_windows'][0]
        assert {'date', 'score', 'quality', 'activity_verdict', 'recommended_windows', 'evidence'} <= set(first)
        assert first['recommended_windows']
        assert first['evidence']['panchanga']


class TestMuhurtaSidecar:
    def test_build_muhurta_sidecar_returns_compact_workflow_shape(self):
        result = build_muhurta_sidecar(
            date_str='2026-07-08',
            activity='business',
            lat=36.42,
            lon=114.2,
            tz=8,
        )
        assert result['status'] == 'ok'
        assert result['source'] == 'local_muhurta.py'
        assert result['activity'] == 'business'
        assert result['report_mode'] == 'muhurta_date_range_solver'
        assert result['panchanga']['query_date'] == '2026-07-08'
        assert result['best_windows']
        assert 'next_action' in result