tracksolid_timescale_grafana_prod/tests/integration/test_webhook_endpoints.py

"""Integration tests for FastAPI webhook endpoints."""
import sys
import os
import json
import pytest
from unittest.mock import MagicMock, patch, call
from contextlib import contextmanager

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(__file__))))

os.environ.setdefault("TRACKSOLID_APP_KEY", "test_key")
os.environ.setdefault("TRACKSOLID_APP_SECRET", "test_secret")
os.environ.setdefault("TRACKSOLID_USER_ID", "test_user")
os.environ.setdefault("TRACKSOLID_PWD_MD5", "test_md5")
os.environ.setdefault("DATABASE_URL", "postgresql://test:test@localhost:5432/test")
os.environ.setdefault("JIMI_WEBHOOK_TOKEN", "")

from fastapi.testclient import TestClient
import webhook_receiver_rev
from tests.fixtures.api_responses import (
    WEBHOOK_ALARM_PAYLOAD,
    WEBHOOK_ALARM_NULL_TYPE,
    WEBHOOK_TRIP_BCD_PAYLOAD,
    WEBHOOK_TRIP_ISO_PAYLOAD,
    WEBHOOK_OBD_PAYLOAD,
)


def make_mock_conn():
    """Create a mock DB connection with cursor support."""
    mock_cur = MagicMock()
    mock_conn = MagicMock()
    mock_conn.cursor.return_value.__enter__ = lambda s: mock_cur
    mock_conn.cursor.return_value.__exit__ = MagicMock(return_value=False)
    return mock_conn, mock_cur


@contextmanager
def mock_get_conn_ctx(mock_conn):
    yield mock_conn


@pytest.fixture
def client():
    return TestClient(webhook_receiver_rev.app, raise_server_exceptions=True)


@pytest.fixture
def mock_db():
    mock_conn, mock_cur = make_mock_conn()
    with patch("webhook_receiver_rev.get_conn") as mock_get_conn:
        mock_get_conn.return_value = mock_get_conn_ctx(mock_conn)
        yield mock_conn, mock_cur


class TestHealth:
    def test_health_returns_ok(self, client):
        response = client.get("/health")
        assert response.status_code == 200
        assert response.json() == {"status": "ok"}


class TestPushAlarm:
    def test_valid_alarm_accepted(self, client, mock_db):
        mock_conn, mock_cur = mock_db
        data_list = json.dumps([WEBHOOK_ALARM_PAYLOAD])
        response = client.post("/pushalarm", data={"token": "", "data_list": data_list})
        assert response.status_code == 200
        assert response.json()["code"] == 0

    def test_null_alarm_type_skipped(self, client, mock_db):
        """BUG-02 guard: NULL alarm_type must be rejected, not inserted."""
        mock_conn, mock_cur = mock_db
        data_list = json.dumps([WEBHOOK_ALARM_NULL_TYPE])
        response = client.post("/pushalarm", data={"token": "", "data_list": data_list})
        assert response.status_code == 200
        # Verify no INSERT was executed (only SAVEPOINT + RELEASE calls)
        insert_calls = [c for c in mock_cur.execute.call_args_list
                       if "INSERT" in str(c)]
        assert len(insert_calls) == 0, "NULL alarm_type must not be inserted"

    def test_empty_data_list_ok(self, client):
        response = client.post("/pushalarm", data={"token": "", "data_list": ""})
        assert response.status_code == 200

    def test_batch_with_bad_item_processes_rest(self, client, mock_db):
        """BUG-04: One bad item must not abort the entire batch."""
        mock_conn, mock_cur = mock_db
        # One valid, one missing alarm_type (will be skipped, not crash)
        items = [WEBHOOK_ALARM_PAYLOAD, WEBHOOK_ALARM_NULL_TYPE]
        data_list = json.dumps(items)
        response = client.post("/pushalarm", data={"token": "", "data_list": data_list})
        assert response.status_code == 200
        assert response.json()["code"] == 0


class TestPushTripReport:
    def test_bcd_timestamp_parsed(self, client, mock_db):
        """BUG-03: BCD timestamp 220415103000 must be parsed correctly."""
        mock_conn, mock_cur = mock_db
        data_list = json.dumps([WEBHOOK_TRIP_BCD_PAYLOAD])
        response = client.post("/pushtripreport", data={"token": "", "data_list": data_list})
        assert response.status_code == 200
        assert response.json()["code"] == 0
        # Verify an INSERT was attempted
        insert_calls = [c for c in mock_cur.execute.call_args_list
                       if "INSERT" in str(c)]
        assert len(insert_calls) > 0, "Trip with BCD timestamp must trigger INSERT"

    def test_iso_timestamp_accepted(self, client, mock_db):
        mock_conn, mock_cur = mock_db
        data_list = json.dumps([WEBHOOK_TRIP_ISO_PAYLOAD])
        response = client.post("/pushtripreport", data={"token": "", "data_list": data_list})
        assert response.status_code == 200

    def test_missing_imei_skipped(self, client, mock_db):
        mock_conn, mock_cur = mock_db
        bad_trip = {"beginTime": "2024-04-12 07:00:00", "miles": 10.0}
        data_list = json.dumps([bad_trip])
        response = client.post("/pushtripreport", data={"token": "", "data_list": data_list})
        assert response.status_code == 200


class TestPushObd:
    def test_valid_obd_accepted(self, client, mock_db):
        mock_conn, mock_cur = mock_db
        data_list = json.dumps([WEBHOOK_OBD_PAYLOAD])
        response = client.post("/pushobd", data={"token": "", "data_list": data_list})
        assert response.status_code == 200
        assert response.json()["code"] == 0