Add test showing error

This commit is contained in:
Carlton Gibson 2016-11-02 22:26:52 +01:00
parent d55e176a1e
commit 35e3563e7a

View File

@ -41,6 +41,10 @@ class OneFieldModel(models.Model):
char_field = models.CharField(max_length=100)
class ExtendedModel(OneFieldModel):
text_field = models.TextField(max_length=100)
class RegularFieldsModel(models.Model):
"""
A model class for testing regular flat fields.
@ -141,6 +145,26 @@ class TestModelSerializer(TestCase):
msginitial = 'Cannot use ModelSerializer with Abstract Models.'
assert str(excinfo.exception).startswith(msginitial)
def test_inheritance(self):
"""
Check serializer contains only correct fields
"""
class TestSerializer(serializers.ModelSerializer):
class Meta:
model = ExtendedModel
fields = '__all__'
instance = ExtendedModel.objects.create(
char_field="An Instance",
text_field="""To check only the two fields plus id are present in data"""
)
serializer = TestSerializer(instance)
data = serializer.data
assert len(data.keys()) == 3
assert data.keys() == ['id', 'char_field', 'text_field',]
class TestRegularFieldMappings(TestCase):
def test_regular_fields(self):