jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-19 12:00:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api): migrate dataset document response schemas to BaseModel (#35298)

Browse Source 
Co-authored-by: ai-hpc <ai-hpc@users.noreply.github.com>
Co-authored-by: Asuka Minato <i@asukaminato.eu.org>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
NVIDIAN
2026-04-16 22:02:04 -07:00
committed by GitHub
parent dc3f992e6e
commit af21dc7df8
2 changed files with 130 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/tests/unit_tests/controllers/console/datasets/test_datasets_document.py
View File

@@ -1,3 +1,4 @@
from types import SimpleNamespace
from unittest.mock import MagicMock, patch
import pytest
@@ -215,17 +216,23 @@ class TestDatasetDocumentListApi:
method = unwrap(api.post)
payload = {"indexing_technique": "economy"}
created_dataset = SimpleNamespace(id="ds-1", name="Dataset", indexing_technique="economy")
created_document = SimpleNamespace(id="doc-1", name="Document", doc_metadata_details=None)
with (
app.test_request_context("/", json=payload),
patch.object(type(console_ns), "payload", payload),
patch(
"controllers.console.datasets.datasets_document.DatasetService.get_dataset",
return_value=created_dataset,
),
patch(
"controllers.console.datasets.datasets_document.DocumentService.document_create_args_validate",
return_value=None,
),
patch(
"controllers.console.datasets.datasets_document.DocumentService.save_document_with_dataset_id",
return_value=([MagicMock()], "batch-1"),
return_value=([created_document], "batch-1"),
),
):
response = method(api, "ds-1")