def test_filters(writer, tracer):
if writer == "sync":
writer = AgentWriter(
tracer.writer.agent_url,
priority_sampler=tracer.priority_sampler,
sync_mode=True,
)
# Need to copy the headers which contain the test token to associate
# traces with this test case.
writer._headers = tracer.writer._headers
else:
writer = tracer.writer
class FilterMutate(object):
def __init__(self, key, value):
self.key = key
self.value = value
def process_trace(self, trace):
for s in trace:
s.set_tag(self.key, self.value)
return trace
tracer.configure(
settings={
"FILTERS": [FilterMutate("boop", "beep")],
},
writer=writer,
)
with tracer.trace("root"):
with tracer.trace("child"):
pass
tracer.shutdown()
def test_sampling(writer, tracer):
if writer == "sync":
writer = AgentWriter(
tracer.writer.agent_url,
priority_sampler=tracer.priority_sampler,
sync_mode=True,
)
# Need to copy the headers which contain the test token to associate
# traces with this test case.
writer._headers = tracer.writer._headers
else:
writer = tracer.writer
tracer.configure(writer=writer)
with tracer.trace("trace1"):
with tracer.trace("child"):
pass
sampler = DatadogSampler(default_sample_rate=1.0)
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace2"):
with tracer.trace("child"):
pass
sampler = DatadogSampler(default_sample_rate=0.000001)
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace3"):
with tracer.trace("child"):
pass
sampler = DatadogSampler(default_sample_rate=1, rules=[SamplingRule(1.0)])
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace4"):
with tracer.trace("child"):
pass
sampler = DatadogSampler(default_sample_rate=1, rules=[SamplingRule(0)])
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace5"):
with tracer.trace("child"):
pass
sampler = DatadogSampler(default_sample_rate=1)
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace6"):
with tracer.trace("child") as span:
span.set_tag(MANUAL_DROP_KEY)
sampler = DatadogSampler(default_sample_rate=1)
tracer.configure(sampler=sampler, writer=writer)
with tracer.trace("trace7"):
with tracer.trace("child") as span:
span.set_tag(MANUAL_KEEP_KEY)
sampler = RateSampler(0.0000000001)
tracer.configure(sampler=sampler, writer=writer)
# This trace should not appear in the snapshot
with tracer.trace("trace8"):
with tracer.trace("child"):
pass
tracer.shutdown()
