From 23e59c677803520386092c2cce24bea8493d89c3 Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Thu, 7 May 2026 10:47:44 +0800 Subject: [PATCH] test(api): update import names in tests Upstream project graphon renamed some classes. Modify the tests to keep import names consistent with upstream. --- ...qlalchemy_execution_extra_content_repository.py | 4 ++-- .../repositories/test_human_input_repository.py | 2 +- .../test_parallel_human_input_join_resume.py | 14 +++++++++++--- .../unit_tests/core/workflow/test_node_runtime.py | 5 +++-- 4 files changed, 17 insertions(+), 8 deletions(-) diff --git a/api/tests/test_containers_integration_tests/repositories/test_sqlalchemy_execution_extra_content_repository.py b/api/tests/test_containers_integration_tests/repositories/test_sqlalchemy_execution_extra_content_repository.py index c074e9f8ee..fbc171a303 100644 --- a/api/tests/test_containers_integration_tests/repositories/test_sqlalchemy_execution_extra_content_repository.py +++ b/api/tests/test_containers_integration_tests/repositories/test_sqlalchemy_execution_extra_content_repository.py @@ -15,7 +15,7 @@ import pytest from sqlalchemy import Engine, delete, select from sqlalchemy.orm import Session, sessionmaker -from graphon.nodes.human_input.entities import FormDefinition, UserAction +from graphon.nodes.human_input.entities import FormDefinition, UserActionConfig from graphon.nodes.human_input.enums import HumanInputFormStatus from libs.datetime_utils import naive_utc_now from models.account import Account, Tenant, TenantAccountJoin, TenantAccountRole @@ -212,7 +212,7 @@ def _create_waiting_form( form_definition = FormDefinition( form_content="content", inputs=[], - user_actions=[UserAction(id="approve", title="Approve")], + user_actions=[UserActionConfig(id="approve", title="Approve")], rendered_content="rendered", expiration_time=expiration_time, default_values=default_values or {"name": "John"}, diff --git a/api/tests/unit_tests/core/repositories/test_human_input_repository.py b/api/tests/unit_tests/core/repositories/test_human_input_repository.py index 1f9e1b21f8..418537675d 100644 --- a/api/tests/unit_tests/core/repositories/test_human_input_repository.py +++ b/api/tests/unit_tests/core/repositories/test_human_input_repository.py @@ -467,7 +467,7 @@ def test_create_form_adds_console_and_backstage_recipients(monkeypatch: pytest.M delivery_methods=[], form_content="hello", inputs=[], - user_actions=[UserAction(id="submit", title="Submit")], + user_actions=[UserActionConfig(id="submit", title="Submit")], ) params = FormCreateParams( workflow_execution_id=None, diff --git a/api/tests/unit_tests/core/workflow/graph_engine/test_parallel_human_input_join_resume.py b/api/tests/unit_tests/core/workflow/graph_engine/test_parallel_human_input_join_resume.py index 624c1fb01d..4ffd377967 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/test_parallel_human_input_join_resume.py +++ b/api/tests/unit_tests/core/workflow/graph_engine/test_parallel_human_input_join_resume.py @@ -12,6 +12,7 @@ from core.repositories.human_input_repository import ( from core.workflow.node_runtime import DifyHumanInputNodeRuntime from core.workflow.system_variables import build_system_variables from graphon.entities import WorkflowStartReason +from graphon.file import File, FileTransferMethod, FileType from graphon.graph import Graph from graphon.graph_engine import GraphEngine, GraphEngineConfig from graphon.graph_engine.command_channels import InMemoryChannel @@ -32,7 +33,7 @@ from graphon.nodes.human_input.entities import ( StringListSource, UserActionConfig, ) -from graphon.nodes.human_input.enums import HumanInputFormStatus +from graphon.nodes.human_input.enums import HumanInputFormStatus, ValueSourceType from graphon.nodes.human_input.human_input_node import HumanInputNode from graphon.nodes.start.entities import StartNodeData from graphon.nodes.start.start_node import StartNode @@ -155,8 +156,15 @@ def _build_graph(runtime_state: GraphRuntimeState, repo: HumanInputFormRepositor human_data = HumanInputNodeData( title="Human Input", form_content="Human input required", - inputs=[], - user_actions=[UserAction(id="approve", title="Approve")], + inputs=[ + SelectInputConfig( + output_variable_name="decision", + option_source=StringListSource(type=ValueSourceType.CONSTANT, value=["approve", "reject"]), + ), + FileInputConfig(output_variable_name="attachment"), + FileListInputConfig(output_variable_name="attachments", number_limits=2), + ], + user_actions=[UserActionConfig(id="approve", title="Approve")], ) human_a_config = {"id": "human_a", "data": human_data.model_dump()} diff --git a/api/tests/unit_tests/core/workflow/test_node_runtime.py b/api/tests/unit_tests/core/workflow/test_node_runtime.py index 5a43369a1a..1389ea3957 100644 --- a/api/tests/unit_tests/core/workflow/test_node_runtime.py +++ b/api/tests/unit_tests/core/workflow/test_node_runtime.py @@ -26,10 +26,11 @@ from core.workflow.node_runtime import ( build_dify_llm_file_saver, resolve_dify_run_context, ) -from graphon.file import FileTransferMethod, FileType +from graphon.file import File, FileTransferMethod, FileType from graphon.model_runtime.entities.common_entities import I18nObject from graphon.model_runtime.entities.model_entities import AIModelEntity, FetchFrom, ModelType -from graphon.nodes.human_input.entities import HumanInputNodeData +from graphon.nodes.human_input.entities import FileInputConfig, FileListInputConfig, HumanInputNodeData +from graphon.variables.segments import ArrayFileSegment, FileSegment from tests.workflow_test_utils import build_test_run_context