import json

import pytest
from openpyxl import load_workbook

from review_agent.application_form_fill.schemas import MergedField, TemplateSpec
from review_agent.application_form_fill.services.traceability_export import save_traceability_exports
from review_agent.models import (
    ApplicationFormFillArtifact,
    ApplicationFormFillBatch,
    Conversation,
    ExportedSummaryFile,
    FileSummaryBatch,
)


pytestmark = pytest.mark.django_db


def test_traceability_exports_excel_json_and_records(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-TRACE")
    batch = ApplicationFormFillBatch.objects.create(
        conversation=conversation,
        user=user,
        source_summary_batch=summary,
        batch_no="AFF-TRACE",
        work_dir=str(tmp_path / "aff" / "AFF-TRACE"),
    )
    spec = TemplateSpec(
        code="registration_certificate",
        name="注册证格式",
        source_file="template.docx",
        output_label="注册证格式",
        applies_when={},
        file_format="docx",
        fields=[{"key": "product_name", "label": "产品名称"}],
    )
    merged_fields = {
        "product_name": MergedField(
            "product_name",
            "产品名称",
            "甲胎蛋白检测试剂盒",
            "说明书.txt",
            "产品名称：甲胎蛋白检测试剂盒",
            0.8,
        )
    }
    conflicts = [
        {
            "field_key": "storage_condition",
            "field_label": "储存条件",
            "selected_value": "2-8℃",
            "conflict_values": [{"value": "-20℃", "source_file": "产品技术要求.txt"}],
            "handling": "说明书优先",
        }
    ]

    exports = save_traceability_exports(
        batch,
        merged_fields,
        conflicts,
        [spec],
        [{"template_label": "注册证格式", "word_status": "success", "pdf_status": "待增强"}],
    )

    assert {export.export_type for export in exports} == {
        ExportedSummaryFile.ExportType.EXCEL,
        ExportedSummaryFile.ExportType.JSON,
    }
    excel_export = next(export for export in exports if export.export_type == ExportedSummaryFile.ExportType.EXCEL)
    workbook = load_workbook(excel_export.storage_path)
    assert workbook.sheetnames == ["字段追溯", "冲突字段", "低置信度条目", "生成结果"]
    assert workbook["字段追溯"]["B2"].value == "产品名称"
    assert workbook["冲突字段"]["C2"].value == "-20℃"

    json_export = next(export for export in exports if export.export_type == ExportedSummaryFile.ExportType.JSON)
    payload = json.loads(open(json_export.storage_path, encoding="utf-8").read())
    assert payload["merged_fields"]["product_name"]["value"] == "甲胎蛋白检测试剂盒"
    assert ApplicationFormFillArtifact.objects.filter(
        batch=batch,
        artifact_type=ApplicationFormFillArtifact.ArtifactType.TRACEABILITY,
    ).exists()