def test_set_flattened_tags_keys():
"""Ensure expected keys in flattened dictionary"""
d = dict(A=1, B=2, C=dict(A=3, B=4, C=dict(A=5, B=6)))
e = dict(A=1, B=2, C_A=3, C_B=4, C_C_A=5, C_C_B=6)
span = Span("test")
trace_utils.set_flattened_tags(span, d.items(), sep="_")
assert span.get_metrics() == e
def test_set_flattened_tags_exclude_policy():
"""Ensure expected keys in flattened dictionary with exclusion set"""
d = dict(A=1, B=2, C=dict(A=3, B=4, C=dict(A=5, B=6)))
e = dict(A=1, B=2, C_B=4)
span = Span("test")
trace_utils.set_flattened_tags(
span,
d.items(),
sep="_",
exclude_policy=lambda tag: tag in {"C_A", "C_C"})
assert span.get_metrics() == e
