mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-08-04 20:40:14 +03:00
Add test to demonstrate that nested add/remove order needs addressing
This commit is contained in:
parent
8495cd898a
commit
8c5fbfd088
|
@ -1,5 +1,6 @@
|
||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
from django.db import models
|
from django.db import models
|
||||||
|
from django.utils.timezone import now
|
||||||
from django.utils.translation import ugettext_lazy as _
|
from django.utils.translation import ugettext_lazy as _
|
||||||
from rest_framework import serializers
|
from rest_framework import serializers
|
||||||
|
|
||||||
|
@ -103,6 +104,10 @@ class Person(RESTFrameworkModel):
|
||||||
class BlogPost(RESTFrameworkModel):
|
class BlogPost(RESTFrameworkModel):
|
||||||
title = models.CharField(max_length=100)
|
title = models.CharField(max_length=100)
|
||||||
writer = models.ForeignKey(Person, null=True, blank=True)
|
writer = models.ForeignKey(Person, null=True, blank=True)
|
||||||
|
created = models.DateTimeField(default=now)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
unique_together = [('writer', 'title')]
|
||||||
|
|
||||||
def get_first_comment(self):
|
def get_first_comment(self):
|
||||||
return self.blogpostcomment_set.all()[0]
|
return self.blogpostcomment_set.all()[0]
|
||||||
|
|
|
@ -1448,7 +1448,7 @@ class DepthTest(TestCase):
|
||||||
def test_implicit_nesting(self):
|
def test_implicit_nesting(self):
|
||||||
|
|
||||||
writer = Person.objects.create(name="django", age=1)
|
writer = Person.objects.create(name="django", age=1)
|
||||||
post = BlogPost.objects.create(title="Test blog post", writer=writer)
|
post = BlogPost.objects.create(title="Test blog post", writer=writer, created=datetime.datetime(1970, 1, 1))
|
||||||
comment = BlogPostComment.objects.create(text="Test blog post comment", blog_post=post)
|
comment = BlogPostComment.objects.create(text="Test blog post comment", blog_post=post)
|
||||||
|
|
||||||
class BlogPostCommentSerializer(serializers.ModelSerializer):
|
class BlogPostCommentSerializer(serializers.ModelSerializer):
|
||||||
|
@ -1458,7 +1458,7 @@ class DepthTest(TestCase):
|
||||||
|
|
||||||
serializer = BlogPostCommentSerializer(instance=comment)
|
serializer = BlogPostCommentSerializer(instance=comment)
|
||||||
expected = {'id': 1, 'text': 'Test blog post comment', 'blog_post': {'id': 1, 'title': 'Test blog post',
|
expected = {'id': 1, 'text': 'Test blog post comment', 'blog_post': {'id': 1, 'title': 'Test blog post',
|
||||||
'writer': {'id': 1, 'name': 'django', 'age': 1}}}
|
'created': datetime.datetime(1970, 1, 1), 'writer': {'id': 1, 'name': 'django', 'age': 1}}}
|
||||||
|
|
||||||
self.assertEqual(serializer.data, expected)
|
self.assertEqual(serializer.data, expected)
|
||||||
|
|
||||||
|
@ -1476,6 +1476,7 @@ class DepthTest(TestCase):
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
model = BlogPost
|
model = BlogPost
|
||||||
|
exclude = ('created',)
|
||||||
|
|
||||||
class BlogPostCommentSerializer(serializers.ModelSerializer):
|
class BlogPostCommentSerializer(serializers.ModelSerializer):
|
||||||
blog_post = BlogPostSerializer()
|
blog_post = BlogPostSerializer()
|
||||||
|
|
|
@ -347,3 +347,45 @@ class NestedModelSerializerUpdateTests(TestCase):
|
||||||
result = deserialize.object
|
result = deserialize.object
|
||||||
result.save()
|
result.save()
|
||||||
self.assertEqual(result.id, john.id)
|
self.assertEqual(result.id, john.id)
|
||||||
|
|
||||||
|
def test_remove_nested_before_adding(self):
|
||||||
|
john = models.Person.objects.create(name="john")
|
||||||
|
|
||||||
|
post = john.blogpost_set.create(title="Test blog post")
|
||||||
|
post.blogpostcomment_set.create(text="I hate this blog post")
|
||||||
|
post.blogpostcomment_set.create(text="I love this blog post")
|
||||||
|
|
||||||
|
class BlogPostCommentSerializer(serializers.ModelSerializer):
|
||||||
|
class Meta:
|
||||||
|
model = models.BlogPostComment
|
||||||
|
exclude = ('blog_post',)
|
||||||
|
|
||||||
|
class BlogPostSerializer(serializers.ModelSerializer):
|
||||||
|
comments = BlogPostCommentSerializer(many=True, source='blogpostcomment_set')
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = models.BlogPost
|
||||||
|
fields = ('id', 'title', 'comments')
|
||||||
|
|
||||||
|
class PersonSerializer(serializers.ModelSerializer):
|
||||||
|
posts = BlogPostSerializer(many=True, source='blogpost_set', allow_add_remove=True)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
model = models.Person
|
||||||
|
fields = ('id', 'name', 'age', 'posts')
|
||||||
|
|
||||||
|
serialize = PersonSerializer(instance=john)
|
||||||
|
|
||||||
|
# Remove the ID from John's post so that it is no longer the same post.
|
||||||
|
john_data = serialize.data
|
||||||
|
self.assertEqual(len(john_data['posts']), 1)
|
||||||
|
del john_data['posts'][0]['id']
|
||||||
|
|
||||||
|
deserialize = PersonSerializer(data=john_data, instance=john)
|
||||||
|
self.assertTrue(deserialize.is_valid(), deserialize.errors)
|
||||||
|
|
||||||
|
deserialize.save()
|
||||||
|
result = deserialize.object
|
||||||
|
self.assertEqual(result.id, john.id)
|
||||||
|
self.assertEqual(result.blogpost_set.count(), 1)
|
||||||
|
self.assertNotEqual(result.blogpost_set.all()[0].created, post.created)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user