115 lines
4.3 KiB
Python
115 lines
4.3 KiB
Python
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"
|