007 add telemetry

tests/observability/test_logging_context.py

@@ -0,0 +1,50 @@
+import structlog
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
+from structlog.testing import LogCapture
+from structlog.contextvars import bind_contextvars, clear_contextvars
+
+from app.observability.logging import add_trace_context
+from app.observability.tracing import get_tracer, reset_tracing_for_tests, setup_tracing
+
+
+def test_logging_context_includes_request_id_and_trace_id() -> None:
+    reset_tracing_for_tests()
+    span_exporter = InMemorySpanExporter()
+    tracer_provider = TracerProvider(resource=Resource.create({"service.name": "tests"}))
+    tracer_provider.add_span_processor(SimpleSpanProcessor(span_exporter))
+    setup_tracing(
+        service_name="tests",
+        otlp_endpoint="",
+        tracer_provider=tracer_provider,
+    )
+
+    log_capture = LogCapture()
+    structlog.configure(
+        processors=[
+            structlog.contextvars.merge_contextvars,
+            add_trace_context,
+            log_capture,
+        ],
+        logger_factory=structlog.stdlib.LoggerFactory(),
+        wrapper_class=structlog.make_filtering_bound_logger(20),
+        cache_logger_on_first_use=False,
+    )
+    logger = structlog.get_logger("test-logger")
+
+    clear_contextvars()
+    bind_contextvars(request_id="req-test-id")
+    with get_tracer(__name__).start_as_current_span("test-span"):
+        logger.info("request_log")
+
+    assert len(log_capture.entries) == 1
+    entry = log_capture.entries[0]
+    assert entry["event"] == "request_log"
+    assert entry["request_id"] == "req-test-id"
+    assert len(entry["trace_id"]) == 32
+    assert len(entry["span_id"]) == 16
+
+    clear_contextvars()
+    reset_tracing_for_tests()