encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-10 23:34:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test case should ensure view.request exists.

Browse Source
This commit is contained in:
Tom Christie 2016-08-11 11:13:55 +01:00
parent 30bad6a453
commit 65f552d880
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
rest_framework/schemas.py
View File

@ -263,19 +263,18 @@ class SchemaGenerator(object):
if method not in ('PUT', 'PATCH', 'POST'): if method not in ('PUT', 'PATCH', 'POST'):
return [] return []
if not hasattr(view, 'get_serializer_class'): if not hasattr(view, 'get_serializer'):
return [] return []
fields = []
serializer = view.get_serializer() serializer = view.get_serializer()
if isinstance(serializer, serializers.ListSerializer): if isinstance(serializer, serializers.ListSerializer):
return coreapi.Field(name='data', location='body', required=True) return [coreapi.Field(name='data', location='body', required=True)]
if not isinstance(serializer, serializers.Serializer): if not isinstance(serializer, serializers.Serializer):
return [] return []
fields = []
for field in serializer.fields.values(): for field in serializer.fields.values():
if field.read_only: if field.read_only:
continue continue

4
tests/test_schemas.py
View File

@ -43,6 +43,10 @@ class ExampleViewSet(ModelViewSet):
def custom_action(self, request, pk): def custom_action(self, request, pk):
return super(ExampleSerializer, self).retrieve(self, request) return super(ExampleSerializer, self).retrieve(self, request)
def get_serializer(self, *args, **kwargs):
assert self.request
return super(ExampleViewSet, self).get_serializer(*args, **kwargs)
class ExampleView(APIView): class ExampleView(APIView):
permission_classes = [permissions.IsAuthenticatedOrReadOnly] permission_classes = [permissions.IsAuthenticatedOrReadOnly]