src/test_headers.py

#!/usr/bin/env python3
"""Test script to verify OpenTelemetry W3C Trace Context header injection."""

from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
from opentelemetry import trace
from opentelemetry.sdk.trace import TracerProvider

# Setup tracer
provider = TracerProvider()
trace.set_tracer_provider(provider)
tracer = trace.get_tracer('test')

# Create span and inject headers
headers = {}
with tracer.start_as_current_span('test-span'):
    TraceContextTextMapPropagator().inject(headers)
    print("Headers injected by OpenTelemetry:")
    for key, value in headers.items():
        print(f"  {key}: {value}")
    
    if not headers:
        print("  (no headers - check if span is current)")