Test one to one with inheritance (encode#4575)

tomchristie · web-flow · commit de08f28a9185 · 2016-10-13T14:21:23.000+01:00
diff --git a/tests/test_one_to_one_with_inheritance.py b/tests/test_one_to_one_with_inheritance.py
@@ -0,0 +1,46 @@
+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)
+    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']))
