mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-30 05:24:00 +03:00
ffe3dbb1b1
Calling dict.keys() is unnecessary. The two are functionally equivalent on modern Pythons. Inspired by Lennart Regebro's talk "Prehistoric Patterns in Python" from PyCon 2017. https://www.youtube.com/watch?v=V5-JH23Vk0I
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
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
|
|
class ParentModel(RESTFrameworkModel):
|
|
name1 = models.CharField(max_length=100)
|
|
|
|
|
|
class ChildModel(ParentModel):
|
|
name2 = models.CharField(max_length=100)
|
|
|
|
|
|
class AssociatedModel(RESTFrameworkModel):
|
|
ref = models.OneToOneField(ParentModel, primary_key=True, on_delete=models.CASCADE)
|
|
name = models.CharField(max_length=100)
|
|
|
|
|
|
# Serializers
|
|
class DerivedModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = ChildModel
|
|
fields = '__all__'
|
|
|
|
|
|
class AssociatedModelSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = AssociatedModel
|
|
fields = '__all__'
|
|
|
|
|
|
# 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)
|
|
assert set(serializer.data) == {'name1', 'name2', 'id'}
|
|
|
|
def test_onetoone_primary_key_model_fields_as_expected(self):
|
|
"""
|
|
Assert that a model with a onetoone field that is the primary key is
|
|
not treated like a derived model
|
|
"""
|
|
parent = ParentModel.objects.create(name1='parent name')
|
|
associate = AssociatedModel.objects.create(name='hello', ref=parent)
|
|
serializer = AssociatedModelSerializer(associate)
|
|
assert set(serializer.data) == {'name', 'ref'}
|
|
|
|
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)
|
|
assert serializer.is_valid() is True
|