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