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"