From 043f4a5a4abdec5cef2e568728a406e56ccbea26 Mon Sep 17 00:00:00 2001 From: Ben Roberts Date: Tue, 11 Dec 2012 22:47:50 -0700 Subject: [PATCH] made things prettier --- rest_framework/tests/serializer.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/rest_framework/tests/serializer.py b/rest_framework/tests/serializer.py index 76b37e4f3..0c11c07b8 100644 --- a/rest_framework/tests/serializer.py +++ b/rest_framework/tests/serializer.py @@ -769,8 +769,6 @@ class DepthTest(TestCase): class NestedSerializerContextTests(TestCase): def test_nested_serializer_context(self): - class AlbumCollection(object): - albums = None class PhotoSerializer(serializers.ModelSerializer): class Meta: @@ -797,15 +795,24 @@ class NestedSerializerContextTests(TestCase): raise RuntimeError("context isn't getting passed into 1st level nested serializer") return "success" + class AlbumCollection(object): + albums = None + class AlbumCollectionSerializer(serializers.Serializer): - albums = AlbumSerializer(source="albums", ) + albums = AlbumSerializer(source="albums") + 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.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.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