def testGetEnumValuesetUrl_withEnumValueSet_returnsCorrectValue(self):
"""Tests get_enum_valueset_url with a valid enum valueset."""
body_length_descriptor = valuesets_pb2.BodyLengthUnitsValueSet().DESCRIPTOR
value_enum_descriptor = body_length_descriptor.enum_types_by_name['Value']
self.assertEqual(
annotation_utils.get_enum_valueset_url(value_enum_descriptor),
_BODY_LENGTH_UNITS_VALUESET_URL)
def testGetEnumValuesetUrl_withMessageDescriptor_raisesException(self):
"""Tests get_enum_valueset_url raises without passing an EnumDescriptor."""
with self.assertRaises(KeyError) as ke:
annotation_utils.get_enum_valueset_url(patient_pb2.Patient())
self.assertIsInstance(ke.exception, KeyError)