smart tdee

backend/tests/test_openmeteo_helpers.py

@@ -1,13 +1,17 @@
 from unittest.mock import patch
 
 from app.homelab.openmeteo import (
+    PRECIP_PROB_HINT,
     RECOMMENDED_SYNC_DOMAINS,
     RECOMMENDED_SYNC_VARIABLES,
     SYNC_HINT,
     _coverage_sufficient,
     _field_coverage,
     _hourly_start_index,
+    _local_needs_sync_hint,
     build_weather_dashboard,
+    format_weather_snapshot,
+    weather_query_relevant,
 )
 
 
@@ -21,49 +25,89 @@ def test_coverage_sufficient():
     assert _coverage_sufficient(
         {
             "current": ["temperature_2m", "weather_code", "wind_speed_10m"],
-            "hourly": ["temperature_2m", "precipitation_probability", "weather_code"],
+            "hourly": ["temperature_2m", "weather_code"],
         }
     ) is True
 
 
 def test_field_coverage_partial():
     raw = {
-        "current": {"time": "2026-06-14T18:15", "temperature_2m": 20.6},
+        "current": {"time": "2026-06-14T18:15", "temperature_2m": 20.6, "weather_code": 2},
         "hourly": {
             "time": ["2026-06-14T18:00", "2026-06-14T19:00"],
             "temperature_2m": [20.0, 19.5],
             "precipitation": [0.0, 0.0],
+            "weather_code": [2, 3],
+        },
+        "daily": {
+            "time": ["2026-06-14", "2026-06-15"],
+            "temperature_2m_max": [21.0, 18.0],
+            "temperature_2m_min": [12.0, 10.0],
+            "weather_code": [2, 3],
         },
     }
     coverage = _field_coverage(raw)
-    assert coverage["current"] == ["temperature_2m"]
-    assert "temperature_2m" in coverage["hourly"]
-    assert "precipitation" in coverage["hourly"]
-    assert "weather_code" not in coverage["hourly"]
+    assert "temperature_2m" in coverage["current"]
+    assert "weather_code" in coverage["hourly"]
+    assert "temperature_2m_max" in coverage["daily"]
 
 
-def test_build_weather_dashboard_includes_sync_hint():
+def test_local_needs_sync_hint():
+    assert _local_needs_sync_hint({"current": ["temperature_2m"], "hourly": ["temperature_2m"]}) is True
+    assert _local_needs_sync_hint(
+        {"current": ["temperature_2m", "weather_code"], "hourly": ["temperature_2m", "weather_code"]}
+    ) is False
+
+
+def test_weather_query_relevant():
+    assert weather_query_relevant("какая погода завтра")
+    assert not weather_query_relevant("напиши код на python")
+
+
+def test_format_weather_snapshot_includes_tomorrow():
+    snap = {
+        "ok": True,
+        "location": "СПб",
+        "current": {"temperature_c": 20, "conditions": "ясно"},
+        "hourly": [],
+        "daily": [
+            {"label": "Сегодня", "temperature_min_c": 10, "temperature_max_c": 20, "conditions": "ясно"},
+            {"label": "Завтра", "temperature_min_c": 12, "temperature_max_c": 18, "conditions": "дождь"},
+        ],
+    }
+    text = format_weather_snapshot(snap)
+    assert "Завтра:" in text
+    assert "None" not in text
+
+
+def test_build_weather_dashboard_includes_daily():
     fake_weather = {
         "ok": True,
         "location": "Test",
         "data_source": "local",
-        "local_field_coverage": {"current": ["temperature_2m"], "hourly": ["temperature_2m"]},
-        "field_coverage": {"current": ["temperature_2m"], "hourly": ["temperature_2m"]},
-        "sync_hint": SYNC_HINT,
-        "current": {"temperature_c": 10, "conditions": "неизвестно"},
+        "local_field_coverage": {"current": ["temperature_2m", "weather_code"], "hourly": ["temperature_2m"], "daily": []},
+        "field_coverage": {"current": ["temperature_2m"], "hourly": ["temperature_2m"], "daily": []},
+        "sync_hint": PRECIP_PROB_HINT,
+        "merged_fields": [],
+        "current": {"temperature_c": 10, "conditions": "ясно"},
         "hourly": [],
+        "daily": [{"label": "Завтра", "temperature_min_c": 5, "temperature_max_c": 12, "conditions": "дождь"}],
     }
     with patch("app.homelab.openmeteo.OpenMeteoClient") as mock_cls:
         client = mock_cls.return_value
-        client.fetch_current_and_hourly.return_value = fake_weather
+        client.fetch_forecast.return_value = fake_weather
         client.rain_summary.return_value = "ok"
+        client.daily_summary.return_value = "Завтра: 5–12°C"
         client.cache_status.return_value = {"source": "local", "has_data": True, "cached": True, "ttl_sec": 300}
         client.location_name = "Test"
         client.lat = 1.0
         client.lon = 2.0
         client.base_url = "http://local"
         client.cache_ttl = 300
-        result = build_weather_dashboard()
-    assert result["sync_hint"]
+        client.forecast_days = 7
+        result = build_weather_dashboard(days_ahead=7)
+
+    assert result["daily_summary"] == "Завтра: 5–12°C"
     assert result["recommended_sync"]["domains"] == RECOMMENDED_SYNC_DOMAINS
     assert result["recommended_sync"]["variables"] == RECOMMENDED_SYNC_VARIABLES
+    assert SYNC_HINT  # constant exists