test(hitl): align service API continuation fixtures

This commit is contained in:
Yansong Zhang
2026-07-29 04:22:28 +08:00
parent 5f49fb4e4b
commit f0c38e7da1
@@ -27,6 +27,7 @@ from core.app.entities.queue_entities import QueueWorkflowPausedEvent
from core.app.entities.task_entities import (
AdvancedChatPausedBlockingResponse,
HumanInputRequiredResponse,
StreamEvent,
WorkflowAppPausedBlockingResponse,
WorkflowPauseStreamResponse,
)
@@ -319,7 +320,7 @@ class TestHitlServiceApi:
msg_generator.retrieve_events.assert_called_once_with(
AppMode.WORKFLOW,
"run-1",
terminal_events=[],
terminal_events=[StreamEvent.WORKFLOW_FINISHED],
)
workflow_generator.convert_to_event_stream.assert_called_once_with(["raw-event"])
@@ -370,6 +371,7 @@ class TestHitlServiceApi:
session_maker=ANY,
human_input_surface=HumanInputSurface.SERVICE_API,
close_on_pause=False,
cursor=None,
)
snapshot_session_maker = snapshot_builder.call_args.kwargs["session_maker"]
assert isinstance(snapshot_session_maker, sessionmaker)
@@ -455,7 +457,10 @@ class TestHitlServiceApi:
def test_advanced_chat_blocking_pipeline_pause_payload_contract(self) -> None:
from core.app.app_config.entities import AppAdditionalFeatures
from core.app.apps.advanced_chat.generate_task_pipeline import AdvancedChatAppGenerateTaskPipeline
from core.app.apps.advanced_chat.generate_task_pipeline import (
AdvancedChatAppGenerateTaskPipeline,
MessageSnapshot,
)
from models.enums import MessageStatus
from models.model import EndUser
@@ -485,10 +490,10 @@ class TestHitlServiceApi:
workflow=SimpleNamespace(id="workflow-id", tenant_id="tenant", features_dict={}),
queue_manager=SimpleNamespace(invoke_from=InvokeFrom.WEB_APP, graph_runtime_state=None),
conversation=SimpleNamespace(id="conv-id", mode=AppMode.ADVANCED_CHAT),
message=SimpleNamespace(
message=MessageSnapshot(
id="message-id",
query="hello",
created_at=datetime.utcnow(),
created_at=datetime.now(UTC),
status=MessageStatus.NORMAL,
answer="",
),