refactor(db)!: finalize pooled WNDB v2 migration

tests/unit/test_project_scada_metadata.py

@@ -33,12 +33,19 @@ def _patch_project_scadas(monkeypatch):
 
 def test_realtime_scada_simulation_uses_current_project_metadata(monkeypatch):
     _patch_project_scadas(monkeypatch)
-    query_mock = AsyncMock(return_value=[{"time": START_TIME, "value": 26.5}])
+    query_mock = AsyncMock(
+        return_value={"J1": [{"time": START_TIME, "value": 26.5}]}
+    )
     monkeypatch.setattr(
         composite_queries.RealtimeRepository,
-        "get_node_field_by_time_range",
+        "get_node_fields_by_ids_time_range",
         query_mock,
     )
+    monkeypatch.setattr(
+        composite_queries.RealtimeRepository,
+        "get_link_fields_by_ids_time_range",
+        AsyncMock(return_value={}),
+    )
 
     result = asyncio.run(
         composite_queries.CompositeQueries.get_scada_associated_realtime_simulation_data(
@@ -55,17 +62,22 @@ def test_realtime_scada_simulation_uses_current_project_metadata(monkeypatch):
     assert query_mock.await_args.args[1:] == (
         START_TIME,
         END_TIME,
-        "J1",
+        ["J1"],
         "pressure",
     )
 
 
 def test_analysis_scada_simulation_uses_current_project_metadata(monkeypatch):
     _patch_project_scadas(monkeypatch)
-    query_mock = AsyncMock(return_value=[{"time": START_TIME, "value": 26.5}])
+    async def query_series(_conn, _run_id, element_type, element_ids, *_args):
+        if element_type == "node":
+            return {"J1": [{"time": START_TIME, "value": 26.5}]}
+        return {}
+
+    query_mock = AsyncMock(side_effect=query_series)
     monkeypatch.setattr(
         composite_queries.AnalysisResultsRepository,
-        "get_node_series",
+        "get_series_by_ids",
         query_mock,
     )
 
@@ -81,7 +93,11 @@ def test_analysis_scada_simulation_uses_current_project_metadata(monkeypatch):
     )
 
     assert result[PROJECT_SCADA["device_id"]][0]["scada_id"] == PROJECT_SCADA["device_id"]
-    assert query_mock.await_args.args[2:] == ("J1", START_TIME, END_TIME, "pressure")
+    assert query_mock.await_count == 2
+    node_call = next(
+        call for call in query_mock.await_args_list if call.args[2] == "node"
+    )
+    assert node_call.args[3:] == (["J1"], START_TIME, END_TIME, "pressure")
 
 
 def test_element_scada_query_uses_current_project_metadata(monkeypatch):