Max Isbey
1b74b06753
Cut comment and docstring volume roughly in half across src, tests, examples, and docs_src: removed comments that restate the adjacent code, leftover development narration, section banners, and self-evident Args/Returns blocks, and compressed the remaining docstrings to a Google-style summary line plus only the detail that earns its place. Kept (and tightened) the load-bearing content: Raises sections, deprecation and version-availability notes, spec/RFC/issue references, why-comments for non-obvious decisions, and all coverage pragmas. The generated mcp_types.v* wire modules are untouched.
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
"""`docs/advanced/opentelemetry.md`: every claim the page makes, proved against the real SDK."""
|
|
|
|
import pytest
|
|
from logfire.testing import CaptureLogfire
|
|
|
|
from docs_src.opentelemetry import tutorial001
|
|
from mcp import Client
|
|
|
|
# See test_index.py for why this is a per-module mark and not a conftest hook.
|
|
pytestmark = [pytest.mark.anyio, pytest.mark.filterwarnings("error::mcp.MCPDeprecationWarning")]
|
|
|
|
|
|
async def test_a_plain_server_is_traced_with_no_extra_code(capfire: CaptureLogfire) -> None:
|
|
async with Client(tutorial001.mcp) as client:
|
|
await client.call_tool("search_books", {"query": "dune"})
|
|
|
|
spans = {s["name"]: s for s in capfire.exporter.exported_spans_as_dict()}
|
|
assert "tools/call search_books" in spans
|
|
|
|
attributes = spans["tools/call search_books"]["attributes"]
|
|
assert attributes["mcp.method.name"] == "tools/call"
|
|
assert attributes["gen_ai.operation.name"] == "execute_tool"
|
|
assert attributes["gen_ai.tool.name"] == "search_books"
|
|
|
|
|
|
async def test_client_and_server_share_one_trace(capfire: CaptureLogfire) -> None:
|
|
"""When both sides run the SDK, the client and server spans land in one trace (SEP-414)."""
|
|
async with Client(tutorial001.mcp, mode="legacy") as client:
|
|
await client.call_tool("search_books", {"query": "dune"})
|
|
|
|
spans = {s["name"]: s for s in capfire.exporter.exported_spans_as_dict()}
|
|
client_span = spans["MCP send tools/call search_books"]
|
|
server_span = spans["tools/call search_books"]
|
|
assert server_span["context"]["trace_id"] == client_span["context"]["trace_id"]
|