mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-27 20:14:01 +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
44 lines
1.3 KiB
Python
44 lines
1.3 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
|
|
from tests.test_multitable_inheritance import ChildModel
|
|
|
|
|
|
# Regression test for #4290
|
|
class ChildAssociatedModel(RESTFrameworkModel):
|
|
child_model = models.OneToOneField(ChildModel, on_delete=models.CASCADE)
|
|
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),
|
|
{'name1', 'name2', 'id', 'childassociatedmodel'})
|