encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-27 08:29:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

made things prettier

Browse Source
This commit is contained in:
Ben Roberts 2012-12-11 22:47:50 -07:00
parent 8bbfa90ff4
commit 043f4a5a4a
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
rest_framework/tests/serializer.py
View File

@ -769,8 +769,6 @@ class DepthTest(TestCase):
class NestedSerializerContextTests(TestCase): class NestedSerializerContextTests(TestCase):
def test_nested_serializer_context(self): def test_nested_serializer_context(self):
class AlbumCollection(object):
albums = None
class PhotoSerializer(serializers.ModelSerializer): class PhotoSerializer(serializers.ModelSerializer):
class Meta: class Meta:
@ -797,15 +795,24 @@ class NestedSerializerContextTests(TestCase):
raise RuntimeError("context isn't getting passed into 1st level nested serializer") raise RuntimeError("context isn't getting passed into 1st level nested serializer")
return "success" return "success"
class AlbumCollection(object):
albums = None
class AlbumCollectionSerializer(serializers.Serializer): class AlbumCollectionSerializer(serializers.Serializer):
albums = AlbumSerializer(source="albums", ) albums = AlbumSerializer(source="albums")
album1 = Album.objects.create(title="album 1") album1 = Album.objects.create(title="album 1")
album1.photo_set.add(Photo(description="Dog"), Photo(description="Cat")) album1.photo_set.add(
Photo.objects.create(description="Bigfoot"),
Photo.objects.create(description="Unicorn"))
album2 = Album.objects.create(title="album 2") album2 = Album.objects.create(title="album 2")
album2.photo_set.add(Photo(description="Yeti"), Photo(description="Sasquatch"), Photo(description="Bigfoot")) album2.photo_set.add(
Photo.objects.create(description="Yeti"),
Photo.objects.create(description="Sasquatch"))
album_collection = AlbumCollection() album_collection = AlbumCollection()
album_collection.albums = [album1, album2] album_collection.albums = [album1, album2]
#the test. (will raise RuntimeError if context doesn't get passed correctly to the nested Serializers)
data = AlbumCollectionSerializer(album_collection, context={'context_item': 'album context'}).data data = AlbumCollectionSerializer(album_collection, context={'context_item': 'album context'}).data