Function __assert_attr

tests/util/otel_trace.py:609–634

Helper function to assert that a specific OpenTelemetry attribute exists in a span.

(span_name, attributes, expected_key, expected_value,
    expected_type="stringValue")

def __assert_attr(span_name, attributes, expected_key, expected_value,
    expected_type="stringValue"):
  """
    Helper function to assert that a specific OpenTelemetry attribute exists in a span.
  """

  assert expected_type in ("stringValue", "boolValue", "intValue"), "Invalid " \
      "expected_type '{}', must be one of 'stringValue', 'boolValue', or 'intValue'" \
      .format(expected_type)

  val = attributes[expected_key]
  assert val is not None, "Span '{}' attribute not found: '{}', actual attributes: {}" \
      .format(span_name, expected_key, attributes)
  assert val.value == expected_value, "Span '{}' attribute '{}' expected: '{}', " \
      "actual: '{}'".format(span_name, expected_key, expected_value, val.value)

  if expected_type == "boolValue":
    expected_type = ATTR_VAL_TYPE_BOOL
  elif expected_type == "intValue":
    expected_type = ATTR_VAL_TYPE_INT
  else:
    expected_type = ATTR_VAL_TYPE_STRING

  assert val.get_type() == expected_type, "Span '{}' attribute '{}' expected to be " \
      "of type '{}', actual: '{}'".format(span_name, expected_key, expected_type,
        val.get_type())


def __assert_span_events(span, expected_events=[]):