test(api): fix tests broken by code adjustment

pull/20699/head
QuantumGhost 11 months ago
parent fbc9480343
commit b67a32db97

@ -7,6 +7,7 @@ import pytest
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from core.app.entities.app_invoke_entities import InvokeFrom from core.app.entities.app_invoke_entities import InvokeFrom
from core.variables.types import SegmentType
from core.workflow.constants import SYSTEM_VARIABLE_NODE_ID from core.workflow.constants import SYSTEM_VARIABLE_NODE_ID
from core.workflow.nodes import NodeType from core.workflow.nodes import NodeType
from models.enums import DraftVariableType from models.enums import DraftVariableType
@ -175,6 +176,7 @@ class TestWorkflowDraftVariableService:
mock_variable.id = "var-id" mock_variable.id = "var-id"
mock_variable.name = "test_var" mock_variable.name = "test_var"
mock_variable.node_id = "node-id" mock_variable.node_id = "node-id"
mock_variable.value_type = SegmentType.STRING
# Create mock execution record # Create mock execution record
mock_execution = Mock(spec=WorkflowNodeExecutionModel) mock_execution = Mock(spec=WorkflowNodeExecutionModel)

Loading…
Cancel
Save