from src.data_collection.russia_station_sources import (
    RUSSIA_MOSCOW_MAP_STATION_LIMIT,
    RUSSIA_MOSCOW_STATIONS,
    RussiaStationSourceMixin,
)


def test_moscow_station_registry_includes_vnukovo_nearby_ring():
    assert "27524" in RUSSIA_MOSCOW_STATIONS
    assert "27500" in RUSSIA_MOSCOW_STATIONS
    assert RUSSIA_MOSCOW_STATIONS["27500"]["station_label"] == "Tolstopaltsevo"
    assert RUSSIA_MOSCOW_MAP_STATION_LIMIT == 10


def test_fetch_moscow_official_nearby_keeps_nearest_ten_station_rows():
    class FakeRussiaSource(RussiaStationSourceMixin):
        CITY_REGISTRY = {
            "moscow": {
                "lat": 55.5915,
                "lon": 37.2615,
            }
        }

        def _ru_cached_station_current(self, station_code, station_meta, use_fahrenheit=False):
            return {
                "station_code": station_code,
                "station_label": station_meta["station_label"],
                "name": station_meta["station_label"],
                "lat": station_meta["lat"],
                "lon": station_meta["lon"],
                "temp": 10.0,
                "source_code": "ru_station_web",
            }

    rows = FakeRussiaSource().fetch_russia_moscow_official_nearby("moscow")

    assert len(rows) == RUSSIA_MOSCOW_MAP_STATION_LIMIT
    assert rows[0]["station_code"] == "27524"
    assert rows[1]["station_code"] == "27500"
    assert all(row["distance_km"] is not None for row in rows)


def test_russia_weather_archive_parser_uses_latest_valid_row():
    html = """
    <table>
      <tr><th>hour</th><th>date</th></tr>
      <tr><td>09</td><td>11.04</td></tr>
      <tr><td>21</td><td>17.04</td></tr>
    </table>
    <table>
      <tr><th>a</th><th>b</th><th>c</th><th>d</th><th>cloud</th><th>temp</th></tr>
      <tr><td>x</td><td>x</td><td>x</td><td>x</td><td>x</td><td>+4.2</td></tr>
      <tr><td>x</td><td>x</td><td>x</td><td>x</td><td>x</td><td>+5.8</td></tr>
    </table>
    """

    parsed = RussiaStationSourceMixin()._ru_parse_station_current_from_weather_html(html)

    assert parsed["temp_c"] == 5.8
    assert parsed["raw_hour"] == "21"
    assert parsed["raw_day_month"] == "17.04"