Add testcase for adding a field in a mixin

This commit is contained in:
Ian Foote 2014-02-10 12:10:52 +00:00
parent 00b1877106
commit c513eaa213

View File

@ -1809,7 +1809,7 @@ class SerializerDefaultTrueBoolean(TestCase):
self.assertEqual(serializer.data['dog'], False)
class BoolenFieldTypeTest(TestCase):
class BooleanFieldTypeTest(TestCase):
'''
Ensure the various Boolean based model fields are rendered as the proper
field type
@ -1840,3 +1840,20 @@ class BoolenFieldTypeTest(TestCase):
'''
bfield = self.serializer.get_fields()['started']
self.assertEqual(type(bfield), fields.BooleanField)
class SerializerMixin(object):
data = serializers.SerializerMethodField('get_data')
def get_data(self, obj):
return 'My data'
class SerializerWithMixin(SerializerMixin, serializers.Serializer):
pass
class SerializerMixinTest(TestCase):
def test_mixin_field(self):
serializer = SerializerWithMixin()
self.assertIn('data', serializer.base_fields)