youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 08:58:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: use EnumText for dataset and replace string literals 4 (#33606)

Browse Source 
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
tmimmanuel
2026-03-18 00:18:08 +00:00
committed by GitHub
parent 0bc6c3a73e
commit 3870b2ad2d
69 changed files with 1027 additions and 849 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -30,6 +30,7 @@ from controllers.console.datasets.error import (
InvalidActionError,
InvalidMetadataError,
)
from models.enums import DataSourceType, IndexingStatus
def unwrap(func):
@ -62,8 +63,8 @@ def document():
return MagicMock(
id="doc-1",
tenant_id="tenant-1",
indexing_status="indexing",
data_source_type="upload_file",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.UPLOAD_FILE,
data_source_info_dict={"upload_file_id": "file-1"},
doc_form="text",
archived=False,
@ -407,7 +408,7 @@ class TestDocumentProcessingApi:
api = DocumentProcessingApi()
method = unwrap(api.patch)
doc = MagicMock(indexing_status="error", is_paused=True)
doc = MagicMock(indexing_status=IndexingStatus.ERROR, is_paused=True)
with (
app.test_request_context("/"),
@ -425,7 +426,7 @@ class TestDocumentProcessingApi:
api = DocumentProcessingApi()
method = unwrap(api.patch)
document = MagicMock(indexing_status="paused", is_paused=True)
document = MagicMock(indexing_status=IndexingStatus.PAUSED, is_paused=True)
with (
app.test_request_context("/"),
@ -461,7 +462,7 @@ class TestDocumentProcessingApi:
api = DocumentProcessingApi()
method = unwrap(api.patch)
document = MagicMock(indexing_status="completed")
document = MagicMock(indexing_status=IndexingStatus.COMPLETED)
with app.test_request_context("/"), patch.object(api, "get_document", return_value=document):
with pytest.raises(InvalidActionError):
@ -630,7 +631,7 @@ class TestDocumentRetryApi:
payload = {"document_ids": ["doc-1"]}
document = MagicMock(indexing_status="indexing", archived=False)
document = MagicMock(indexing_status=IndexingStatus.INDEXING, archived=False)
with (
app.test_request_context("/", json=payload),
@ -659,7 +660,7 @@ class TestDocumentRetryApi:
payload = {"document_ids": ["doc-1"]}
document = MagicMock(indexing_status="completed", archived=False)
document = MagicMock(indexing_status=IndexingStatus.COMPLETED, archived=False)
with (
app.test_request_context("/", json=payload),
@ -817,8 +818,8 @@ class TestDocumentIndexingEstimateApi:
method = unwrap(api.get)
document = MagicMock(
indexing_status="indexing",
data_source_type="upload_file",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.UPLOAD_FILE,
data_source_info_dict={"upload_file_id": "file-1"},
tenant_id="tenant-1",
doc_form="text",
@ -844,8 +845,8 @@ class TestDocumentIndexingEstimateApi:
method = unwrap(api.get)
document = MagicMock(
indexing_status="indexing",
data_source_type="upload_file",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.UPLOAD_FILE,
data_source_info_dict={"upload_file_id": "file-1"},
tenant_id="tenant-1",
doc_form="text",
@ -882,7 +883,7 @@ class TestDocumentIndexingEstimateApi:
api = DocumentIndexingEstimateApi()
method = unwrap(api.get)
document = MagicMock(indexing_status="completed")
document = MagicMock(indexing_status=IndexingStatus.COMPLETED)
with app.test_request_context("/"), patch.object(api, "get_document", return_value=document):
with pytest.raises(DocumentAlreadyFinishedError):
@ -963,8 +964,8 @@ class TestDocumentBatchIndexingEstimateApi:
method = unwrap(api.get)
doc = MagicMock(
indexing_status="indexing",
data_source_type="website_crawl",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.WEBSITE_CRAWL,
data_source_info_dict={
"provider": "firecrawl",
"job_id": "j1",
@ -992,8 +993,8 @@ class TestDocumentBatchIndexingEstimateApi:
method = unwrap(api.get)
doc = MagicMock(
indexing_status="indexing",
data_source_type="notion_import",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.NOTION_IMPORT,
data_source_info_dict={
"credential_id": "c1",
"notion_workspace_id": "w1",
@ -1020,7 +1021,7 @@ class TestDocumentBatchIndexingEstimateApi:
method = unwrap(api.get)
document = MagicMock(
indexing_status="indexing",
indexing_status=IndexingStatus.INDEXING,
data_source_type="unknown",
data_source_info_dict={},
doc_form="text",
@ -1130,7 +1131,7 @@ class TestDocumentProcessingApiResume:
api = DocumentProcessingApi()
method = unwrap(api.patch)
document = MagicMock(indexing_status="completed", is_paused=False)
document = MagicMock(indexing_status=IndexingStatus.COMPLETED, is_paused=False)
with app.test_request_context("/"), patch.object(api, "get_document", return_value=document):
with pytest.raises(InvalidActionError):
@ -1348,8 +1349,8 @@ class TestDocumentIndexingEdgeCases:
method = unwrap(api.get)
document = MagicMock(
indexing_status="indexing",
data_source_type="upload_file",
indexing_status=IndexingStatus.INDEXING,
data_source_type=DataSourceType.UPLOAD_FILE,
data_source_info_dict={"upload_file_id": "file-1"},
tenant_id="tenant-1",
doc_form="text",