86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
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()
|