import json

import pytest
from django.urls import reverse

from review_agent.models import (
    Conversation,
    FileSummaryBatch,
    FileSummaryItem,
    RegulatoryReviewBatch,
    WorkflowEvent,
    WorkflowNodeRun,
)
from review_agent.regulatory_review.services.info_extract import detect_regulatory_condition_candidates
from review_agent.regulatory_review.workflow import (
    create_regulatory_review_batch,
    start_regulatory_review_workflow,
)


pytestmark = pytest.mark.django_db


def test_detect_regulatory_condition_candidates_from_summary_items(django_user_model):
    user = django_user_model.objects.create_user(username="owner", password="pass")
    conversation = Conversation.objects.create(user=user, title="会话")
    summary = FileSummaryBatch.objects.create(
        conversation=conversation,
        user=user,
        batch_no="FS-COND",
        status=FileSummaryBatch.Status.SUCCESS,
        product_name="甲胎蛋白检测试剂盒",
    )
    FileSummaryItem.objects.create(
        batch=summary,
        file_index=1,
        directory_level="临床评价资料",
        file_name="免临床评价资料.docx",
        file_type="docx",
        relative_path="4.临床评价资料/免临床评价资料.docx",
        storage_path="missing.docx",
    )

    candidates = detect_regulatory_condition_candidates(summary)

    assert candidates["product_category"]["suggested"] == "体外诊断试剂"
    assert candidates["registration_type"]["suggested"] == "首次注册"
    assert candidates["clinical_evaluation_path"]["suggested"] == "免临床"
    assert candidates["product_name"]["suggested"] == "甲胎蛋白检测试剂盒"


def test_workflow_pauses_before_rule_scope_until_conditions_confirmed(settings, tmp_path, django_user_model):
    settings.MEDIA_ROOT = tmp_path
    user = django_user_model.objects.create_user(username="owner", password="pass")
    conversation = Conversation.objects.create(user=user, title="会话")
    summary = FileSummaryBatch.objects.create(
        conversation=conversation,
        user=user,
        batch_no="FS-COND",
        status=FileSummaryBatch.Status.SUCCESS,
        product_name="甲胎蛋白检测试剂盒",
    )
    batch = create_regulatory_review_batch(
        conversation=conversation,
        user=user,
        source_summary_batch=summary,
    )

    start_regulatory_review_workflow(batch, async_run=False)

    batch.refresh_from_db()
    condition_node = WorkflowNodeRun.objects.get(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
        node_code="condition_confirm",
    )
    rule_scope_node = WorkflowNodeRun.objects.get(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
        node_code="rule_scope",
    )
    assert batch.status == RegulatoryReviewBatch.Status.WAITING_USER
    assert condition_node.status == WorkflowNodeRun.Status.WAITING_USER
    assert rule_scope_node.status == WorkflowNodeRun.Status.PENDING
    assert batch.condition_json["candidates"]["product_category"]["suggested"] == "体外诊断试剂"
    assert WorkflowEvent.objects.filter(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
        event_type="waiting_user",
    ).exists()


def test_confirm_conditions_endpoint_resumes_workflow(client, settings, tmp_path, django_user_model):
    settings.MEDIA_ROOT = tmp_path
    settings.REGULATORY_REVIEW_ASYNC = False
    user = django_user_model.objects.create_user(username="owner", password="pass")
    conversation = Conversation.objects.create(user=user, title="会话")
    summary = FileSummaryBatch.objects.create(
        conversation=conversation,
        user=user,
        batch_no="FS-COND",
        status=FileSummaryBatch.Status.SUCCESS,
        product_name="甲胎蛋白检测试剂盒",
    )
    batch = create_regulatory_review_batch(
        conversation=conversation,
        user=user,
        source_summary_batch=summary,
    )
    start_regulatory_review_workflow(batch, async_run=False)
    client.force_login(user)

    response = client.post(
        reverse("regulatory_review_confirm_conditions", args=[batch.pk]),
        data=json.dumps(
            {
                "conditions": {
                    "product_category": "体外诊断试剂",
                    "registration_type": "首次注册",
                    "clinical_evaluation_path": "免临床",
                    "product_name": "甲胎蛋白检测试剂盒",
                    "model_spec": "卡型",
                    "intended_use": "用于甲胎蛋白检测",
                }
            }
        ),
        content_type="application/json",
    )

    batch.refresh_from_db()
    assert response.status_code == 200
    assert response.json()["batch"]["status"] == RegulatoryReviewBatch.Status.SUCCESS
    assert batch.condition_json["confirmed"] is True
    assert batch.condition_json["confirmed_conditions"]["model_spec"] == "卡型"
    assert WorkflowNodeRun.objects.get(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
        node_code="condition_confirm",
    ).status == WorkflowNodeRun.Status.SUCCESS