mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
1b39062ecd
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
30 lines
720 B
Python
30 lines
720 B
Python
"""Telemetry coverage for MCP server lifecycle spans."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from contextlib import contextmanager
|
|
|
|
import logfire
|
|
import pytest
|
|
|
|
from basic_memory.mcp.server import lifespan, mcp
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_mcp_lifespan_wraps_startup_and_shutdown(config_manager, monkeypatch) -> None:
|
|
spans: list[tuple[str, dict]] = []
|
|
|
|
@contextmanager
|
|
def fake_span(name: str, **attrs):
|
|
spans.append((name, attrs))
|
|
yield
|
|
|
|
monkeypatch.setattr(logfire, "span", fake_span)
|
|
|
|
async with lifespan(mcp):
|
|
pass
|
|
|
|
span_names = [name for name, _ in spans]
|
|
assert "mcp.lifecycle.startup" in span_names
|
|
assert "mcp.lifecycle.shutdown" in span_names
|