AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-18 21:35:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): handle invalid variable type for document_extractor_node

Browse Source 
- Imported StringVariable to properly mock an invalid variable type.
- Enhanced test case `test_run_invalid_variable_type` to use StringVariable instead of a plain string.
This commit is contained in:
-LAN- 2024-09-30 18:02:37 +08:00
parent 94b946c715
commit f9baabc9f9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/tests/unit_tests/core/workflow/nodes/test_document_extractor_node.py
View File

@ -6,6 +6,7 @@ from werkzeug.datastructures import FileStorage
from core.file import File, FileTransferMethod from core.file import File, FileTransferMethod
from core.variables import ArrayFileSegment from core.variables import ArrayFileSegment
from core.variables.variables import StringVariable
from core.workflow.entities.node_entities import NodeRunResult from core.workflow.entities.node_entities import NodeRunResult
from core.workflow.nodes.document_extractor import DocumentExtractorNode, DocumentExtractorNodeData from core.workflow.nodes.document_extractor import DocumentExtractorNode, DocumentExtractorNodeData
from core.workflow.nodes.document_extractor.document_extractor_node import ( from core.workflow.nodes.document_extractor.document_extractor_node import (
@ -51,7 +52,9 @@ def test_run_variable_not_found(document_extractor_node, mock_graph_runtime_stat
def test_run_invalid_variable_type(document_extractor_node, mock_graph_runtime_state): def test_run_invalid_variable_type(document_extractor_node, mock_graph_runtime_state):
document_extractor_node.graph_runtime_state = mock_graph_runtime_state document_extractor_node.graph_runtime_state = mock_graph_runtime_state
mock_graph_runtime_state.variable_pool.get.return_value = "Not an ArrayFileSegment" mock_graph_runtime_state.variable_pool.get.return_value = StringVariable(
value="Not an ArrayFileSegment", name="test"
)
result = document_extractor_node._run() result = document_extractor_node._run()