feat(application-form-fill): 实现自动填表模板选择

tests/test_application_form_fill_template_select.py

@@ -0,0 +1,114 @@
+import pytest
+
+from review_agent.application_form_fill.services.template_config import load_template_config
+from review_agent.application_form_fill.services.template_select import (
+    detect_registration_type,
+    parse_requested_templates,
+    select_templates,
+)
+from review_agent.models import ApplicationFormFillBatch, Conversation, FileSummaryBatch, RegulatoryReviewBatch
+
+
+pytestmark = pytest.mark.django_db
+
+
+@pytest.mark.parametrize(
+    ("message", "expected"),
+    [
+        ("帮我填注册证", ["registration_certificate"]),
+        ("生成变更注册备案文件", ["change_registration"]),
+        ("生成安全和性能基本原则清单", ["essential_principles"]),
+        ("请生成全部模板", ["registration_certificate", "change_registration", "essential_principles"]),
+        ("普通聊天", []),
+    ],
+)
+def test_parse_requested_templates(message, expected):
+    assert parse_requested_templates(message) == expected
+
+
+def test_detect_registration_type_prefers_user_message(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-SEL")
+    regulatory = RegulatoryReviewBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        source_summary_batch=summary,
+        batch_no="RR-SEL",
+        condition_json={"confirmed_conditions": {"registration_type": "变更注册"}},
+    )
+    batch = ApplicationFormFillBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        source_summary_batch=summary,
+        source_regulatory_batch=regulatory,
+        batch_no="AFF-SEL",
+    )
+
+    value, source = detect_registration_type(batch=batch, message="首次注册资料，请填注册证")
+
+    assert value == "首次注册"
+    assert source == ApplicationFormFillBatch.RegistrationTypeSource.USER_MESSAGE
+
+
+def test_detect_registration_type_falls_back_to_regulatory_batch_and_file_candidates(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-SEL-2")
+    regulatory = RegulatoryReviewBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        source_summary_batch=summary,
+        batch_no="RR-SEL-2",
+        condition_json={"confirmed_conditions": {"registration_type": "变更注册"}},
+    )
+    batch = ApplicationFormFillBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        source_summary_batch=summary,
+        source_regulatory_batch=regulatory,
+        batch_no="AFF-SEL-2",
+    )
+
+    regulatory_value, regulatory_source = detect_registration_type(batch=batch, message="")
+    file_value, file_source = detect_registration_type(
+        message="",
+        file_candidates={"registration_type": {"suggested": "备案"}},
+    )
+
+    assert (regulatory_value, regulatory_source) == (
+        "变更注册",
+        ApplicationFormFillBatch.RegistrationTypeSource.REGULATORY_BATCH,
+    )
+    assert (file_value, file_source) == (
+        "备案",
+        ApplicationFormFillBatch.RegistrationTypeSource.FILE_EXTRACT,
+    )
+
+
+def test_select_default_templates_for_initial_registration():
+    config = load_template_config()
+
+    specs, risk_notes = select_templates(config, [], "首次注册")
+
+    assert [spec.code for spec in specs] == ["registration_certificate", "essential_principles"]
+    assert risk_notes == []
+
+
+def test_select_default_templates_for_change_registration():
+    config = load_template_config()
+
+    specs, risk_notes = select_templates(config, [], "变更注册")
+
+    assert [spec.code for spec in specs] == ["change_registration", "essential_principles"]
+    assert risk_notes == []
+
+
+def test_select_user_requested_mismatch_is_allowed_with_risk_note():
+    config = load_template_config()
+
+    specs, risk_notes = select_templates(config, ["change_registration"], "首次注册")
+
+    assert [spec.code for spec in specs] == ["change_registration"]
+    assert risk_notes
+    assert risk_notes[0]["type"] == "template_registration_mismatch"