import pytest
from django.urls import reverse

from review_agent.models import Conversation, FileSummaryBatch, RegulatoryReviewBatch, WorkflowNodeRun


pytestmark = pytest.mark.django_db


def test_regulatory_batch_status_requires_owner(client, django_user_model):
    owner = django_user_model.objects.create_user(username="owner", password="pass")
    other = django_user_model.objects.create_user(username="other", password="pass")
    conversation = Conversation.objects.create(user=owner, title="会话")
    summary = FileSummaryBatch.objects.create(
        conversation=conversation,
        user=owner,
        batch_no="FS-OK",
        status=FileSummaryBatch.Status.SUCCESS,
    )
    batch = RegulatoryReviewBatch.objects.create(
        conversation=conversation,
        user=owner,
        source_summary_batch=summary,
        batch_no="RR-STATUS",
    )
    WorkflowNodeRun.objects.create(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
        node_group="regulatory_review",
        node_code="prepare",
        node_name="准备",
        progress=50,
    )

    client.force_login(other)
    denied = client.get(reverse("regulatory_review_batch_status", args=[batch.pk]))
    assert denied.status_code == 404

    client.force_login(owner)
    allowed = client.get(reverse("regulatory_review_batch_status", args=[batch.pk]))
    assert allowed.status_code == 200
    payload = allowed.json()
    assert payload["batch"]["workflow_type"] == "regulatory_review"
    assert payload["batch"]["batch_no"] == "RR-STATUS"
    assert payload["nodes"][0]["node_code"] == "prepare"


def test_regulatory_batch_status_exposes_condition_confirmation(client, django_user_model):
    owner = django_user_model.objects.create_user(username="owner", password="pass")
    conversation = Conversation.objects.create(user=owner, title="会话")
    summary = FileSummaryBatch.objects.create(
        conversation=conversation,
        user=owner,
        batch_no="FS-OK",
        status=FileSummaryBatch.Status.SUCCESS,
    )
    batch = RegulatoryReviewBatch.objects.create(
        conversation=conversation,
        user=owner,
        source_summary_batch=summary,
        batch_no="RR-WAIT",
        status=RegulatoryReviewBatch.Status.WAITING_USER,
        condition_json={
            "confirmed": False,
            "candidates": {
                "product_category": {
                    "label": "产品类别",
                    "input_type": "select",
                    "options": ["体外诊断试剂", "医疗器械", "其他"],
                    "suggested": "体外诊断试剂",
                }
            },
        },
    )
    client.force_login(owner)

    response = client.get(reverse("regulatory_review_batch_status", args=[batch.pk]))

    payload = response.json()
    assert payload["batch"]["status"] == RegulatoryReviewBatch.Status.WAITING_USER
    assert payload["condition_confirmation"]["batch_id"] == batch.pk
    assert payload["condition_confirmation"]["candidates"]["product_category"]["suggested"] == "体外诊断试剂"