jiang
fa188af0b1
Reorganize WNDB by responsibility and remove legacy scheme endpoints.\n\nRoute analysis and time-series access through project pools, preserve transactional realtime replacement, and refresh GIS materialized views after writes.\n\nAdd database architecture documentation, live pooling coverage, API contract updates, and executable container verification.\n\nBREAKING CHANGE: legacy scheme APIs and flat app.native.wndb module imports are removed.
123 lines
3.5 KiB
Python
123 lines
3.5 KiB
Python
import asyncio
|
|
from datetime import datetime, timezone
|
|
from unittest.mock import AsyncMock
|
|
from uuid import uuid4
|
|
|
|
from app.api.v1.endpoints import project_data
|
|
from app.infra.db.timescaledb import composite_queries
|
|
|
|
|
|
PROJECT_SCADA = {
|
|
"device_id": "fengyang-pressure-1",
|
|
"device_type": "pressure",
|
|
"node_id": "J1",
|
|
"link_id": None,
|
|
"api_query_id": "query-1",
|
|
"transmission_mode": "realtime",
|
|
"transmission_frequency": None,
|
|
"reliability": 1.0,
|
|
"x": 117.1,
|
|
"y": 32.9,
|
|
}
|
|
START_TIME = datetime(2026, 6, 1, tzinfo=timezone.utc)
|
|
END_TIME = datetime(2026, 6, 2, tzinfo=timezone.utc)
|
|
|
|
|
|
def _patch_project_scadas(monkeypatch):
|
|
monkeypatch.setattr(
|
|
composite_queries.ScadaInfoRepository,
|
|
"get_scadas",
|
|
AsyncMock(return_value=[PROJECT_SCADA.copy()]),
|
|
)
|
|
|
|
|
|
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}])
|
|
monkeypatch.setattr(
|
|
composite_queries.RealtimeRepository,
|
|
"get_node_field_by_time_range",
|
|
query_mock,
|
|
)
|
|
|
|
result = asyncio.run(
|
|
composite_queries.CompositeQueries.get_scada_associated_realtime_simulation_data(
|
|
object(),
|
|
object(),
|
|
[PROJECT_SCADA["device_id"]],
|
|
START_TIME,
|
|
END_TIME,
|
|
)
|
|
)
|
|
|
|
assert result[PROJECT_SCADA["device_id"]][0]["scada_id"] == PROJECT_SCADA["device_id"]
|
|
assert query_mock.await_count == 1
|
|
assert query_mock.await_args.args[1:] == (
|
|
START_TIME,
|
|
END_TIME,
|
|
"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}])
|
|
monkeypatch.setattr(
|
|
composite_queries.AnalysisResultsRepository,
|
|
"get_node_series",
|
|
query_mock,
|
|
)
|
|
|
|
result = asyncio.run(
|
|
composite_queries.CompositeQueries.get_scada_associated_analysis_simulation_data(
|
|
object(),
|
|
object(),
|
|
[PROJECT_SCADA["device_id"]],
|
|
START_TIME,
|
|
END_TIME,
|
|
uuid4(),
|
|
)
|
|
)
|
|
|
|
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")
|
|
|
|
|
|
def test_element_scada_query_uses_current_project_metadata(monkeypatch):
|
|
_patch_project_scadas(monkeypatch)
|
|
query_mock = AsyncMock(
|
|
return_value={
|
|
PROJECT_SCADA["device_id"]: [{"time": START_TIME, "value": 26.5}]
|
|
}
|
|
)
|
|
monkeypatch.setattr(
|
|
composite_queries.ScadaRepository,
|
|
"get_scada_field_by_id_time_range",
|
|
query_mock,
|
|
)
|
|
|
|
result = asyncio.run(
|
|
composite_queries.CompositeQueries.get_element_associated_scada_data(
|
|
object(),
|
|
object(),
|
|
"J1",
|
|
START_TIME,
|
|
END_TIME,
|
|
)
|
|
)
|
|
|
|
assert result == {"J1": [{"time": START_TIME, "value": 26.5}]}
|
|
|
|
|
|
def test_scada_info_endpoint_uses_current_project_connection(monkeypatch):
|
|
monkeypatch.setattr(
|
|
project_data.ScadaInfoRepository,
|
|
"get_scadas",
|
|
AsyncMock(return_value=[PROJECT_SCADA.copy()]),
|
|
)
|
|
|
|
result = asyncio.run(project_data.get_scada_info_with_connection(object()))
|
|
|
|
assert result == {"success": True, "data": [PROJECT_SCADA], "count": 1}
|