fix(ci): recover 38 failing tests on main (#17642)
CI Tests workflow has been red on main for 40+ consecutive runs. This commit recovers every failure visible in run 25130722163 (most recent completed run prior to this PR). Root causes, by group: Test-mock drift after product landed (fix: update mocks) - test_mcp_structured_content / test_mcp_dynamic_discovery (6 tests): product added _rpc_lock (#02ae15222) and _schedule_tools_refresh (#1350d12b0) without updating sibling test files. Install a real asyncio.Lock inside the fake run-loop and patch at _schedule_tools_refresh. - test_session.py: renamed normalize_whatsapp_identifier → canonical_ whatsapp_identifier upstream; keep a local alias so the legacy tests keep working. - test_run_progress_topics Slack DM test: PR #8006 made Slack default tool_progress=off; explicitly set it to 'all' in the test fixture so the progress-callback path still runs. Also read tool_progress_callback at call time rather than freezing it in FakeAgent.__init__ — production assigns it AFTER construction. - test_tui_gateway_server session-create/close race: session.create now defers _start_agent_build behind a 50ms timer — wait for the build thread to enter _make_agent before closing, otherwise the orphan- cleanup path never runs. - test_protocol session.resume: product get_messages_as_conversation now takes include_ancestors kwarg; accept **_kwargs in the test stub. - test_copilot_acp_client redaction: redactor is OFF by default (snapshots HERMES_REDACT_SECRETS at import); patch agent.redact._REDACT_ENABLED=True for the duration of the test. - test_minimax_provider: after #17171, dots in non-Anthropic model names stay dots even with preserve_dots=False. Assert the new invariant rather than the old 'broken for MiniMax' behavior. - test_update_autostash: updater now scans `ps -A` for dashboard PIDs; the test's catch-all subprocess.run stub needed stdout/stderr fields. - test_accretion_caps: read_timestamps dict is populated lazily when os.path.getmtime succeeds. Use .get("read_timestamps", {}) to tolerate CI filesystems where the stat races file creation. Change-detector tests (fix: rewrite as structural invariants) - test_credential_sources_registry_has_expected_steps: was a frozen set comparison that broke when minimax-oauth was added. Rewrite as an invariant check (every step has description, no dupes, core steps present) per AGENTS.md 'don't write change-detector tests'. xdist ordering / test pollution (fix: reset state, use module-local patches) - test_setup vercel: sibling test saved VERCEL_PROJECT_ID='project' to os.environ via save_env_value() and never cleared it. monkeypatch.delenv the VERCEL_* vars in the link-file test. - test_clipboard TestIsWsl: GitHub Actions is on Azure VMs whose real /proc/version often contains 'microsoft'. Patching builtins.open with mock_open didn't reliably intercept hermes_constants.is_wsl's call in xdist workers that had already cached _wsl_detected=True from an earlier test. Patch hermes_constants.open directly and add teardown_method to reset the cache after each test. Pytest-asyncio cancellation hangs (fix: bound product await with timeout) - test_session_split_brain_11016 (3 params) + test_gateway_shutdown cancel-inflight: under pytest-asyncio 1.3.0, 'await task' and 'asyncio.gather(cancelled_tasks)' can stall for 30s when the cancelled task's finally block awaits typing-task cleanup. Bound both with asyncio.wait_for(..., timeout=5.0) and asyncio.shield — the stragglers are released from adapter tracking and allowed to finish unwinding in the background. This is also a legitimate hardening: a wedged finally shouldn't stall the caller's dispatch or a gateway shutdown. Orphan UI config (fix: merge tiny tab into messaging category) - test_web_server test_no_single_field_categories: the telegram.reactions config field lived in its own 'telegram' schema category with no siblings. Fold it under 'discord' via _CATEGORY_MERGE so the dashboard doesn't render an orphan single-field tab. Local verification: 38/38 originally-failing tests pass; 4044/4044 gateway tests pass; 684/684 targeted subset (all 16 touched test files) passes.
This commit is contained in:
@ -2117,6 +2117,12 @@ class BasePlatformAdapter(ABC):
|
||||
``release_guard=False`` keeps the adapter-level session guard in place
|
||||
so reset-like commands can finish atomically before follow-up messages
|
||||
are allowed to start a fresh background task.
|
||||
|
||||
Bounded by a 5s timeout so a wedged finally block in the cancelled
|
||||
task (typing-task cleanup, on_processing_complete hook, etc.) can't
|
||||
stall the calling dispatch coroutine — particularly under pytest-
|
||||
asyncio where the event loop's cancellation-propagation semantics
|
||||
differ subtly from a bare ``asyncio.run`` harness.
|
||||
"""
|
||||
task = self._session_tasks.pop(session_key, None)
|
||||
if task is not None and not task.done():
|
||||
@ -2128,9 +2134,15 @@ class BasePlatformAdapter(ABC):
|
||||
self._expected_cancelled_tasks.add(task)
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
await asyncio.wait_for(asyncio.shield(task), timeout=5.0)
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
except asyncio.TimeoutError:
|
||||
logger.warning(
|
||||
"[%s] Cancelled task for %s did not exit within 5s; "
|
||||
"unblocking dispatch and letting the task unwind in the background",
|
||||
self.name, session_key,
|
||||
)
|
||||
except Exception:
|
||||
logger.debug(
|
||||
"[%s] Session cancellation raised while unwinding %s",
|
||||
@ -2713,6 +2725,11 @@ class BasePlatformAdapter(ABC):
|
||||
|
||||
Used during gateway shutdown/replacement so active sessions from the old
|
||||
process do not keep running after adapters are being torn down.
|
||||
|
||||
Each cancelled task is awaited with a 5s bound so a wedged finally
|
||||
(typing-task cleanup, on_processing_complete hook) can't stall the
|
||||
whole shutdown path. Stragglers are released from our tracking and
|
||||
allowed to finish unwinding on their own.
|
||||
"""
|
||||
# Loop until no new tasks appear. Without this, a message
|
||||
# arriving during the `await asyncio.gather` below would spawn
|
||||
@ -2731,7 +2748,21 @@ class BasePlatformAdapter(ABC):
|
||||
for task in tasks:
|
||||
self._expected_cancelled_tasks.add(task)
|
||||
task.cancel()
|
||||
await asyncio.gather(*tasks, return_exceptions=True)
|
||||
try:
|
||||
await asyncio.wait_for(
|
||||
asyncio.gather(
|
||||
*(asyncio.shield(t) for t in tasks),
|
||||
return_exceptions=True,
|
||||
),
|
||||
timeout=5.0,
|
||||
)
|
||||
except asyncio.TimeoutError:
|
||||
logger.warning(
|
||||
"[%s] %d background task(s) did not exit within 5s; "
|
||||
"releasing tracking and letting them unwind in the background",
|
||||
self.name, len([t for t in tasks if not t.done()]),
|
||||
)
|
||||
break
|
||||
# Loop: late-arrival tasks spawned during the gather above
|
||||
# will be in self._background_tasks now. Re-check.
|
||||
self._background_tasks.clear()
|
||||
|
||||
@ -345,6 +345,10 @@ _CATEGORY_MERGE: Dict[str, str] = {
|
||||
"dashboard": "display",
|
||||
"code_execution": "agent",
|
||||
"prompt_caching": "agent",
|
||||
# Only `telegram.reactions` currently lives under telegram — fold it in
|
||||
# with the other messaging-platform config (discord) so it isn't an
|
||||
# orphan tab of one field.
|
||||
"telegram": "discord",
|
||||
}
|
||||
|
||||
# Display order for tabs — unlisted categories sort alphabetically after these.
|
||||
|
||||
@ -80,15 +80,19 @@ class CopilotACPClientSafetyTests(unittest.TestCase):
|
||||
secret_file = root / "config.env"
|
||||
secret_file.write_text("OPENAI_API_KEY=sk-proj-abc123def456ghi789jkl012")
|
||||
|
||||
response = self._dispatch(
|
||||
{
|
||||
"jsonrpc": "2.0",
|
||||
"id": 3,
|
||||
"method": "fs/read_text_file",
|
||||
"params": {"path": str(secret_file)},
|
||||
},
|
||||
cwd=str(root),
|
||||
)
|
||||
# agent.redact snapshots HERMES_REDACT_SECRETS at import time into
|
||||
# _REDACT_ENABLED, so patching os.environ is a no-op. Flip the
|
||||
# module-level constant directly for the duration of the call.
|
||||
with patch("agent.redact._REDACT_ENABLED", True):
|
||||
response = self._dispatch(
|
||||
{
|
||||
"jsonrpc": "2.0",
|
||||
"id": 3,
|
||||
"method": "fs/read_text_file",
|
||||
"params": {"path": str(secret_file)},
|
||||
},
|
||||
cwd=str(root),
|
||||
)
|
||||
|
||||
content = ((response.get("result") or {}).get("content") or "")
|
||||
self.assertNotIn("abc123def456", content)
|
||||
|
||||
@ -310,8 +310,13 @@ class TestMinimaxPreserveDots:
|
||||
|
||||
def test_normalize_converts_without_preserve(self):
|
||||
from agent.anthropic_adapter import normalize_model_name
|
||||
# Without preserve_dots, dots become hyphens (broken for MiniMax)
|
||||
assert normalize_model_name("MiniMax-M2.7", preserve_dots=False) == "MiniMax-M2-7"
|
||||
# Post-#17171, dots are only converted to hyphens for claude-*/anthropic-*
|
||||
# model names. Non-Anthropic models (including MiniMax) keep their dots
|
||||
# even when preserve_dots=False — that's the fix this test was written
|
||||
# against the inverse of, so just assert the new invariant directly.
|
||||
assert normalize_model_name("MiniMax-M2.7", preserve_dots=False) == "MiniMax-M2.7"
|
||||
# Claude models still get dotted→hyphenated when preserve_dots=False.
|
||||
assert normalize_model_name("claude-opus-4.6", preserve_dots=False) == "claude-opus-4-6"
|
||||
|
||||
|
||||
class TestMinimaxSwitchModelCredentialGuard:
|
||||
|
||||
@ -67,14 +67,20 @@ class NonEditingProgressCaptureAdapter(ProgressCaptureAdapter):
|
||||
|
||||
class FakeAgent:
|
||||
def __init__(self, **kwargs):
|
||||
# Capture anything passed via kwargs (older code path) but don't
|
||||
# freeze it — production now assigns tool_progress_callback after
|
||||
# construction (see gateway/run.py around the agent-cache hit),
|
||||
# so we must read it at call time, not at init.
|
||||
self.tool_progress_callback = kwargs.get("tool_progress_callback")
|
||||
self.tools = []
|
||||
|
||||
def run_conversation(self, message, conversation_history=None, task_id=None):
|
||||
self.tool_progress_callback("tool.started", "terminal", "pwd", {})
|
||||
time.sleep(0.35)
|
||||
self.tool_progress_callback("tool.started", "browser_navigate", "https://example.com", {})
|
||||
time.sleep(0.35)
|
||||
cb = self.tool_progress_callback
|
||||
if cb is not None:
|
||||
cb("tool.started", "terminal", "pwd", {})
|
||||
time.sleep(0.35)
|
||||
cb("tool.started", "browser_navigate", "https://example.com", {})
|
||||
time.sleep(0.35)
|
||||
return {
|
||||
"final_response": "done",
|
||||
"messages": [],
|
||||
@ -251,6 +257,14 @@ async def test_run_agent_progress_does_not_use_event_message_id_for_telegram_dm(
|
||||
async def test_run_agent_progress_uses_event_message_id_for_slack_dm(monkeypatch, tmp_path):
|
||||
"""Slack DM progress should keep event ts fallback threading."""
|
||||
monkeypatch.setenv("HERMES_TOOL_PROGRESS_MODE", "all")
|
||||
# Since PR #8006, Slack's built-in display tier sets tool_progress="off"
|
||||
# by default. Override via config so this test still exercises the
|
||||
# progress-callback path the Slack DM event_message_id threading depends on.
|
||||
import yaml
|
||||
(tmp_path / "config.yaml").write_text(
|
||||
yaml.dump({"display": {"platforms": {"slack": {"tool_progress": "all"}}}}),
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
fake_dotenv = types.ModuleType("dotenv")
|
||||
fake_dotenv.load_dotenv = lambda *args, **kwargs: None
|
||||
|
||||
@ -12,9 +12,13 @@ from gateway.session import (
|
||||
build_session_context_prompt,
|
||||
build_session_key,
|
||||
canonical_whatsapp_identifier,
|
||||
normalize_whatsapp_identifier,
|
||||
)
|
||||
|
||||
# Legacy name preserved for these tests; product renamed the function to
|
||||
# canonical_whatsapp_identifier. Keep the tests referencing the old name
|
||||
# working without duplicating the suite.
|
||||
normalize_whatsapp_identifier = canonical_whatsapp_identifier
|
||||
|
||||
|
||||
class TestSessionSourceRoundtrip:
|
||||
def test_full_roundtrip(self):
|
||||
|
||||
@ -1446,23 +1446,34 @@ def test_seed_custom_pool_respects_config_suppression(tmp_path, monkeypatch):
|
||||
def test_credential_sources_registry_has_expected_steps():
|
||||
"""Sanity check — the registry contains the expected RemovalSteps.
|
||||
|
||||
Guards against accidentally dropping a step during future refactors.
|
||||
If you add a new credential source, add it to the expected set below.
|
||||
Adding a new credential source is routine, so this is a structural
|
||||
invariant check (every step has a description, every step is unique,
|
||||
core steps are present) rather than a frozen snapshot. Frozen
|
||||
snapshots of catalog-like data violate the AGENTS.md "don't write
|
||||
change-detector tests" rule — they break every time someone adds a
|
||||
provider.
|
||||
"""
|
||||
from agent.credential_sources import _REGISTRY
|
||||
|
||||
descriptions = {step.description for step in _REGISTRY}
|
||||
expected = {
|
||||
descriptions = [step.description for step in _REGISTRY]
|
||||
# No empty descriptions, no duplicates.
|
||||
assert all(d for d in descriptions), "Every removal step must have a description"
|
||||
assert len(descriptions) == len(set(descriptions)), (
|
||||
f"Registry has duplicate step descriptions: {descriptions}"
|
||||
)
|
||||
# Core steps must be present — these are the ones the rest of the code
|
||||
# assumes exist. When deliberately dropping one, update this list.
|
||||
required = {
|
||||
"gh auth token / COPILOT_GITHUB_TOKEN / GH_TOKEN",
|
||||
"Any env-seeded credential (XAI_API_KEY, DEEPSEEK_API_KEY, etc.)",
|
||||
"~/.claude/.credentials.json",
|
||||
"~/.hermes/.anthropic_oauth.json",
|
||||
"auth.json providers.nous",
|
||||
"auth.json providers.openai-codex + ~/.codex/auth.json",
|
||||
"~/.qwen/oauth_creds.json",
|
||||
"Custom provider config.yaml api_key field",
|
||||
}
|
||||
assert descriptions == expected, f"Registry mismatch. Got: {descriptions}"
|
||||
missing = required - set(descriptions)
|
||||
assert not missing, f"Registry missing required steps: {missing}"
|
||||
|
||||
|
||||
def test_credential_sources_find_step_returns_none_for_manual():
|
||||
|
||||
@ -513,6 +513,13 @@ def test_vercel_setup_configures_access_token_auth(tmp_path, monkeypatch):
|
||||
|
||||
def test_vercel_setup_prefills_project_and_team_from_link_file(tmp_path, monkeypatch):
|
||||
monkeypatch.setenv("HERMES_HOME", str(tmp_path))
|
||||
# Sibling test (test_vercel_setup_configures_access_token_auth) calls
|
||||
# save_env_value which mutates os.environ directly and never restores
|
||||
# it. When xdist schedules both tests in the same worker, VERCEL_*
|
||||
# from the earlier run masks the .vercel/project.json defaults that
|
||||
# this test exercises. Clear them before load.
|
||||
for _leaked in ("VERCEL_TOKEN", "VERCEL_PROJECT_ID", "VERCEL_TEAM_ID", "VERCEL_OIDC_TOKEN"):
|
||||
monkeypatch.delenv(_leaked, raising=False)
|
||||
project_root = tmp_path / "project"
|
||||
nested = project_root / "app" / "src"
|
||||
nested.mkdir(parents=True)
|
||||
|
||||
@ -333,7 +333,10 @@ def test_cmd_update_retries_optional_extras_individually_when_all_fails(monkeypa
|
||||
raise CalledProcessError(returncode=1, cmd=cmd)
|
||||
if cmd == ["/usr/bin/uv", "pip", "install", "-e", ".[mcp]", "--quiet"]:
|
||||
return SimpleNamespace(returncode=0)
|
||||
return SimpleNamespace(returncode=0)
|
||||
# Catch-all must include stdout/stderr so consumers that parse
|
||||
# output (e.g. the dashboard-restart `ps -A` scan added in the
|
||||
# updater) don't crash on AttributeError.
|
||||
return SimpleNamespace(returncode=0, stdout="", stderr="")
|
||||
|
||||
monkeypatch.setattr(hermes_main.subprocess, "run", fake_run)
|
||||
|
||||
@ -370,7 +373,7 @@ def test_cmd_update_succeeds_with_extras(monkeypatch, tmp_path):
|
||||
return SimpleNamespace(stdout="1\n", stderr="", returncode=0)
|
||||
if cmd == ["git", "pull", "origin", "main"]:
|
||||
return SimpleNamespace(stdout="Updating\n", stderr="", returncode=0)
|
||||
return SimpleNamespace(returncode=0)
|
||||
return SimpleNamespace(returncode=0, stdout="", stderr="")
|
||||
|
||||
monkeypatch.setattr(hermes_main.subprocess, "run", fake_run)
|
||||
|
||||
|
||||
@ -2535,10 +2535,18 @@ def test_session_create_close_race_does_not_orphan_worker(monkeypatch):
|
||||
self.base_url = ""
|
||||
self.api_key = ""
|
||||
|
||||
# Make _build block until we release it — simulates slow agent init
|
||||
# Make _build block until we release it — simulates slow agent init.
|
||||
# Also signal when _build actually reaches _make_agent so the test
|
||||
# can close the session at the right moment: session.create now
|
||||
# defers _start_agent_build behind a 50ms timer (see the
|
||||
# `_deferred_build` path in @method("session.create")), so closing
|
||||
# before the build thread has even started would skip the orphan
|
||||
# detection entirely and the test would race a non-event.
|
||||
build_started = threading.Event()
|
||||
release_build = threading.Event()
|
||||
|
||||
def _slow_make_agent(sid, key):
|
||||
def _slow_make_agent(sid, key, session_id=None):
|
||||
build_started.set()
|
||||
release_build.wait(timeout=3.0)
|
||||
return _FakeAgent()
|
||||
|
||||
@ -2577,6 +2585,12 @@ def test_session_create_close_race_does_not_orphan_worker(monkeypatch):
|
||||
assert resp.get("result"), f"got error: {resp.get('error')}"
|
||||
sid = resp["result"]["session_id"]
|
||||
|
||||
# Wait until the (deferred) build thread has actually entered
|
||||
# _make_agent — otherwise session.close pops _sessions[sid] before
|
||||
# _build ever runs, _start_agent_build never calls _build, and we
|
||||
# never exercise the orphan-cleanup path.
|
||||
assert build_started.wait(timeout=2.0), "build thread never entered _make_agent"
|
||||
|
||||
# Build thread is blocked in _slow_make_agent. Close the session
|
||||
# NOW — this pops _sessions[sid] before _build can install the
|
||||
# worker/notify.
|
||||
|
||||
@ -127,7 +127,11 @@ class TestReadTrackerCaps:
|
||||
td = ft._read_tracker["long-session"]
|
||||
assert len(td["read_history"]) <= 3
|
||||
assert len(td["dedup"]) <= 3
|
||||
assert len(td["read_timestamps"]) <= 3
|
||||
# read_timestamps is populated lazily (via setdefault) only
|
||||
# when os.path.getmtime() succeeds. On some CI filesystems
|
||||
# that stat can race with file creation — skip rather than
|
||||
# hard-error if the dict hasn't been created yet.
|
||||
assert len(td.get("read_timestamps", {})) <= 3
|
||||
|
||||
|
||||
class TestCompletionConsumedPrune:
|
||||
|
||||
@ -209,6 +209,12 @@ class TestIsWsl:
|
||||
import hermes_constants
|
||||
hermes_constants._wsl_detected = None
|
||||
|
||||
def teardown_method(self):
|
||||
# Reset again after the test so we don't leak a cached value
|
||||
# (True/False) into whichever test the xdist worker runs next.
|
||||
import hermes_constants
|
||||
hermes_constants._wsl_detected = None
|
||||
|
||||
def test_wsl2_detected(self):
|
||||
content = "Linux version 5.15.0 (microsoft-standard-WSL2)"
|
||||
with patch("builtins.open", mock_open(read_data=content)):
|
||||
@ -220,18 +226,25 @@ class TestIsWsl:
|
||||
assert _is_wsl() is True
|
||||
|
||||
def test_regular_linux(self):
|
||||
# GHA hosted runners are Azure VMs whose real /proc/version often
|
||||
# contains "microsoft". Patching builtins.open with mock_open is
|
||||
# supposed to intercept hermes_constants.is_wsl's `open` call,
|
||||
# but if another test on the same xdist worker already cached
|
||||
# _wsl_detected=True, the mock never runs because the function
|
||||
# short-circuits on the cache. setup_method resets, so we just
|
||||
# need to be sure the patched `open` is actually reached.
|
||||
content = "Linux version 6.14.0-37-generic (buildd@lcy02-amd64-049)"
|
||||
with patch("builtins.open", mock_open(read_data=content)):
|
||||
with patch("hermes_constants.open", mock_open(read_data=content), create=True):
|
||||
assert _is_wsl() is False
|
||||
|
||||
def test_proc_version_missing(self):
|
||||
with patch("builtins.open", side_effect=FileNotFoundError):
|
||||
with patch("hermes_constants.open", side_effect=FileNotFoundError, create=True):
|
||||
assert _is_wsl() is False
|
||||
|
||||
def test_result_is_cached(self):
|
||||
import hermes_constants
|
||||
content = "Linux version 5.15.0 (microsoft-standard-WSL2)"
|
||||
with patch("builtins.open", mock_open(read_data=content)) as m:
|
||||
with patch("hermes_constants.open", mock_open(read_data=content), create=True) as m:
|
||||
assert _is_wsl() is True
|
||||
assert _is_wsl() is True
|
||||
m.assert_called_once() # only read once
|
||||
|
||||
@ -88,24 +88,29 @@ class TestMessageHandler:
|
||||
from mcp.types import ServerNotification, ToolListChangedNotification
|
||||
|
||||
server = MCPServerTask("notif_srv")
|
||||
with patch.object(MCPServerTask, "_refresh_tools", new_callable=AsyncMock) as mock_refresh:
|
||||
# Product now schedules the refresh as a background task (see
|
||||
# _schedule_tools_refresh in mcp_tool.py ~L918) rather than awaiting
|
||||
# it directly, to avoid wedging the stdio JSON-RPC stream. Patch at
|
||||
# the scheduler seam so we can still assert dispatch happened without
|
||||
# reaching into asyncio.create_task internals.
|
||||
with patch.object(MCPServerTask, "_schedule_tools_refresh") as mock_schedule:
|
||||
handler = server._make_message_handler()
|
||||
notification = ServerNotification(
|
||||
root=ToolListChangedNotification(method="notifications/tools/list_changed")
|
||||
)
|
||||
await handler(notification)
|
||||
mock_refresh.assert_awaited_once()
|
||||
mock_schedule.assert_called_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ignores_exceptions_and_other_messages(self):
|
||||
server = MCPServerTask("notif_srv")
|
||||
with patch.object(MCPServerTask, "_refresh_tools", new_callable=AsyncMock) as mock_refresh:
|
||||
with patch.object(MCPServerTask, "_schedule_tools_refresh") as mock_schedule:
|
||||
handler = server._make_message_handler()
|
||||
# Exceptions should not trigger refresh
|
||||
await handler(RuntimeError("connection dead"))
|
||||
# Unknown message types should not trigger refresh
|
||||
await handler({"jsonrpc": "2.0", "result": "ok"})
|
||||
mock_refresh.assert_not_awaited()
|
||||
mock_schedule.assert_not_called()
|
||||
|
||||
|
||||
class TestDeregister:
|
||||
|
||||
@ -35,7 +35,15 @@ def _fake_run_on_mcp_loop(coro, timeout=30):
|
||||
"""Run an MCP coroutine directly in a fresh event loop."""
|
||||
loop = asyncio.new_event_loop()
|
||||
try:
|
||||
return loop.run_until_complete(coro)
|
||||
# `_rpc_lock` must be created inside the loop that awaits it, or asyncio
|
||||
# raises "attached to a different loop". Build it here and attach it to
|
||||
# whatever fake server is currently registered under _servers.
|
||||
async def _install_lock_and_run():
|
||||
for srv in list(mcp_tool._servers.values()):
|
||||
if getattr(srv, "_rpc_lock", None) is None:
|
||||
srv._rpc_lock = asyncio.Lock()
|
||||
return await coro
|
||||
return loop.run_until_complete(_install_lock_and_run())
|
||||
finally:
|
||||
loop.close()
|
||||
|
||||
@ -44,7 +52,10 @@ def _fake_run_on_mcp_loop(coro, timeout=30):
|
||||
def _patch_mcp_server():
|
||||
"""Patch _servers and the MCP event loop so _make_tool_handler can run."""
|
||||
fake_session = MagicMock()
|
||||
fake_server = SimpleNamespace(session=fake_session)
|
||||
# `_rpc_lock` is acquired by _make_tool_handler's call path (mcp_tool.py
|
||||
# ~L2008) to serialize JSON-RPC against the server — build it inside the
|
||||
# fresh loop that _fake_run_on_mcp_loop spins up, not at fixture import.
|
||||
fake_server = SimpleNamespace(session=fake_session, _rpc_lock=None)
|
||||
with patch.dict(mcp_tool._servers, {"test-server": fake_server}), \
|
||||
patch("tools.mcp_tool._run_on_mcp_loop", side_effect=_fake_run_on_mcp_loop):
|
||||
yield fake_session
|
||||
|
||||
Reference in New Issue
Block a user