o ih @sddlZddlZddlZddlmZddlmZddlmZddlmZddl m Z e e Z de ddfd d Zd d Zdd dZdedejefddZejdejeddffddZejdejeddffddZdS)N)DDTESTOPT_ROOT_SPAN_RESOURCE)DDTraceTestContextPlainTestContext) TestContext) TestOptWriterwriterreturncCsTzddl}WntytdYdSwddlm}||}|jj|gddS)zz Install a trace filter in the global ddtrace tracer that forwards spans generated by ddtrace to ddtrace.testing. rNz4ddrace is not available, not installing trace filter)TestOptSpanProcessortrace_processors)ddtrace ImportErrorlogdebugspan_processorr tracer configure)rrr rr_/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/testing/internal/tracer_api/context.pyinstall_global_trace_filters    rcCs8z ddlm}|WdStytdYdSw)Nr _patch_allz#Error enabling ddtrace integrations)ddtrace._monkeyr Exceptionr exceptionrrrrenable_all_ddtrace_integrations s    rcCs4zddl}Wn tyYdSw|jjgddS)zA Uninstall trace filters from the global ddtrace tracer. rNr )rrrr)rrrruninstall_global_trace_filter)s   rddtrace_enabledcCs<|rzddl}tWStytdYtSwtS)a Create a trace context for a test to run. If ddtrace is enabled, a ddtrace context will be started; any spans created inside the test (e.g., instrumented HTTP requests) will be children of this context. This context manager yields a `TestContext` object containing the trace_id and span_id of created context. If ddtrace is not enabled, yields a dummy context with a freshly generated trace_id and span_id. rNzs"