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

refactor(api): move workflow knowledge nodes and trigger nodes (#33445)

Browse Source
This commit is contained in:
-LAN-
2026-03-15 15:24:59 +08:00
committed by GitHub
parent 1b6e695520
commit fb41b215c8
232 changed files with 1575 additions and 1421 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
api/tests/unit_tests/core/trigger/debug/test_debug_event_selectors.py
View File

@ -13,6 +13,11 @@ from unittest.mock import MagicMock, patch
import pytest
from core.plugin.entities.request import TriggerInvokeEventResponse
from core.trigger.constants import (
TRIGGER_PLUGIN_NODE_TYPE,
TRIGGER_SCHEDULE_NODE_TYPE,
TRIGGER_WEBHOOK_NODE_TYPE,
)
from core.trigger.debug.event_selectors import (
PluginTriggerDebugEventPoller,
ScheduleTriggerDebugEventPoller,
@ -21,7 +26,7 @@ from core.trigger.debug.event_selectors import (
select_trigger_debug_events,
)
from core.trigger.debug.events import PluginTriggerDebugEvent, WebhookDebugEvent
from dify_graph.enums import NodeType
from dify_graph.enums import BuiltinNodeTypes, NodeType
from tests.unit_tests.core.trigger.conftest import VALID_PROVIDER_ID
@ -215,24 +220,24 @@ class TestCreateEventPoller:
return wf
def test_creates_plugin_poller(self):
wf = self._workflow_with_node(NodeType.TRIGGER_PLUGIN)
wf = self._workflow_with_node(TRIGGER_PLUGIN_NODE_TYPE)
poller = create_event_poller(wf, "t1", "u1", "a1", "n1")
assert isinstance(poller, PluginTriggerDebugEventPoller)
def test_creates_webhook_poller(self):
wf = self._workflow_with_node(NodeType.TRIGGER_WEBHOOK)
wf = self._workflow_with_node(TRIGGER_WEBHOOK_NODE_TYPE)
poller = create_event_poller(wf, "t1", "u1", "a1", "n1")
assert isinstance(poller, WebhookTriggerDebugEventPoller)
def test_creates_schedule_poller(self):
wf = self._workflow_with_node(NodeType.TRIGGER_SCHEDULE)
wf = self._workflow_with_node(TRIGGER_SCHEDULE_NODE_TYPE)
poller = create_event_poller(wf, "t1", "u1", "a1", "n1")
assert isinstance(poller, ScheduleTriggerDebugEventPoller)
def test_raises_for_unknown_type(self):
wf = MagicMock()
wf.get_node_config_by_id.return_value = {"data": {}}
wf.get_node_type_from_node_config.return_value = NodeType.START
wf.get_node_type_from_node_config.return_value = BuiltinNodeTypes.START
with pytest.raises(ValueError):
create_event_poller(wf, "t1", "u1", "a1", "n1")
@ -249,7 +254,7 @@ class TestSelectTriggerDebugEvents:
def test_returns_first_non_none_event(self):
wf = MagicMock()
wf.get_node_config_by_id.return_value = {"data": {}}
wf.get_node_type_from_node_config.return_value = NodeType.TRIGGER_WEBHOOK
wf.get_node_type_from_node_config.return_value = TRIGGER_WEBHOOK_NODE_TYPE
app_model = MagicMock()
app_model.tenant_id = "t1"
app_model.id = "a1"
@ -265,7 +270,7 @@ class TestSelectTriggerDebugEvents:
def test_returns_none_when_no_events(self):
wf = MagicMock()
wf.get_node_config_by_id.return_value = {"data": {}}
wf.get_node_type_from_node_config.return_value = NodeType.TRIGGER_WEBHOOK
wf.get_node_type_from_node_config.return_value = TRIGGER_WEBHOOK_NODE_TYPE
app_model = MagicMock()
app_model.tenant_id = "t1"
app_model.id = "a1"