encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-07 13:54:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use get_schema_view in tests

Browse Source
This commit is contained in:
Tom Christie 2016-10-07 12:09:41 +01:00
parent 072442025c
commit 3eb7fe6f64
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_schemas.py
View File

@ -7,7 +7,7 @@ from rest_framework import filters, pagination, permissions, serializers
from rest_framework.compat import coreapi from rest_framework.compat import coreapi
from rest_framework.decorators import detail_route, list_route from rest_framework.decorators import detail_route, list_route
from rest_framework.routers import DefaultRouter from rest_framework.routers import DefaultRouter
from rest_framework.schemas import SchemaGenerator from rest_framework.schemas import SchemaGenerator, get_schema_view
from rest_framework.test import APIClient from rest_framework.test import APIClient
from rest_framework.views import APIView from rest_framework.views import APIView
from rest_framework.viewsets import ModelViewSet from rest_framework.viewsets import ModelViewSet
@ -65,9 +65,16 @@ class ExampleViewSet(ModelViewSet):
return super(ExampleViewSet, self).get_serializer(*args, **kwargs) return super(ExampleViewSet, self).get_serializer(*args, **kwargs)
router = DefaultRouter(schema_title='Example API' if coreapi else None) if coreapi:
schema_view = get_schema_view(title='Example API')
else:
def schema_view(request):
pass
router = DefaultRouter()
router.register('example', ExampleViewSet, base_name='example') router.register('example', ExampleViewSet, base_name='example')
urlpatterns = [ urlpatterns = [
url(r'^$', schema_view),
url(r'^', include(router.urls)) url(r'^', include(router.urls))
] ]