114 lines
3.2 KiB
Python
114 lines
3.2 KiB
Python
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
|