From b67a32db972496a3263c7fdf16f7acf993e941a4 Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Mon, 23 Jun 2025 10:54:25 +0800 Subject: [PATCH] test(api): fix tests broken by code adjustment --- .../services/workflow/test_workflow_draft_variable_service.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/tests/unit_tests/services/workflow/test_workflow_draft_variable_service.py b/api/tests/unit_tests/services/workflow/test_workflow_draft_variable_service.py index f85570ea78..8ae69c8d64 100644 --- a/api/tests/unit_tests/services/workflow/test_workflow_draft_variable_service.py +++ b/api/tests/unit_tests/services/workflow/test_workflow_draft_variable_service.py @@ -7,6 +7,7 @@ import pytest from sqlalchemy.orm import Session 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.nodes import NodeType from models.enums import DraftVariableType @@ -175,6 +176,7 @@ class TestWorkflowDraftVariableService: mock_variable.id = "var-id" mock_variable.name = "test_var" mock_variable.node_id = "node-id" + mock_variable.value_type = SegmentType.STRING # Create mock execution record mock_execution = Mock(spec=WorkflowNodeExecutionModel)