Skip to content

Commit

Permalink
Add test for #382
Browse files Browse the repository at this point in the history
  • Loading branch information
axnsan12 committed Jun 13, 2019
1 parent 1635e5e commit acc204e
Show file tree
Hide file tree
Showing 4 changed files with 27 additions and 2 deletions.
6 changes: 6 additions & 0 deletions testproj/people/views.py
@@ -1,13 +1,19 @@
from rest_framework import viewsets
from rest_framework.pagination import BasePagination

from .models import Identity, Person
from .serializers import IdentitySerializer, PersonSerializer


class UnknownPagination(BasePagination):
paginator_query_args = ['unknown_paginator']


class PersonViewSet(viewsets.ModelViewSet):
model = Person
queryset = Person.objects
serializer_class = PersonSerializer
pagination_class = UnknownPagination


class IdentityViewSet(viewsets.ModelViewSet):
Expand Down
11 changes: 11 additions & 0 deletions testproj/testproj/inspectors.py
@@ -0,0 +1,11 @@
from drf_yasg import openapi
from drf_yasg.inspectors import PaginatorInspector, NotHandled


class UnknownPaginatorInspector(PaginatorInspector):
def get_paginator_parameters(self, paginator):
if hasattr(paginator, 'paginator_query_args'):
return [openapi.Parameter(name=arg, in_=openapi.IN_QUERY, type=openapi.TYPE_STRING)
for arg in getattr(paginator, 'paginator_query_args')]

return NotHandled
7 changes: 6 additions & 1 deletion testproj/testproj/settings/base.py
Expand Up @@ -140,7 +140,12 @@
'clientId': OAUTH2_CLIENT_ID,
'clientSecret': OAUTH2_CLIENT_SECRET,
'appName': OAUTH2_APP_NAME,
}
},
"DEFAULT_PAGINATOR_INSPECTORS": [
'testproj.inspectors.UnknownPaginatorInspector',
'drf_yasg.inspectors.DjangoRestResponsePagination',
'drf_yasg.inspectors.CoreAPICompatInspector',
]
}

REDOC_SETTINGS = {
Expand Down
5 changes: 4 additions & 1 deletion tests/reference.yaml
Expand Up @@ -276,7 +276,10 @@ paths:
get:
operationId: people_list
description: ''
parameters: []
parameters:
- name: unknown_paginator
in: query
type: string
responses:
'200':
description: ''
Expand Down

0 comments on commit acc204e

Please sign in to comment.