def test_sampling_attributes(self):
decision_attributes = {
"sampler-attr": "sample-val",
"attr-in-both": "decision-attr",
}
tracer_provider = trace.TracerProvider(
sampling.StaticSampler(
sampling.Decision(sampled=True,
attributes=decision_attributes)))
self.tracer = tracer_provider.get_tracer(__name__)
with self.tracer.start_as_current_span("root2") as root:
self.assertEqual(len(root.attributes), 2)
self.assertEqual(root.attributes["sampler-attr"], "sample-val")
self.assertEqual(root.attributes["attr-in-both"], "decision-attr")
attributes = {
"attr-key": "val",
"attr-key2": "val2",
"attr-in-both": "span-attr",
}
with self.tracer.start_as_current_span("root2",
attributes=attributes) as root:
self.assertEqual(len(root.attributes), 4)
self.assertEqual(root.attributes["attr-key"], "val")
self.assertEqual(root.attributes["attr-key2"], "val2")
self.assertEqual(root.attributes["sampler-attr"], "sample-val")
self.assertEqual(root.attributes["attr-in-both"], "decision-attr")
def test_sampling_attributes(self):
sampling_attributes = {
"sampler-attr": "sample-val",
"attr-in-both": "decision-attr",
}
tracer_provider = trace.TracerProvider(
sampling.StaticSampler(sampling.Decision.RECORD_AND_SAMPLE, ))
self.tracer = tracer_provider.get_tracer(__name__)
with self.tracer.start_as_current_span(
name="root2", attributes=sampling_attributes) as root:
self.assertEqual(len(root.attributes), 2)
self.assertEqual(root.attributes["sampler-attr"], "sample-val")
self.assertEqual(root.attributes["attr-in-both"], "decision-attr")
self.assertEqual(root.get_context().trace_flags,
trace_api.TraceFlags.SAMPLED)
