def get_type(self, obj):
"""A get method for the AffiliateSerializer class' ``type`` attribute.
Args:
obj: The instance of the ContactInfo model class that is being serialized.
Returns:
The verbose label associated with the ``type`` of the ContactInfo object.
"""
return ContactInfo.InfoType(obj.type).label
def test_contact_info_type_serializer_method_field(self):
"""Ensure that the contact info `type` serializer method field has the correct value in an API response.
"""
url = reverse('event-list')
response = self.client.get(f'{url}/{self.event.pk}/')
contact = response.data['contacts'][0]
self.assertEqual(
ContactInfo.InfoType(self.contact.type).label, contact['type'])