2569718930@qq.com
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
from web.services.city_payloads import build_city_detail_payload, build_city_summary_payload
|
|
from web.services import city_runtime
|
|
|
|
|
|
def test_city_summary_payload_preserves_deb_metadata():
|
|
payload = build_city_summary_payload(
|
|
{
|
|
"name": "shanghai",
|
|
"display_name": "Shanghai",
|
|
"temp_symbol": "°C",
|
|
"current": {},
|
|
"risk": {},
|
|
"deb": {
|
|
"prediction": 29.7,
|
|
"raw_prediction": 27.6,
|
|
"version": "deb_v1_recent_bias_corrected",
|
|
"bias_adjustment": 1.3,
|
|
"bias_samples": 18,
|
|
},
|
|
}
|
|
)
|
|
|
|
assert payload["deb"] == {
|
|
"prediction": 29.7,
|
|
"raw_prediction": 27.6,
|
|
"version": "deb_v1_recent_bias_corrected",
|
|
"bias_adjustment": 1.3,
|
|
"bias_samples": 18,
|
|
}
|
|
|
|
|
|
def test_city_payloads_expose_wunderground_current():
|
|
data = {
|
|
"name": "shanghai",
|
|
"display_name": "Shanghai",
|
|
"temp_symbol": "°C",
|
|
"current": {},
|
|
"risk": {},
|
|
"wunderground_current": {
|
|
"source": "wunderground_historical",
|
|
"station_code": "ZSPD",
|
|
"max_so_far": 26,
|
|
"today_obs": [{"time": "13:30", "temp": 26}],
|
|
},
|
|
}
|
|
|
|
summary = build_city_summary_payload(data)
|
|
detail = build_city_detail_payload(data)
|
|
|
|
assert summary["wunderground_current"]["max_so_far"] == 26
|
|
assert detail["wunderground_current"]["max_so_far"] == 26
|
|
assert detail["official"]["wunderground_current"]["station_code"] == "ZSPD"
|
|
assert detail["timeseries"]["wunderground_today_obs"] == [{"time": "13:30", "temp": 26}]
|
|
|
|
|
|
def test_api_payload_overlay_strips_cached_wunderground_state_without_fetch(monkeypatch):
|
|
stale_payload = {
|
|
"name": "guangzhou",
|
|
"temp_symbol": "°C",
|
|
"utc_offset_seconds": 28800,
|
|
"wunderground_current": {
|
|
"source": "wunderground_historical",
|
|
"station_code": "ZGGG",
|
|
"temp": 36,
|
|
"max_so_far": 36,
|
|
"today_obs": [{"time": "14:00", "temp": 36}],
|
|
},
|
|
"official": {
|
|
"wunderground_current": {
|
|
"source": "wunderground_historical",
|
|
"station_code": "ZGGG",
|
|
"temp": 36,
|
|
"max_so_far": 36,
|
|
},
|
|
},
|
|
"timeseries": {
|
|
"wunderground_today_obs": [{"time": "14:00", "temp": 36}],
|
|
},
|
|
}
|
|
|
|
fetch_calls = []
|
|
|
|
def fail_fetch(*args, **kwargs):
|
|
fetch_calls.append((args, kwargs))
|
|
raise AssertionError("city API overlay must not fetch Wunderground directly")
|
|
|
|
monkeypatch.setattr(city_runtime._weather, "fetch_wunderground_historical", fail_fetch)
|
|
|
|
overlay = getattr(city_runtime, "_overlay_latest_wunderground_current", None)
|
|
assert callable(overlay), "city API must strip cached WU state without direct fetch"
|
|
payload = overlay("guangzhou", stale_payload)
|
|
|
|
assert "wunderground_current" not in payload
|
|
assert "wunderground_current" not in payload["official"]
|
|
assert "wunderground_today_obs" not in payload["timeseries"]
|
|
assert stale_payload["wunderground_current"]["max_so_far"] == 36
|
|
assert fetch_calls == []
|