mirror of https://github.com/langgenius/dify.git
fix typo
This commit is contained in:
parent
4a0ccd0823
commit
42178b9ecb
|
|
@ -710,7 +710,7 @@ class TestDocumentIndexingTasks:
|
||||||
)
|
)
|
||||||
document_ids = [doc.id for doc in documents]
|
document_ids = [doc.id for doc in documents]
|
||||||
tenant_id = dataset.tenant_id
|
tenant_id = dataset.tenant_id
|
||||||
dateset_id = dataset.id
|
dataset_id = dataset.id
|
||||||
|
|
||||||
# Mock the task function
|
# Mock the task function
|
||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
@ -743,7 +743,7 @@ class TestDocumentIndexingTasks:
|
||||||
|
|
||||||
# Verify correct parameters for each call
|
# Verify correct parameters for each call
|
||||||
calls = mock_task_func.delay.call_args_list
|
calls = mock_task_func.delay.call_args_list
|
||||||
assert calls[0][1] == {"tenant_id": tenant_id, "dataset_id": dateset_id, "document_ids": ["waiting-doc-1"]}
|
assert calls[0][1] == {"tenant_id": tenant_id, "dataset_id": dataset_id, "document_ids": ["waiting-doc-1"]}
|
||||||
|
|
||||||
# Verify queue is empty after processing (tasks were pulled)
|
# Verify queue is empty after processing (tasks were pulled)
|
||||||
remaining_tasks = queue.pull_tasks(count=10) # Pull more than we added
|
remaining_tasks = queue.pull_tasks(count=10) # Pull more than we added
|
||||||
|
|
@ -768,7 +768,7 @@ class TestDocumentIndexingTasks:
|
||||||
)
|
)
|
||||||
document_ids = [doc.id for doc in documents]
|
document_ids = [doc.id for doc in documents]
|
||||||
tenant_id = dataset.tenant_id
|
tenant_id = dataset.tenant_id
|
||||||
dateset_id = dataset.id
|
dataset_id = dataset.id
|
||||||
|
|
||||||
# Mock IndexingRunner to raise an exception
|
# Mock IndexingRunner to raise an exception
|
||||||
mock_external_service_dependencies["indexing_runner_instance"].run.side_effect = Exception("Test error")
|
mock_external_service_dependencies["indexing_runner_instance"].run.side_effect = Exception("Test error")
|
||||||
|
|
@ -807,7 +807,7 @@ class TestDocumentIndexingTasks:
|
||||||
|
|
||||||
# Verify correct parameters for the call
|
# Verify correct parameters for the call
|
||||||
call = mock_task_func.delay.call_args
|
call = mock_task_func.delay.call_args
|
||||||
assert call[1] == {"tenant_id": tenant_id, "dataset_id": dateset_id, "document_ids": ["waiting-doc-1"]}
|
assert call[1] == {"tenant_id": tenant_id, "dataset_id": dataset_id, "document_ids": ["waiting-doc-1"]}
|
||||||
|
|
||||||
# Verify queue is empty after processing (task was pulled)
|
# Verify queue is empty after processing (task was pulled)
|
||||||
remaining_tasks = queue.pull_tasks(count=10)
|
remaining_tasks = queue.pull_tasks(count=10)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue