Added test for None found while resolving the source attribute of a field

This commit is contained in:
dorenthas 2013-02-07 11:23:38 -05:00
parent 4dcc34ecae
commit badf75e5e1

View File

@ -785,6 +785,23 @@ class RelatedTraversalTest(TestCase):
self.assertEqual(serializer.data, expected) self.assertEqual(serializer.data, expected)
def test_nested_traversal_with_none(self):
from rest_framework.tests.models import NullableForeignKeySource
instance = NullableForeignKeySource.objects.create(name='Source with null FK')
class NullableSourceSerializer(serializers.Serializer):
target_name = serializers.Field(source='target.name')
serializer = NullableSourceSerializer(instance=instance)
expected = {
'name': 'Source with null FK',
'target_name': None,
}
self.assertEqual(serializer.data, expected)
class SerializerMethodFieldTests(TestCase): class SerializerMethodFieldTests(TestCase):
def setUp(self): def setUp(self):