fix(burst-location): normalize simulation ids

2026-07-08 17:17:41 +08:00
parent 4c0a4b29e9
commit d62bcae85e
3 changed files with 100 additions and 19 deletions
@@ -248,6 +248,54 @@ def test_run_burst_location_requires_simulation_scheme_name(monkeypatch, tmp_pat
        )


+def test_build_observed_series_from_simulation_normalizes_result_ids(monkeypatch):
+    module = _load_burst_location_module()
+    query_calls = []
+
+    monkeypatch.setattr(
+        module,
+        "get_all_scada_info",
+        lambda network: [
+            {
+                "type": "pressure",
+                "associated_element_id": " 100026 ",
+                "api_query_id": " pressure-query ",
+            }
+        ],
+    )
+
+    def fake_scheme_query(**kwargs):
+        query_calls.append(kwargs)
+        return {
+            100026: [
+                {"time": kwargs["start_time"], "value": 10.0},
+                {"time": kwargs["end_time"], "value": 14.0},
+            ]
+        }
+
+    monkeypatch.setattr(
+        module.InternalQueries,
+        "query_scheme_simulation_by_ids_timerange",
+        staticmethod(fake_scheme_query),
+    )
+
+    series, sample_count = module._build_observed_series_from_simulation(
+        network="tjwater",
+        sensor_ids=["100026"],
+        start_dt=datetime(2025, 1, 1, 0, 0, 0, tzinfo=timezone.utc),
+        end_dt=datetime(2025, 1, 1, 1, 0, 0, tzinfo=timezone.utc),
+        data_type="pressure",
+        series_name="burst_pressure",
+        simulation_source="scheme",
+        simulation_scheme_name="BurstSchemeA",
+        simulation_scheme_type="burst_analysis",
+    )
+
+    assert query_calls[0]["element_ids"] == ["100026"]
+    assert sample_count == 2
+    assert series["100026"] == pytest.approx(12.0)
+
+
 def test_run_burst_location_monitoring_uses_scada_for_burst_and_realtime_for_normal(
    monkeypatch, tmp_path
 ):