diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_fields.py | 25 | 
1 files changed, 25 insertions, 0 deletions
| diff --git a/tests/test_fields.py b/tests/test_fields.py index ab3418bd..7f5f8102 100644 --- a/tests/test_fields.py +++ b/tests/test_fields.py @@ -93,6 +93,31 @@ class TestSource:              "same as the field name. Remove the `source` keyword argument."          ) +    def test_callable_source(self): +        class ExampleSerializer(serializers.Serializer): +            example_field = serializers.CharField(source='example_callable') + +        class ExampleInstance(object): +            def example_callable(self): +                return 'example callable value' + +        serializer = ExampleSerializer(ExampleInstance()) +        assert serializer.data['example_field'] == 'example callable value' + +    def test_callable_source_raises(self): +        class ExampleSerializer(serializers.Serializer): +            example_field = serializers.CharField(source='example_callable', read_only=True) + +        class ExampleInstance(object): +            def example_callable(self): +                raise AttributeError('method call failed') + +        with pytest.raises(ValueError) as exc_info: +            serializer = ExampleSerializer(ExampleInstance()) +            serializer.data.items() + +        assert 'method call failed' in str(exc_info.value) +  class TestReadOnly:      def setup(self): | 
