保存 NAS 健康系统 UI7 代码与本机维护基线

tests/test_mi_fitness_cloud.py

@@ -1,4 +1,5 @@
 import pytest
+from datetime import datetime, timezone, timedelta
 
 from mi_fitness_mcp.adapters.mi_fitness_cloud import (
     MiFitnessCloudAdapter,
@@ -34,39 +35,46 @@ def test_record_datetime_uses_zone_offset():
 
 
 @pytest.mark.asyncio
-async def test_iter_daily_activity_aggregates_steps_and_calories(monkeypatch):
-    adapter = MiFitnessCloudAdapter(user_id="u1", pass_token="p1")
+async def test_iter_daily_activity_uses_latest_cloud_daily_report(monkeypatch):
+    adapter = MiFitnessCloudAdapter(user_id="u1", pass_token="p1", region="cn")
     adapter._connected = True
     adapter._client = object()
 
-    async def fake_fetch(key, start_date, end_date, region=None):
-        if key == "steps":
-            return [
-                {
-                    "time": 1743467400,
-                    "zone_offset": 0,
-                    "value": '{"steps": 10, "distance": 8, "calories": 1}',
-                },
-                {
-                    "time": 1743467460,
-                    "zone_offset": 0,
-                    "value": '{"steps": 20, "distance": 16, "calories": 2}',
-                },
-            ]
-        if key == "calories":
-            return [
-                {"time": 1743467400, "zone_offset": 0, "value": '{"calories": 5}'},
-                {"time": 1743467460, "zone_offset": 0, "value": '{"calories": 7}'},
-            ]
-        return []
+    async def fake_request(base_url, path, payload):
+        # CN 午夜对应 UTC 前一天，日报没有 zone_offset；重复日报取最新版本。
+        if path.endswith("get_aggregated_fitness_data_by_time"):
+            assert payload["tag"] == "daily_report"
+            values = (
+                ['{"steps": 10, "distance": 8}', '{"steps": 20, "distance": 16}']
+                if payload["key"] == "steps" else ['{"calories": 7}']
+            )
+            return {"data_list": [{"time": 1743436800, "update_time": i + 1, "value": v} for i, v in enumerate(values)], "has_more": False}
+        # 旧实现会误把多来源明细相加。
+        return {"data_list": [{"time": 1743467400, "value": '{"steps": 99, "distance": 99, "calories": 99}'}], "has_more": False}
 
-    monkeypatch.setattr(adapter, "_fetch_key", fake_fetch)
+    monkeypatch.setattr(adapter, "_request", fake_request)
     items = await _collect(adapter.iter_daily_activity("2025-04-01", "2025-04-01"))
 
     assert len(items) == 1
-    assert items[0].steps == 30
-    assert items[0].distance_m == 24
-    assert items[0].active_kcal == 12
+    assert items[0].date == "2025-04-01"
+    assert items[0].steps == 20
+    assert items[0].distance_m == 16
+    assert items[0].active_kcal == 7
+
+
+@pytest.mark.asyncio
+async def test_daily_activity_rejects_missing_calorie_report(monkeypatch):
+    adapter = MiFitnessCloudAdapter(user_id="u1", pass_token="p1", region="cn")
+    adapter._connected = True
+    adapter._client = object()
+
+    async def fake_request(base_url, path, payload):
+        items = [{"time": 1743436800, "value": '{"steps": 20, "distance": 16}'}] if payload["key"] == "steps" else []
+        return {"data_list": items, "has_more": False}
+
+    monkeypatch.setattr(adapter, "_request", fake_request)
+    with pytest.raises(ValueError, match="Incomplete daily report"):
+        await _collect(adapter.iter_daily_activity("2025-04-01", "2025-04-01"))
 
 
 @pytest.mark.asyncio
@@ -100,3 +108,43 @@ def test_authentication_error_detection():
     assert _is_authentication_error(0, "session expired")
     assert _is_authentication_error(-10001, "unknown")
     assert not _is_authentication_error(500, "temporary server failure")
+
+
+@pytest.mark.asyncio
+async def test_sleep_daily_report_does_not_subtract_awake_twice(monkeypatch):
+    adapter = MiFitnessCloudAdapter(user_id="u1", pass_token="p1", region="cn")
+    adapter._connected = True
+    adapter._client = object()
+    cn = timezone(timedelta(hours=8))
+    bedtime = int(datetime(2025, 4, 1, 23, 40, tzinfo=cn).timestamp())
+    wake = bedtime + 40 * 60
+    segment = {"bedtime": bedtime, "wake_up_time": wake, "duration": 30, "sleep_awake_duration": 10, "sleep_deep_duration": 10, "sleep_light_duration": 20}
+
+    async def fake_request(base_url, path, payload):
+        if path.endswith("get_aggregated_fitness_data_by_time"):
+            return {"data_list": [{"time": wake, "update_time": wake, "value": {"total_duration": 30, "sleep_score": 84, "segment_details": [segment]}}], "has_more": False}
+        return {"data_list": [{"time": wake, "zone_offset": 28800, "value": segment}], "has_more": False}
+
+    monkeypatch.setattr(adapter, "_request", fake_request)
+    sessions = await _collect(adapter.iter_sleep_sessions("2025-04-02", "2025-04-02"))
+    assert len(sessions) == 1
+    assert sessions[0].duration_minutes == 30
+    assert sessions[0].time_asleep_minutes == 30
+    assert sessions[0].time_awake_minutes == 10
+    assert sessions[0].sleep_score == 84
+
+
+def test_sleep_queries_use_wake_date_and_latest_canonical_version(tmp_path):
+    from mi_fitness_mcp.models import SleepSession
+    from mi_fitness_mcp.storage import Database
+
+    cn = timezone(timedelta(hours=8))
+    db = Database(tmp_path / "test.sqlite")
+    start = datetime(2025, 4, 1, 23, 40, tzinfo=cn)
+    end = start + timedelta(minutes=40)
+    for name, source, collected in [("raw", "raw:1", end), ("old", "daily_report:2025-04-02", end), ("new", "daily_report:2025-04-02", end + timedelta(minutes=5))]:
+        db.insert_sleep_session(SleepSession(id=name, sleep_id=name, provider="mi_fitness", source_type="cloud_session", source_record_id=source, user_id="u1", timezone="Asia/Shanghai", collected_at=collected, start_at=start, end_at=end, duration_minutes=30, time_asleep_minutes=30, time_awake_minutes=10))
+    sessions = db.query_sleep_sessions("u1", "2025-04-02", "2025-04-02")
+    assert [row["id"] for row in sessions] == ["new"]
+    with db._get_connection() as conn:
+        assert conn.execute("SELECT COUNT(*) FROM sleep_sessions").fetchone()[0] == 3