import pytest

from review_agent.application_form_fill.services.field_merge import merge_fields, normalize_field_value, rank_source


def test_normalize_field_value_removes_whitespace():
    assert normalize_field_value(" 2-8℃ 保存 \n 有效期12个月 ") == "2-8℃保存有效期12个月"


def test_rank_source_prefers_instructions():
    assert rank_source("说明书") < rank_source("产品技术要求")


def test_merge_fields_prefers_instructions_and_marks_conflict():
    regex_results = {
        "fields": [
            {
                "key": "storage_condition_and_validity",
                "label": "产品储存条件及有效期",
                "value": "2-8℃保存，有效期12个月",
                "source_file": "说明书.txt",
                "source_role": "说明书",
                "evidence": "产品储存条件及有效期：2-8℃保存，有效期12个月",
                "confidence": 0.75,
            },
            {
                "key": "storage_condition_and_validity",
                "label": "产品储存条件及有效期",
                "value": "-20℃保存",
                "source_file": "产品技术要求.txt",
                "source_role": "产品技术要求",
                "evidence": "产品储存条件及有效期：-20℃保存",
                "confidence": 0.8,
            },
        ]
    }

    merged, conflicts = merge_fields(regex_results, {"fields": []})

    field = merged["storage_condition_and_validity"]
    assert field.value == "2-8℃保存，有效期12个月"
    assert field.has_conflict is True
    assert conflicts[0]["selected_value"] == "2-8℃保存，有效期12个月"
    assert conflicts[0]["conflict_values"][0]["value"] == "-20℃保存"


def test_merge_fields_combines_consistent_values_without_conflict():
    regex_results = {
        "fields": [
            {
                "key": "product_name",
                "label": "产品名称",
                "value": "甲胎蛋白检测试剂盒",
                "source_file": "说明书.txt",
                "source_role": "说明书",
                "evidence": "产品名称：甲胎蛋白检测试剂盒",
                "confidence": 0.75,
            }
        ]
    }
    llm_results = {
        "fields": [
            {
                "key": "product_name",
                "label": "产品名称",
                "value": "甲胎蛋白 检测试剂盒",
                "source_file": "产品技术要求.txt",
                "source_role": "产品技术要求",
                "evidence": "产品名称：甲胎蛋白 检测试剂盒",
                "confidence": 0.9,
            }
        ]
    }

    merged, conflicts = merge_fields(regex_results, llm_results)

    assert merged["product_name"].value == "甲胎蛋白检测试剂盒"
    assert merged["product_name"].has_conflict is False
    assert conflicts == []


def test_merge_fields_fills_agent_from_applicant_for_now():
    regex_results = {
        "fields": [
            {
                "key": "applicant_name",
                "label": "注册人名称",
                "value": "卡尤迪生物科技宜兴有限公司",
                "source_file": "目标产品说明书.docx",
                "source_role": "说明书",
                "evidence": "生产企业名称：卡尤迪生物科技宜兴有限公司",
                "confidence": 0.75,
            },
            {
                "key": "applicant_address",
                "label": "注册人住所",
                "value": "江苏省宜兴经济技术开发区杏里路10号",
                "source_file": "目标产品说明书.docx",
                "source_role": "说明书",
                "evidence": "生产企业住所：江苏省宜兴经济技术开发区杏里路10号",
                "confidence": 0.75,
            },
        ]
    }

    merged, conflicts = merge_fields(regex_results, {"fields": []})

    assert merged["agent_name"].value == "卡尤迪生物科技宜兴有限公司"
    assert merged["agent_name"].label == "代理人名称"
    assert merged["agent_address"].value == "江苏省宜兴经济技术开发区杏里路10号"
    assert conflicts == []