import asyncio
from types import SimpleNamespace

from gfa_core.mas.agents.base import AgentMessage, AgentState
from gfa_core.mas.agents.intent_v2 import IntentAgentV2


def test_fallback_does_not_turn_unrecognized_continuation_into_identify():
    intent = IntentAgentV2(None)._fallback_recognition("执行接下来的分析")

    assert intent["stages"] == []
    assert intent["needs_clarification"] is True
    assert intent["clarification_type"] == "analysis_stage"


def test_validation_does_not_turn_empty_model_stages_into_identify():
    intent = IntentAgentV2(None)._validate_intent(
        {
            "gene_family": "SPL",
            "species": "Ft",
            "stages": [],
            "params": {},
        },
        "执行接下来的步骤",
    )

    assert intent["stages"] == []
    assert intent["needs_clarification"] is True
    assert intent["clarification_type"] == "analysis_stage"


def test_fallback_still_recognizes_explicit_identify():
    intent = IntentAgentV2(None)._fallback_recognition("执行 identify 分析")

    assert intent["stages"] == ["identify"]
    assert intent.get("needs_clarification") is not True


def test_empty_model_response_cannot_authorize_identify_and_keeps_goal_context():
    class _EmptyLLM:
        async def acall(self, *_args, **_kwargs):
            return SimpleNamespace(content="")

    state = AgentState(session_id="empty-intent", job_id="job-empty-intent")
    message = AgentMessage(
        sender="router",
        receiver="intent_v2",
        message_type="recognize_intent",
        payload={
            "query": "执行接下来的分析",
            "job_id": "job-empty-intent",
            "workspace": "/tmp/job-empty-intent",
            "goal_context": {
                "user_terms": {
                    "gene_family": "SPL",
                    "target_species": ["Ft"],
                    "reference_species": ["At"],
                    "comparative_species": [],
                }
            },
        },
    )

    returned = asyncio.run(IntentAgentV2(_EmptyLLM()).execute(state, message))
    intent = returned.results["results"]["intent"]

    assert intent["stages"] == []
    assert intent["needs_clarification"] is True
    assert intent["target_species"] == ["Ft"]
    assert intent["reference_species"] == ["At"]