53 lines
3.6 KiB
Python
53 lines
3.6 KiB
Python
import unittest
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, Mock
|
|
from models import AssistantConfig, RuntimeTool, RuntimeToolBinding
|
|
from services.brains.prompt_tasks import PromptTaskBridge
|
|
from services.tool_executor import ToolExecutor
|
|
from services.runtime_variables import DynamicVariableStore
|
|
|
|
|
|
class PromptTaskBridgeTest(unittest.IsolatedAsyncioTestCase):
|
|
async def test_receipt_and_single_terminal_notification(self):
|
|
cfg = AssistantConfig(conversation_id="conv_1", dynamic_variables={"system__agent_id": "asst_1"})
|
|
runtime = SimpleNamespace(session_id="conv_1", transport=object(), tool_executor_factory=None, call_end=SimpleNamespace(ending=False), queue_frame=AsyncMock(), llm=SimpleNamespace(queue_frame=AsyncMock()))
|
|
detach = Mock()
|
|
service = SimpleNamespace(subscribe=Mock(return_value=detach), submit=AsyncMock(return_value={"id": "task_1", "status": "queued"}))
|
|
bridge = PromptTaskBridge(cfg, runtime, ToolExecutor(DynamicVariableStore({})), service)
|
|
tool = RuntimeTool(id="tool_task", name="coding", function_name="coding_task", type="task", definition={"type": "task", "config": {}})
|
|
binding = RuntimeToolBinding(toolId=tool.id, argumentBindings={"instruction": {"source": "model"}})
|
|
functions = bridge.functions(tool, binding)
|
|
callback = AsyncMock()
|
|
await functions[0][1](SimpleNamespace(arguments={"instruction": "修复测试"}, tool_call_id="call_1", result_callback=callback))
|
|
self.assertEqual(callback.await_args.args[0]["status"], "accepted")
|
|
task = {"id": "task_1", "sequence": 3, "updatedAt": "2026-10-23", "inputRequest": {}, "status": "succeeded", "summary": "完成", "result": "测试通过"}
|
|
await bridge.on_task(task)
|
|
await bridge.on_task(task)
|
|
runtime.llm.queue_frame.assert_awaited_once()
|
|
self.assertFalse(runtime.llm.queue_frame.await_args.args[0].interruptible)
|
|
bridge.close()
|
|
detach.assert_called_once()
|
|
|
|
async def test_test_runtime_never_submits_a_real_task(self):
|
|
runtime = SimpleNamespace(session_id="conv_1", transport=None, tool_executor_factory=object())
|
|
service = SimpleNamespace(submit=AsyncMock())
|
|
bridge = PromptTaskBridge(AssistantConfig(), runtime, ToolExecutor(DynamicVariableStore({})), service)
|
|
tool = RuntimeTool(id="tool_task", name="coding", function_name="coding_task", type="task", definition={"type": "task", "config": {}})
|
|
callback = AsyncMock()
|
|
await bridge.functions(tool, None)[0][1](SimpleNamespace(result_callback=callback))
|
|
service.submit.assert_not_awaited()
|
|
self.assertEqual(callback.await_args.args[0]["status"], "error")
|
|
|
|
async def test_test_runtime_never_controls_an_existing_real_task(self):
|
|
runtime = SimpleNamespace(session_id="conv_1", transport=None, tool_executor_factory=object())
|
|
service = SimpleNamespace(get=AsyncMock(), cancel=AsyncMock(), send_input=AsyncMock())
|
|
bridge = PromptTaskBridge(AssistantConfig(), runtime, ToolExecutor(DynamicVariableStore({})), service)
|
|
tool = RuntimeTool(id="tool_task", name="coding", function_name="coding_task", type="task", definition={"type": "task", "config": {}})
|
|
for schema, handler in bridge.functions(tool, None)[1:]:
|
|
callback = AsyncMock()
|
|
await handler(SimpleNamespace(function_name=schema.name, arguments={"task_id": "real_task", "message": "yes"}, result_callback=callback))
|
|
self.assertEqual(callback.await_args.args[0]["status"], "error")
|
|
service.get.assert_not_awaited()
|
|
service.cancel.assert_not_awaited()
|
|
service.send_input.assert_not_awaited()
|