diff options
| author | Craig de Stigter | 2013-10-03 11:34:42 +1300 | 
|---|---|---|
| committer | Craig de Stigter | 2013-10-03 11:34:42 +1300 | 
| commit | dc650f77b5f377bbfab3da66455feb57b195a1da (patch) | |
| tree | 41859c69f42912a15bd1e7224b90187287f5d475 /rest_framework | |
| parent | 94cd1369437f84adefb46462439b46dc5208ab1d (diff) | |
| download | django-rest-framework-dc650f77b5f377bbfab3da66455feb57b195a1da.tar.bz2 | |
add tests for transform_fieldname methods
Diffstat (limited to 'rest_framework')
| -rw-r--r-- | rest_framework/tests/test_serializer.py | 35 | 
1 files changed, 35 insertions, 0 deletions
diff --git a/rest_framework/tests/test_serializer.py b/rest_framework/tests/test_serializer.py index c2497660..ca876eae 100644 --- a/rest_framework/tests/test_serializer.py +++ b/rest_framework/tests/test_serializer.py @@ -1643,3 +1643,38 @@ class SerializerSupportsManyRelationships(TestCase):          serializer = SimpleSlugSourceModelSerializer(data={'text': 'foo', 'targets': [1, 2]})          self.assertTrue(serializer.is_valid())          self.assertEqual(serializer.data, {'text': 'foo', 'targets': [1, 2]}) + + +class TransformMethodsSerializer(serializers.Serializer): +    a = serializers.CharField() +    b_renamed = serializers.CharField(source='b') + +    def transform_a(self, obj, value): +        return value.lower() + +    def transform_b_renamed(self, obj, value): +        if value is not None: +            return 'and ' + value + + +class TestSerializerTransformMethods(TestCase): +    def setUp(self): +        self.s = TransformMethodsSerializer() + +    def test_transform_methods(self): +        self.assertEqual( +            self.s.to_native({'a': 'GREEN EGGS', 'b': 'HAM'}), +            { +                'a': 'green eggs', +                'b_renamed': 'and HAM', +            } +        ) + +    def test_missing_fields(self): +        self.assertEqual( +            self.s.to_native({'a': 'GREEN EGGS'}), +            { +                'a': 'green eggs', +                'b_renamed': None, +            } +        )  | 
