feat(file-summary): 添加文件汇总数据模型

tests/test_file_summary_models.py

@@ -0,0 +1,113 @@
+import pytest
+from django.contrib.auth import get_user_model
+from django.db import IntegrityError, transaction
+
+from review_agent.models import (
+    Conversation,
+    ExportedSummaryFile,
+    FileAttachment,
+    FileSummaryBatch,
+    FileSummaryBatchAttachment,
+    FileSummaryItem,
+)
+
+
+pytestmark = pytest.mark.django_db
+
+
+def create_user(username="u1"):
+    return get_user_model().objects.create_user(username=username, password="pass")
+
+
+def test_attachment_versions_are_unique_per_conversation_and_name():
+    user = create_user()
+    conversation = Conversation.objects.create(user=user, title="会话")
+
+    first = FileAttachment.objects.create(
+        conversation=conversation,
+        user=user,
+        original_name="资料.docx",
+        version_no=1,
+        is_active=False,
+        storage_path="media/a.docx",
+        file_size=10,
+    )
+    second = FileAttachment.objects.create(
+        conversation=conversation,
+        user=user,
+        original_name="资料.docx",
+        version_no=2,
+        storage_path="media/b.docx",
+        file_size=12,
+    )
+
+    assert first.version_no == 1
+    assert second.version_no == 2
+
+    with pytest.raises(IntegrityError), transaction.atomic():
+        FileAttachment.objects.create(
+            conversation=conversation,
+            user=user,
+            original_name="资料.docx",
+            version_no=2,
+            storage_path="media/c.docx",
+            file_size=14,
+        )
+
+
+def test_batch_attachment_and_item_unique_constraints():
+    user = create_user()
+    conversation = Conversation.objects.create(user=user, title="会话")
+    attachment = FileAttachment.objects.create(
+        conversation=conversation,
+        user=user,
+        original_name="资料.docx",
+        storage_path="media/a.docx",
+        file_size=10,
+    )
+    batch = FileSummaryBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        batch_no="FS-001",
+    )
+
+    FileSummaryBatchAttachment.objects.create(batch=batch, attachment=attachment)
+    with pytest.raises(IntegrityError), transaction.atomic():
+        FileSummaryBatchAttachment.objects.create(batch=batch, attachment=attachment)
+
+    FileSummaryItem.objects.create(
+        batch=batch,
+        file_index=1,
+        file_name="资料.docx",
+        file_type="docx",
+        relative_path="资料.docx",
+        storage_path="media/a.docx",
+    )
+    with pytest.raises(IntegrityError), transaction.atomic():
+        FileSummaryItem.objects.create(
+            batch=batch,
+            file_index=2,
+            file_name="资料.docx",
+            file_type="docx",
+            relative_path="资料.docx",
+            storage_path="media/a.docx",
+        )
+
+
+def test_exported_file_traces_to_user_and_conversation():
+    user = create_user()
+    conversation = Conversation.objects.create(user=user, title="会话")
+    batch = FileSummaryBatch.objects.create(
+        conversation=conversation,
+        user=user,
+        batch_no="FS-002",
+    )
+    exported = ExportedSummaryFile.objects.create(
+        batch=batch,
+        export_type=ExportedSummaryFile.ExportType.MARKDOWN,
+        file_name="summary.md",
+        storage_path="media/summary.md",
+    )
+
+    assert exported.batch.user == user
+    assert exported.batch.conversation == conversation