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

test: added test cases for core.workflow module (#33126)

Browse Source
This commit is contained in:
Rajat Agarwal
2026-03-12 13:05:25 +05:30
committed by GitHub
parent 157208ab1e
commit dc50e4c4f2
6 changed files with 1266 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/tests/unit_tests/services/workflow/test_workflow_service.py
View File

@ -311,7 +311,9 @@ class TestWorkflowService:
mock_workflow.conversation_variables = []
# Mock node config
mock_workflow.get_node_config_by_id.return_value = {"id": "node-1", "data": {"type": "llm"}}
mock_workflow.get_node_config_by_id.return_value = NodeConfigDictAdapter.validate_python(
{"id": "node-1", "data": {"type": NodeType.LLM.value}}
)
mock_workflow.get_enclosing_node_type_and_id.return_value = None
# Mock class methods
@ -376,7 +378,9 @@ class TestWorkflowService:
mock_workflow.tenant_id = "tenant-1"
mock_workflow.environment_variables = []
mock_workflow.conversation_variables = []
mock_workflow.get_node_config_by_id.return_value = {"id": "node-1", "data": {"type": "llm"}}
mock_workflow.get_node_config_by_id.return_value = NodeConfigDictAdapter.validate_python(
{"id": "node-1", "data": {"type": NodeType.LLM.value}}
)
mock_workflow.get_enclosing_node_type_and_id.return_value = None
monkeypatch.setattr(workflow_service_module, "WorkflowDraftVariableService", MagicMock())