From e43ad2a66562a9a3586d66a2517b89a623dc1141 Mon Sep 17 00:00:00 2001 From: "T. Franzel" Date: Mon, 14 Dec 2020 19:25:27 +0100 Subject: [PATCH] add regression test #217 --- tests/test_regressions.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/tests/test_regressions.py b/tests/test_regressions.py index 81feb209..cd8e09ad 100644 --- a/tests/test_regressions.py +++ b/tests/test_regressions.py @@ -1097,6 +1097,20 @@ def view_func(request, format=None): view_func.cls.authentication_classes = [CustomTokenAuthentication] schema = generate_schema('x', view_function=view_func) - validate_schema(schema) - assert schema['components']['securitySchemes']['tokenAuth']['scheme'] == 'bearer' + + +@pytest.mark.parametrize('responses', [ + str, + OpenApiTypes.STR, + {'200': str}, + {'200': OpenApiTypes.STR}, +]) +def test_string_response_variations(no_warnings, responses): + @extend_schema(responses=responses) + @api_view(['GET']) + def view_func(request, format=None): + pass # pragma: no cover + + schema = generate_schema('x', view_function=view_func) + assert get_response_schema(schema['paths']['/x']['get'])['type'] == 'string'