feat(file-summary): 实现文件处理技能链路
This commit is contained in:
29
tests/test_file_summary_product_detect.py
Normal file
29
tests/test_file_summary_product_detect.py
Normal file
@@ -0,0 +1,29 @@
|
||||
import pytest
|
||||
|
||||
from review_agent.file_summary.services.product_detect import detect_product_name
|
||||
from review_agent.models import Conversation, FileSummaryBatch, FileSummaryItem
|
||||
|
||||
|
||||
pytestmark = pytest.mark.django_db
|
||||
|
||||
|
||||
def test_detect_product_name_from_top_level_directory(django_user_model):
|
||||
user = django_user_model.objects.create_user(username="owner", password="pass")
|
||||
conversation = Conversation.objects.create(user=user, title="新对话 06-06")
|
||||
batch = FileSummaryBatch.objects.create(conversation=conversation, user=user, batch_no="FS-D")
|
||||
FileSummaryItem.objects.create(
|
||||
batch=batch,
|
||||
file_index=1,
|
||||
file_name="说明书.docx",
|
||||
file_type="docx",
|
||||
relative_path="甲型试剂盒/说明书.docx",
|
||||
storage_path="x",
|
||||
)
|
||||
|
||||
product_name = detect_product_name(batch)
|
||||
|
||||
batch.refresh_from_db()
|
||||
conversation.refresh_from_db()
|
||||
assert product_name == "甲型试剂盒"
|
||||
assert batch.product_name == "甲型试剂盒"
|
||||
assert conversation.title == "甲型试剂盒-文件汇总"
|
||||
Reference in New Issue
Block a user