def test_to_representation_handles_none(self):
field = EnumField(enum_class=SomeEnum)
value = field.to_representation(None)
self.assertIsNone(value)
def test_to_representation_works(self):
field = EnumField(enum_class=SomeEnum)
value = field.to_representation(SomeEnum.test1)
self.assertEqual(value, 'test1')
def test_to_internal_value_works(self):
field = EnumField(enum_class=SomeEnum)
value = field.to_internal_value('test2')
self.assertEqual(value, SomeEnum.test2)
def test_to_internal_value_validates(self):
field = EnumField(enum_class=SomeEnum)
with self.assertRaises(ValidationError):
field.to_internal_value('test3')