Раис Юсупалиев
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
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()
|