diff options
Diffstat (limited to 'tests/serializers')
| -rw-r--r-- | tests/serializers/models/__init__.py | 1 | ||||
| -rw-r--r-- | tests/serializers/models/multi_table.py | 22 | ||||
| -rw-r--r-- | tests/serializers/test_natural.py | 33 |
3 files changed, 55 insertions, 1 deletions
diff --git a/tests/serializers/models/__init__.py b/tests/serializers/models/__init__.py index a786b54be9..9ac2381d17 100644 --- a/tests/serializers/models/__init__.py +++ b/tests/serializers/models/__init__.py @@ -1,3 +1,4 @@ from .base import * # NOQA from .data import * # NOQA +from .multi_table import * # NOQA from .natural import * # NOQA diff --git a/tests/serializers/models/multi_table.py b/tests/serializers/models/multi_table.py new file mode 100644 index 0000000000..467ab50955 --- /dev/null +++ b/tests/serializers/models/multi_table.py @@ -0,0 +1,22 @@ +from django.db import models + + +class ParentManager(models.Manager): + def get_by_natural_key(self, parent_data): + return self.get(parent_data=parent_data) + + +class Parent(models.Model): + parent_data = models.CharField(max_length=30, unique=True) + + objects = ParentManager() + + def natural_key(self): + return (self.parent_data, ) + + +class Child(Parent): + child_data = models.CharField(max_length=30, unique=True) + + class Meta: + manager_inheritance_from_future = True diff --git a/tests/serializers/test_natural.py b/tests/serializers/test_natural.py index d8c4e9a7db..1a4f4d5ea0 100644 --- a/tests/serializers/test_natural.py +++ b/tests/serializers/test_natural.py @@ -4,7 +4,7 @@ from django.core import serializers from django.db import connection from django.test import TestCase -from .models import FKDataNaturalKey, NaturalKeyAnchor +from .models import Child, FKDataNaturalKey, NaturalKeyAnchor from .tests import register_tests @@ -69,6 +69,37 @@ def natural_key_test(format, self): self.assertIsNone(books[1].object.pk) +def natural_pk_mti_test(format, self): + """ + If serializing objects in a multi-table inheritance relationship using + natural primary keys, the natural foreign key for the parent is output in + the fields of the child so it's possible to relate the child to the parent + when deserializing. + """ + child_1 = Child.objects.create(parent_data='1', child_data='1') + child_2 = Child.objects.create(parent_data='2', child_data='2') + + string_data = serializers.serialize( + format, + [child_1.parent_ptr, child_2.parent_ptr, child_2, child_1], + use_natural_foreign_keys=True, use_natural_primary_keys=True, + ) + + child_1.delete() + child_2.delete() + + for obj in serializers.deserialize(format, string_data): + obj.save() + + children = Child.objects.all() + self.assertEqual(len(children), 2) + for child in children: + # If it's possible to find the superclass from the subclass and it's + # the correct superclass, it's working. + self.assertEqual(child.child_data, child.parent_data) + + # Dynamically register tests for each serializer register_tests(NaturalKeySerializerTests, 'test_%s_natural_key_serializer', natural_key_serializer_test) register_tests(NaturalKeySerializerTests, 'test_%s_serializer_natural_keys', natural_key_test) +register_tests(NaturalKeySerializerTests, 'test_%s_serializer_natural_pks_mti', natural_pk_mti_test) |
