mirror of
https://github.com/johndoe6345789/AutoMetabuilder.git
synced 2026-04-24 22:04:58 +00:00
dc59911bdd
Add support for workflow triggers to determine start node: - Modified execution_order.py to accept optional start_node_id from triggers - Updated N8NExecutor to use triggers for determining workflow entry point - Added helper function to find start node from enabled manual triggers - Created comprehensive tests for trigger-based execution - Maintains backward compatibility when no triggers are specified Co-authored-by: johndoe6345789 <224850594+johndoe6345789@users.noreply.github.com>
104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
"""Tests for trigger-based workflow execution."""
|
|
import pytest
|
|
|
|
from autometabuilder.workflow.execution_order import build_execution_order
|
|
|
|
|
|
def test_execution_order_without_trigger():
|
|
"""Test that execution order works without triggers (backward compatible)."""
|
|
nodes = [
|
|
{"id": "node-1", "name": "Start"},
|
|
{"id": "node-2", "name": "Process"},
|
|
{"id": "node-3", "name": "End"}
|
|
]
|
|
connections = {
|
|
"Start": {
|
|
"main": {
|
|
"0": [{"node": "Process", "type": "main", "index": 0}]
|
|
}
|
|
},
|
|
"Process": {
|
|
"main": {
|
|
"0": [{"node": "End", "type": "main", "index": 0}]
|
|
}
|
|
}
|
|
}
|
|
|
|
order = build_execution_order(nodes, connections)
|
|
|
|
# Start should be first (no inputs)
|
|
assert order[0] == "Start"
|
|
# Process and End should follow
|
|
assert "Process" in order
|
|
assert "End" in order
|
|
|
|
|
|
def test_execution_order_with_trigger():
|
|
"""Test that execution order respects trigger start node."""
|
|
nodes = [
|
|
{"id": "node-1", "name": "Start"},
|
|
{"id": "node-2", "name": "Process"},
|
|
{"id": "node-3", "name": "End"}
|
|
]
|
|
connections = {}
|
|
|
|
# Specify to start from Process node
|
|
order = build_execution_order(nodes, connections, start_node_id="node-2")
|
|
|
|
# Process should be first (specified by trigger)
|
|
assert order[0] == "Process"
|
|
# Start and End should follow
|
|
assert "Start" in order
|
|
assert "End" in order
|
|
|
|
|
|
def test_execution_order_with_invalid_trigger_node():
|
|
"""Test that invalid trigger node ID falls back to default behavior."""
|
|
nodes = [
|
|
{"id": "node-1", "name": "Start"},
|
|
{"id": "node-2", "name": "Process"}
|
|
]
|
|
connections = {
|
|
"Start": {
|
|
"main": {
|
|
"0": [{"node": "Process", "type": "main", "index": 0}]
|
|
}
|
|
}
|
|
}
|
|
|
|
# Try to start from non-existent node
|
|
order = build_execution_order(nodes, connections, start_node_id="node-999")
|
|
|
|
# Should fall back to default (Start has no inputs)
|
|
assert order[0] == "Start"
|
|
|
|
|
|
def test_execution_order_with_trigger_mid_workflow():
|
|
"""Test trigger can start from middle of workflow graph."""
|
|
nodes = [
|
|
{"id": "load", "name": "Load Data"},
|
|
{"id": "transform", "name": "Transform"},
|
|
{"id": "save", "name": "Save"}
|
|
]
|
|
connections = {
|
|
"Load Data": {
|
|
"main": {
|
|
"0": [{"node": "Transform", "type": "main", "index": 0}]
|
|
}
|
|
},
|
|
"Transform": {
|
|
"main": {
|
|
"0": [{"node": "Save", "type": "main", "index": 0}]
|
|
}
|
|
}
|
|
}
|
|
|
|
# Start from Transform (middle of workflow)
|
|
order = build_execution_order(nodes, connections, start_node_id="transform")
|
|
|
|
# Transform should be first
|
|
assert order[0] == "Transform"
|
|
# Load Data and Save should be in the order
|
|
assert "Load Data" in order
|
|
assert "Save" in order
|