from src.data_collection.amsc_awos_sources import (
    AmscAwosSourceMixin,
    _amsc_parse_wind_plate_payload,
    _amsc_supported_city_codes,
)


ZBAA_SAMPLE = {
    "code": 200,
    "msg": "操作成功",
    "data": {
        "18R/36L": {
            "RNO": "18R/36L",
            "OTIME": "2026-05-14 17:19:00",
            "TDZ_TEMP": "20.8",
            "MID_TEMP": "-",
            "END_TEMP": "20.8",
            "TDZ_HUMID": "67",
            "END_HUMID": "67",
            "METAR": "METAR ZBAA 141700Z 12003MPS 050V170 7000 NSC 21/15 Q1014 NOSIG=",
        },
        "18L/36R": {
            "RNO": "18L/36R",
            "OTIME": "2026-05-14 17:19:00",
            "TDZ_TEMP": "21.0",
            "MID_TEMP": "-",
            "END_TEMP": "20.4",
            "METAR": "METAR ZBAA 141700Z 12003MPS 050V170 7000 NSC 21/15 Q1014 NOSIG=",
        },
        "19/01": {
            "RNO": "19/01",
            "OTIME": "2026-05-14 17:19:00",
            "TDZ_TEMP": "20.2",
            "MID_TEMP": "-",
            "END_TEMP": "20.2",
            "METAR": "METAR ZBAA 141700Z 12003MPS 050V170 7000 NSC 21/15 Q1014 NOSIG=",
        },
    },
}


def test_parse_wind_plate_payload_normalizes_runway_point_temperatures():
    parsed = _amsc_parse_wind_plate_payload(ZBAA_SAMPLE, city_key="beijing", icao="ZBAA")

    assert parsed["source"] == "amsc_awos"
    assert parsed["source_label"] == "AMSC AWOS Beijing Capital (ZBAA)"
    assert parsed["observation_source_zh"] == "AMSC AWOS 跑道观测气温"
    assert parsed["icao"] == "ZBAA"
    assert parsed["temp_c"] == 21.0
    assert parsed["temp_source"] == "runway_max"
    assert parsed["runway_temp_range"] == (20.2, 21.0)
    assert parsed["observation_time"] == "2026-05-14T17:19:00+00:00"
    assert parsed["observation_time_local"] == "2026-05-15 01:19:00"
    assert parsed["raw_metar"].startswith("METAR ZBAA 141700Z")

    runway_obs = parsed["runway_obs"]
    assert runway_obs["runway_pairs"] == [("18R", "36L"), ("18L", "36R"), ("19", "01")]
    assert runway_obs["temperatures"] == [(20.8, None), (21.0, None), (20.2, None)]
    pt0 = runway_obs["point_temperatures"][0]
    assert pt0["runway"] == "18R/36L"
    assert pt0["tdz_temp"] == 20.8
    assert pt0["mid_temp"] is None
    assert pt0["end_temp"] == 20.8
    assert pt0["target_runway_max"] == 20.8
    assert pt0["wind_dir"] is None
    assert pt0["wind_speed"] is None
    assert pt0["rvr"] is None
    assert pt0["mor"] is None
    assert pt0["humidity"] == 67.0


def test_parse_wind_plate_payload_rejects_unauthorized_or_empty_payloads():
    assert _amsc_parse_wind_plate_payload(
        {"errCode": -12010, "errMsg": "无权访问此接口"},
        city_key="beijing",
        icao="ZBAA",
    ) is None
    assert _amsc_parse_wind_plate_payload({"code": 200, "data": {}}, city_key="beijing", icao="ZBAA") is None


def test_fetch_amsc_official_current_uses_domestic_city_whitelist(monkeypatch):
    import src.data_collection.amsc_awos_sources as _amsc
    monkeypatch.setattr(_amsc, "AMSC_AWOS_BASE_URL", "https://www.amsc.net.cn/gateway/api/saas/rest/amc/AwosController/getWindPlate")
    assert _amsc_supported_city_codes()["beijing"] == "ZBAA"
    assert "new york" not in _amsc_supported_city_codes()

    class FakeCollector(AmscAwosSourceMixin):
        timeout = 1.0

        def _amsc_http_get_json(self, url, *, headers=None):
            assert "getWindPlate?cccc=ZBAA" in url
            return ZBAA_SAMPLE

    data = FakeCollector().fetch_amsc_awos_current("beijing")

    assert data is not None
    assert data["icao"] == "ZBAA"
    assert data["runway_temp_range"] == (20.2, 21.0)
    assert FakeCollector().fetch_amsc_awos_current("new york") is None