"""Tests for POST /agent/run — uses a stub orchestrator factory."""
from __future__ import annotations

from typing import Any
from unittest.mock import patch

import pytest
from fastapi.testclient import TestClient

from src.agents.schemas import AgentResult, ToolTraceItem
from src.api.main import app


client = TestClient(app)


class _FakeOrchestrator:
    """Returns a canned AgentResult; ignores input."""
    def __init__(self, *args: Any, **kwargs: Any) -> None:
        pass

    def run(self, user_input: str, context: dict[str, Any] | None = None) -> AgentResult:
        return AgentResult(
            text=f"Synthesized answer for: {user_input}",
            trace=[
                ToolTraceItem(name="run_bbb_pipeline", args={"smiles": user_input},
                              result={"label": 1, "label_text": "permeable"}),
                ToolTraceItem(name="retrieve_context", args={"query": "BBB"},
                              result={"chunks": []}),
            ],
            model="stub-model",
            finish_reason="complete",
        )


class TestAgentRoute:
    def test_post_returns_synthesized_text_and_trace(self) -> None:
        with patch("src.api.routes._build_orchestrator", return_value=_FakeOrchestrator()):
            r = client.post("/agent/run", json={"user_input": "CCO"})
        assert r.status_code == 200
        body = r.json()
        assert "Synthesized answer for: CCO" in body["text"]
        assert len(body["trace"]) == 2
        assert body["trace"][0]["name"] == "run_bbb_pipeline"
        assert body["model"] == "stub-model"
        assert body["finish_reason"] == "complete"

    def test_empty_user_input_422(self) -> None:
        r = client.post("/agent/run", json={"user_input": ""})
        assert r.status_code == 422

    def test_missing_user_input_422(self) -> None:
        r = client.post("/agent/run", json={})
        assert r.status_code == 422