xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-13 01:06:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: harden agent runner runtime boundaries

Browse Source
This commit is contained in:
huanghuoguoguo
2026-06-13 00:17:40 +08:00
parent 897a708a13
commit 1113bafe28
24 changed files with 400 additions and 897 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/unit_tests/api/service/test_mcp_service.py
View File

@@ -630,7 +630,9 @@ class TestMCPServiceTestMCPServer:
ap.tool_mgr.mcp_tool_loader = SimpleNamespace()
mock_session = MagicMock()
mock_session.server_name = 'transient-test-server'
mock_session.start = AsyncMock()
mock_session.shutdown = AsyncMock()
ap.tool_mgr.mcp_tool_loader.load_mcp_server = AsyncMock(return_value=mock_session)
ap.task_mgr = SimpleNamespace()
@@ -645,4 +647,9 @@ class TestMCPServiceTestMCPServer:
# Verify - load_mcp_server called
ap.tool_mgr.mcp_tool_loader.load_mcp_server.assert_called_once()
assert task_id == 456
assert task_id == 456
coroutine = ap.task_mgr.create_user_task.call_args.args[0]
await coroutine
mock_session.start.assert_awaited_once()
mock_session.shutdown.assert_awaited_once()