opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-11 23:18:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: avoid MagicMock dependency in segment signing tests

Browse Source
This commit is contained in:
Yanli 盐粒 2026-03-01 17:23:18 +08:00
parent 0d6a4bac0f
commit 30af50cb47
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
api/tests/unit_tests/models/test_dataset_models.py
View File

@ -490,6 +490,15 @@ class TestDocumentModelRelationships:
class TestDocumentSegmentIndexing:
"""Test suite for DocumentSegment model indexing and operations."""
@staticmethod
def _mock_scalars_result(upload_file_ids: list[str]):
class _ScalarsResult:
@staticmethod
def all() -> list[str]:
return upload_file_ids
return _ScalarsResult()
def test_document_segment_creation_with_required_fields(self):
"""Test creating a document segment with all required fields."""
# Arrange
@ -562,8 +571,7 @@ class TestDocumentSegmentIndexing:
tokens=1,
created_by=str(uuid4()),
)
mock_scalars_result = MagicMock()
mock_scalars_result.all.return_value = [upload_file_id]
mock_scalars_result = self._mock_scalars_result([upload_file_id])
# Act
with (
@ -594,8 +602,7 @@ class TestDocumentSegmentIndexing:
tokens=1,
created_by=str(uuid4()),
)
mock_scalars_result = MagicMock()
mock_scalars_result.all.return_value = [upload_file_id]
mock_scalars_result = self._mock_scalars_result([upload_file_id])
# Act
with (
@ -630,8 +637,7 @@ class TestDocumentSegmentIndexing:
tokens=1,
created_by=str(uuid4()),
)
mock_scalars_result = MagicMock()
mock_scalars_result.all.return_value = [allowed_upload_file_id]
mock_scalars_result = self._mock_scalars_result([allowed_upload_file_id])
# Act
with (