From 2978f148642c09c71dfbf016a685ff3af59b91ac Mon Sep 17 00:00:00 2001 From: Dhaval Mehta <20968146+dhaval-mehta@users.noreply.github.com> Date: Tue, 26 May 2020 02:49:46 +0530 Subject: [PATCH] add test case for get_schema_view function --- tests/schemas/test_get_schema_view.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/schemas/test_get_schema_view.py b/tests/schemas/test_get_schema_view.py index f582c6495..91f055529 100644 --- a/tests/schemas/test_get_schema_view.py +++ b/tests/schemas/test_get_schema_view.py @@ -7,6 +7,7 @@ from rest_framework.schemas import coreapi, get_schema_view, openapi class GetSchemaViewTests(TestCase): """For the get_schema_view() helper.""" + def test_openapi(self): schema_view = get_schema_view(title="With OpenAPI") assert isinstance(schema_view.initkwargs['schema_generator'], openapi.SchemaGenerator) @@ -18,3 +19,11 @@ class GetSchemaViewTests(TestCase): schema_view = get_schema_view(title="With CoreAPI") assert isinstance(schema_view.initkwargs['schema_generator'], coreapi.SchemaGenerator) 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'}]