jiang
9b095c7439
- make realtime replacement and analysis result writes transactional\n- consolidate SCADA repositories and remove process-global project state\n- validate SCADA batches and use indexed GIS-backed business queries\n\nBREAKING CHANGE: remove the public analysis result writer and the pipeline-health network_name query parameter.
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import asyncio
|
|
from datetime import datetime, timezone
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from fastapi import HTTPException
|
|
|
|
from app.api.v1.endpoints.timeseries import scada as scada_endpoint
|
|
|
|
|
|
def _reading(device_id: str) -> scada_endpoint.ScadaReadingBatchItem:
|
|
return scada_endpoint.ScadaReadingBatchItem(
|
|
time=datetime(2026, 8, 27, tzinfo=timezone.utc),
|
|
device_id=device_id,
|
|
monitored_value=12.5,
|
|
)
|
|
|
|
|
|
def test_scada_batch_rejects_devices_missing_from_bizdb(monkeypatch):
|
|
monkeypatch.setattr(
|
|
scada_endpoint.ScadaInfoRepository,
|
|
"get_existing_device_ids",
|
|
AsyncMock(return_value={"known"}),
|
|
)
|
|
insert = AsyncMock()
|
|
monkeypatch.setattr(scada_endpoint.ScadaRepository, "insert_scada_batch", insert)
|
|
|
|
with pytest.raises(HTTPException, match="missing") as exc_info:
|
|
asyncio.run(
|
|
scada_endpoint.insert_scada_data(
|
|
[_reading("known"), _reading("missing")],
|
|
conn=object(),
|
|
postgres_conn=object(),
|
|
)
|
|
)
|
|
|
|
assert exc_info.value.status_code == 422
|
|
insert.assert_not_awaited()
|
|
|
|
|
|
def test_scada_batch_writes_only_after_bizdb_validation(monkeypatch):
|
|
monkeypatch.setattr(
|
|
scada_endpoint.ScadaInfoRepository,
|
|
"get_existing_device_ids",
|
|
AsyncMock(return_value={"known"}),
|
|
)
|
|
insert = AsyncMock()
|
|
monkeypatch.setattr(scada_endpoint.ScadaRepository, "insert_scada_batch", insert)
|
|
|
|
result = asyncio.run(
|
|
scada_endpoint.insert_scada_data(
|
|
[_reading(" known ")],
|
|
conn=object(),
|
|
postgres_conn=object(),
|
|
)
|
|
)
|
|
|
|
assert result == {"message": "Inserted 1 records"}
|
|
assert insert.await_args.args[1][0]["device_id"] == "known"
|