from unittest.mock import AsyncMock

import pytest

from app.service.relationship_service import GraphDegradedResult, RelationshipService


class FailingGraph:
    async def execute_query(self, query: str, **parameters: object) -> list[object]:
        raise TimeoutError("neo4j timeout")


@pytest.mark.asyncio
async def test_neo4j_timeout_returns_explicit_degraded_result() -> None:
    result = await RelationshipService(FailingGraph()).neighbors(1, "HOLDS")
    assert len(result) == 1
    assert isinstance(result[0], GraphDegradedResult)
    assert result[0].degraded is True


@pytest.mark.asyncio
async def test_neo4j_rejects_unapproved_relationship() -> None:
    with pytest.raises(ValueError, match="not allowed"):
        await RelationshipService(FailingGraph()).neighbors(1, "DELETE_ALL")


@pytest.mark.asyncio
async def test_model_router_excludes_unhealthy_endpoint() -> None:
    from app.model.configuration import ModelEndpointConfig
    from app.service.model_router_service import ModelRouterService

    primary = ModelEndpointConfig(
        id=1,
        endpoint_code="primary",
        provider="x",
        model_name="m1",
        base_url="http://x",
        secret_ref="secret://one",
        capabilities=["chat"],
        status="active",
        timeout_ms=1000,
    )
    fallback = ModelEndpointConfig(
        id=2,
        endpoint_code="fallback",
        provider="x",
        model_name="m2",
        base_url="http://y",
        secret_ref="secret://two",
        capabilities=["chat"],
        status="active",
        timeout_ms=1000,
    )
    session = AsyncMock()
    session.scalars.return_value = [primary, fallback]
    router = ModelRouterService(session)
    router.mark_health("primary", False)
    selected = await router.select_with_fallback(
        agent_type="customer_service", task_type="chat", fallback_codes=["primary", "fallback"]
    )
    assert [endpoint.endpoint_code for endpoint in selected] == ["fallback"]