mirror of
https://github.com/turnstonelabs/turnstone.git
synced 2026-08-12 23:12:23 -06:00
Apply MCP formatting
This commit is contained in:
@@ -157,9 +157,7 @@ class TestTransportOwnerLifecycle:
|
||||
assert state.owner_task is None
|
||||
assert state.close_requested is None
|
||||
|
||||
def test_tools_only_connect_skips_unchanged_catalog_listeners(
|
||||
self, running_loop_mgr
|
||||
) -> None:
|
||||
def test_tools_only_connect_skips_unchanged_catalog_listeners(self, running_loop_mgr) -> None:
|
||||
"""A tools-only registration must not rebuild every live chat twice."""
|
||||
mgr, loop, _ = running_loop_mgr
|
||||
patches: dict[str, Any] = {}
|
||||
@@ -465,9 +463,7 @@ class TestTransportOwnerLifecycle:
|
||||
assert len(state.tools) == 1
|
||||
assert mgr.is_mcp_tool("mcp__new__late") is True
|
||||
|
||||
def test_blocked_loop_cannot_return_failure_before_add_outcome(
|
||||
self, running_loop_mgr
|
||||
) -> None:
|
||||
def test_blocked_loop_cannot_return_failure_before_add_outcome(self, running_loop_mgr) -> None:
|
||||
"""A synchronous listener stall may delay success, never expose a ghost."""
|
||||
mgr, _loop, _ = running_loop_mgr
|
||||
|
||||
|
||||
@@ -2468,14 +2468,8 @@ class MCPClientManager:
|
||||
# exact wiring immediately before commit; there are no awaits from
|
||||
# this check through publication, so a callable catalog can never
|
||||
# be installed behind a dead/replaced transport.
|
||||
if (
|
||||
owner.done()
|
||||
or state.owner_task is not owner
|
||||
or state.session is not session
|
||||
):
|
||||
raise ConnectionError(
|
||||
f"MCP server '{name}' transport died before catalog commit"
|
||||
)
|
||||
if owner.done() or state.owner_task is not owner or state.session is not session:
|
||||
raise ConnectionError(f"MCP server '{name}' transport died before catalog commit")
|
||||
except BaseException:
|
||||
await self._teardown_static_session(name)
|
||||
raise
|
||||
@@ -3154,11 +3148,7 @@ class MCPClientManager:
|
||||
# Mirror the static commit guard. A transport owner can finish in the
|
||||
# same scheduling turn as the final discovery response; never publish
|
||||
# that response into the per-user maps after its session was evicted.
|
||||
if (
|
||||
owner.done()
|
||||
or entry.owner_task is not owner
|
||||
or entry.session is not session
|
||||
):
|
||||
if owner.done() or entry.owner_task is not owner or entry.session is not session:
|
||||
await self._teardown_pool_entry(key)
|
||||
raise ConnectionError(
|
||||
f"MCP pool server '{server_name}' transport died before catalog commit"
|
||||
@@ -5835,9 +5825,7 @@ class MCPClientManager:
|
||||
# run after this cleanup releases the lock.
|
||||
_clear_failed_add_state()
|
||||
if isinstance(exc, TimeoutError):
|
||||
raise TimeoutError(
|
||||
f"MCP server '{name}' registration timed out"
|
||||
) from None
|
||||
raise TimeoutError(f"MCP server '{name}' registration timed out") from None
|
||||
raise
|
||||
|
||||
future = asyncio.run_coroutine_threadsafe(_add(), self._loop)
|
||||
|
||||
Reference in New Issue
Block a user