63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
import asyncio
|
|
|
|
from app.infra.db.postgresql.scada import ScadaInfoRepository
|
|
|
|
|
|
class _FakeCursor:
|
|
def __init__(self):
|
|
self.query = None
|
|
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
return False
|
|
|
|
async def execute(self, query):
|
|
self.query = query
|
|
|
|
async def fetchall(self):
|
|
return [
|
|
{
|
|
"id": " 25470001 ",
|
|
"type": " PRESSURE ",
|
|
"associated_element_id": " J1 ",
|
|
"api_query_id": "query-1",
|
|
"transmission_mode": "realtime",
|
|
"transmission_frequency": None,
|
|
"reliability": "0.95",
|
|
"x_coor": "117.1",
|
|
"y_coor": "32.9",
|
|
}
|
|
]
|
|
|
|
|
|
class _FakeConnection:
|
|
def __init__(self):
|
|
self.cursor_instance = _FakeCursor()
|
|
|
|
def cursor(self):
|
|
return self.cursor_instance
|
|
|
|
|
|
def test_get_scadas_normalizes_id_and_type():
|
|
conn = _FakeConnection()
|
|
|
|
result = asyncio.run(ScadaInfoRepository.get_scadas(conn))
|
|
|
|
assert result == [
|
|
{
|
|
"id": "25470001",
|
|
"type": "pressure",
|
|
"associated_element_id": "J1",
|
|
"api_query_id": "query-1",
|
|
"transmission_mode": "realtime",
|
|
"transmission_frequency": None,
|
|
"reliability": 0.95,
|
|
"x": 117.1,
|
|
"y": 32.9,
|
|
}
|
|
]
|
|
assert "associated_element_id" in conn.cursor_instance.query
|
|
assert "FROM public.scada_info" in conn.cursor_instance.query
|