mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-08-07 13:54:47 +03:00
Moved tests to exhibit the inheritance problem to separate file
This commit is contained in:
parent
0f61c9ec29
commit
8ac0c0cbea
56
tests/test_onetoone_with_inheritance.py
Normal file
56
tests/test_onetoone_with_inheritance.py
Normal file
|
@ -0,0 +1,56 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
from django.db import models
|
||||
from django.test import TestCase
|
||||
|
||||
from rest_framework import serializers
|
||||
from tests.models import RESTFrameworkModel
|
||||
|
||||
|
||||
# Models
|
||||
from tests.test_multitable_inheritance import ChildModel, DerivedModelSerializer
|
||||
|
||||
|
||||
class ChildAssociatedModel(RESTFrameworkModel):
|
||||
child_model = models.OneToOneField(ChildModel)
|
||||
child_name = models.CharField(max_length=100)
|
||||
|
||||
|
||||
# Serializers
|
||||
class DerivedModelSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = ChildModel
|
||||
fields = ['id', 'name1', 'name2', 'childassociatedmodel']
|
||||
|
||||
|
||||
class ChildAssociatedModelSerializer(serializers.ModelSerializer):
|
||||
|
||||
class Meta:
|
||||
model = ChildAssociatedModel
|
||||
fields = ['id', 'child_name']
|
||||
|
||||
|
||||
# Tests
|
||||
class InheritedModelSerializationTests(TestCase):
|
||||
|
||||
def test_multitable_inherited_model_fields_as_expected(self):
|
||||
"""
|
||||
Assert that the parent pointer field is not included in the fields
|
||||
serialized fields
|
||||
"""
|
||||
child = ChildModel(name1='parent name', name2='child name')
|
||||
serializer = DerivedModelSerializer(child)
|
||||
self.assertEqual(set(serializer.data.keys()),
|
||||
set(['name1', 'name2', 'id', 'childassociatedmodel']))
|
||||
|
||||
def test_data_is_valid_without_parent_ptr(self):
|
||||
"""
|
||||
Assert that the pointer to the parent table is not a required field
|
||||
for input data
|
||||
"""
|
||||
data = {
|
||||
'name1': 'parent name',
|
||||
'name2': 'child name',
|
||||
}
|
||||
serializer = DerivedModelSerializer(data=data)
|
||||
self.assertEqual(serializer.is_valid(), True)
|
Loading…
Reference in New Issue
Block a user