import pytest

from app.core.contracts import AgentDefinition, CoreResult, RequestContext
from app.core.errors import RecoverableAgentError
from app.service.agent.base import BaseAgent
from app.service.agent.factory import AgentFactory
from app.service.model_gateway import ModelDispatchService, ModelGenerationService


class DemoAgent(BaseAgent):
    async def handle(self, request, context):
        return CoreResult(text="ok")


class Gateway:
    async def generate(self, *, endpoint_code, prompt, timeout_ms):
        return "model answer"


@pytest.mark.asyncio
async def test_factory_injects_model_generation_entrypoint(governance) -> None:
    definition = AgentDefinition(agent_type="demo", version="1", allowed_roles=("customer",),
                                 allowed_portals=("api",))
    factory = AgentFactory(governance, ModelGenerationService(ModelDispatchService(Gateway())))
    factory.register(definition, lambda _: DemoAgent(definition))
    context = RequestContext(user_id="1", trace_id="t", roles=("customer",),
                             permissions=("agent:run",))
    agent = factory.create("demo", context)
    endpoint = type("Endpoint", (), {"endpoint_code": "primary", "timeout_ms": 1000})()
    result = await agent.generate_with_model([endpoint], "classify this")
    assert result.text == "model answer"


@pytest.mark.asyncio
async def test_agent_without_model_dependency_fails_closed() -> None:
    definition = AgentDefinition(agent_type="demo", version="1", allowed_roles=("customer",),
                                 allowed_portals=("api",))
    agent = DemoAgent(definition)
    with pytest.raises(RecoverableAgentError, match="模型服务"):
        await agent.generate_with_model([], "classify this")