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

add test case for get_schema_view function

Browse Source
This commit is contained in:
Dhaval Mehta 2020-05-26 02:49:46 +05:30
parent 91df8d29ab
commit 2978f14864
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/schemas/test_get_schema_view.py
View File

@ -7,6 +7,7 @@ from rest_framework.schemas import coreapi, get_schema_view, openapi
class GetSchemaViewTests(TestCase): class GetSchemaViewTests(TestCase):
"""For the get_schema_view() helper.""" """For the get_schema_view() helper."""
def test_openapi(self): def test_openapi(self):
schema_view = get_schema_view(title="With OpenAPI") schema_view = get_schema_view(title="With OpenAPI")
assert isinstance(schema_view.initkwargs['schema_generator'], openapi.SchemaGenerator) assert isinstance(schema_view.initkwargs['schema_generator'], openapi.SchemaGenerator)
@ -18,3 +19,11 @@ class GetSchemaViewTests(TestCase):
schema_view = get_schema_view(title="With CoreAPI") schema_view = get_schema_view(title="With CoreAPI")
assert isinstance(schema_view.initkwargs['schema_generator'], coreapi.SchemaGenerator) assert isinstance(schema_view.initkwargs['schema_generator'], coreapi.SchemaGenerator)
assert renderers.CoreAPIOpenAPIRenderer in schema_view.cls().renderer_classes assert renderers.CoreAPIOpenAPIRenderer in schema_view.cls().renderer_classes
def test_tag_objects(self):
schema_view = get_schema_view(
title="With OpenAPI",
tag_objects=[{'name': 'pet', 'description': 'store description'}]
)
assert schema_view.initkwargs['schema_generator'].tag_objects, [
{'name': 'pet', 'description': 'store description'}]