from django.urls import reverse

from agent_core.results import AgentResult
from apps.audit.models import AgentAuditLog
from apps.chat.models import Conversation
from apps.documents.models import SubmissionBatch, UploadedDocument


def _create_conversation_with_batch():
    batch = SubmissionBatch.objects.create(
        batch_id="SUB-20260604-001",
        product_name="新型冠状病毒 2019-nCoV 核酸检测试剂盒",
        workflow_type="registration",
        conversation_id="conv-001",
        file_count=2,
        page_count=12,
        import_status="completed",
    )
    conversation = Conversation.objects.create(
        conversation_id="conv-001",
        title="新型冠状病毒 2019-nCoV 核酸检测试剂盒",
        product_name=batch.product_name,
        batch_id=batch.batch_id,
        task_status="processing",
        node_results=[
            {"label": "资料包导入", "status": "已完成"},
            {"label": "目录汇总", "status": "处理中"},
        ],
    )
    return batch, conversation


def test_chat_post_returns_agent_result_and_audit_log(client, db, monkeypatch):
    batch, conversation = _create_conversation_with_batch()
    UploadedDocument.objects.create(
        batch=batch,
        scenario_id="document_review",
        original_name="说明书.md",
        file_type="md",
        size=1,
        status=UploadedDocument.STATUS_INDEXED,
    )

    monkeypatch.setattr(
        "apps.chat.views.run_agent",
        lambda *args, **kwargs: AgentResult(answer="模拟回答", status="success"),
    )

    response = client.post(
        reverse("chat:detail", args=[conversation.conversation_id]),
        {"message": "如何处理异常？"},
    )

    assert response.status_code == 200
    content = response.content.decode("utf-8")
    assert "审核智能体" in content
    assert "模拟回答" in content
    assert AgentAuditLog.objects.count() == 1
    assert AgentAuditLog.objects.get().batch_id == batch.batch_id


def test_chat_rejects_empty_message(client, db):
    _batch, conversation = _create_conversation_with_batch()

    response = client.post(reverse("chat:detail", args=[conversation.conversation_id]), {"message": ""})

    assert response.status_code == 200
    assert AgentAuditLog.objects.count() == 0
    assert "请输入要咨询的问题" in response.content.decode("utf-8")


def test_chat_passes_selected_document_ids_to_agent_core(client, db, monkeypatch):
    batch, conversation = _create_conversation_with_batch()
    selected = UploadedDocument.objects.create(
        batch=batch,
        scenario_id="document_review",
        original_name="selected.md",
        file_type="md",
        size=1,
        status=UploadedDocument.STATUS_INDEXED,
    )
    UploadedDocument.objects.create(
        batch=batch,
        scenario_id="document_review",
        original_name="other.md",
        file_type="md",
        size=1,
        status=UploadedDocument.STATUS_INDEXED,
    )
    captured = {}

    def fake_run_agent(scenario_config, user_input, options=None):
        captured["options"] = options or {}
        return AgentResult(answer="ok", status="success")

    monkeypatch.setattr("apps.chat.views.run_agent", fake_run_agent)

    response = client.post(
        reverse("chat:detail", args=[conversation.conversation_id]),
        {"message": "只查选中文档", "document_ids": [str(selected.id)]},
    )

    assert response.status_code == 200
    assert captured["options"]["document_ids"] == [selected.id]
    assert captured["options"]["conversation_id"] == conversation.conversation_id
    assert captured["options"]["batch_id"] == batch.batch_id


def test_chat_renders_three_column_workspace_and_node_results(client, db):
    batch, conversation = _create_conversation_with_batch()
    UploadedDocument.objects.create(
        batch=batch,
        scenario_id="document_review",
        original_name="说明书.md",
        file_type="md",
        size=1,
        status=UploadedDocument.STATUS_INDEXED,
    )

    response = client.get(reverse("chat:detail", args=[conversation.conversation_id]))

    content = response.content.decode("utf-8")
    assert response.status_code == 200
    assert "会话历史" in content
    assert "对话区与节点导航" in content
    assert "上传区" in content
    assert "资料包导入 / 已完成" in content
    assert "目录汇总 / 处理中" in content