"""Tests for the YAML loader and news-window logic in calendar_parse."""

from __future__ import annotations

import sys
import textwrap
from datetime import date, time
from pathlib import Path

import pytest

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

from scripts.calendar_parse import (  # noqa: E402
    is_in_news_window,
    load_calendar,
)


REPO_ROOT = Path(__file__).resolve().parent.parent
CALENDAR_PATH = REPO_ROOT / "calendar_evenimente.yaml"


def test_load_calendar() -> None:
    events = load_calendar(CALENDAR_PATH)
    assert isinstance(events, list)
    assert len(events) > 0
    required = {"name", "cadence", "time_ro", "severity", "window_before_min", "window_after_min"}
    for ev in events:
        missing = required - set(ev.keys())
        assert not missing, f"event {ev.get('name')!r} missing fields: {missing}"


def test_load_calendar_bad_version(tmp_path: Path) -> None:
    bad = tmp_path / "bad.yaml"
    bad.write_text(
        textwrap.dedent(
            """
            schema_version: 99
            events: []
            """
        ).strip()
        + "\n",
        encoding="utf-8",
    )
    with pytest.raises(ValueError):
        load_calendar(bad)


def _scheduled(date_str: str, time_str: str, before: int, after: int, severity: str = "extrem") -> dict:
    return {
        "name": "Test",
        "cadence": "scheduled",
        "date": date_str,
        "time_ro": time_str,
        "severity": severity,
        "window_before_min": before,
        "window_after_min": after,
    }


class TestWindowBoundaries:
    def setup_method(self) -> None:
        self.cal = [_scheduled("2026-05-06", "15:30", 15, 15)]
        self.d = date(2026, 5, 6)

    def test_window_inside_boundary(self) -> None:
        assert is_in_news_window(self.d, time(15, 14), self.cal) is False  # 1 min before lower bound
        assert is_in_news_window(self.d, time(15, 15), self.cal) is True   # lower bound inclusive
        assert is_in_news_window(self.d, time(15, 45), self.cal) is True   # upper bound inclusive

    def test_window_outside(self) -> None:
        assert is_in_news_window(self.d, time(15, 14), self.cal) is False
        assert is_in_news_window(self.d, time(15, 46), self.cal) is False


def test_severity_filter_mediu_excluded() -> None:
    # JOLTS-like event with severity 'mediu' at 17:00 — even smack on time, no Set C trigger.
    cal = [_scheduled("2026-05-06", "17:00", 10, 10, severity="mediu")]
    assert is_in_news_window(date(2026, 5, 6), time(17, 0), cal) is False
    assert is_in_news_window(date(2026, 5, 6), time(17, 5), cal) is False


def test_fomc_powell_window() -> None:
    """Real FOMC Powell Press Apr from calendar_evenimente.yaml (2026-04-29 21:30 RO, 0/45)."""
    cal = load_calendar(CALENDAR_PATH)
    assert is_in_news_window(date(2026, 4, 29), time(21, 35), cal) is True
    assert is_in_news_window(date(2026, 4, 29), time(22, 16), cal) is False