def test_get_operation_with_serializer():
view = View()
route = get_route(Controller.post_with_serializer)
View.post.route = route
reference_resolver = openapi.ReferenceResolver('definitions',
'parameters',
force_init=True)
auto_schema = SwaggerAutoSchema(view, 'path', route.http_method,
reference_resolver, 'request', {})
operation = auto_schema.get_operation(['test_app', 'post'])
schema_ref = openapi.SchemaRef(
resolver=reference_resolver,
schema_name='User',
)
parameters = [
openapi.Parameter(
name='data',
in_=openapi.IN_BODY,
required=True,
schema=schema_ref,
),
]
responses = openapi.Responses({
'200':
openapi.Response(
description='',
schema=openapi.Schema(
type=openapi.TYPE_OBJECT,
properties={
'name': {
'type': 'string',
},
},
),
),
})
assert operation == openapi.Operation(
operation_id='Controller.post_with_serializer',
responses=responses,
consumes=['application/json'],
produces=['application/json'],
tags=['test_app'],
parameters=parameters,
)
del View.post.route
def test_get_operation_without_route():
view = View()
reference_resolver = openapi.ReferenceResolver('definitions',
'parameters',
force_init=True)
auto_schema = SwaggerAutoSchema(view, 'path', 'get', reference_resolver,
'request', {})
operation = auto_schema.get_operation(['test_app', 'post'])
parameters = []
responses = openapi.Responses({
'200': openapi.Response(description=''),
})
assert operation == openapi.Operation(
description='',
operation_id='test_app_post',
responses=responses,
consumes=['application/json'],
produces=['application/json'],
tags=['test_app'],
parameters=parameters,
)
def test_get_operation():
view = View()
route = get_route(Controller.post)
View.post.route = route
auto_schema = SwaggerAutoSchema(view, 'path', route.http_method,
'components', 'request', {})
operation = auto_schema.get_operation(['test_app', 'post'])
parameters = [
openapi.Parameter(
name='data',
in_=openapi.IN_BODY,
required=True,
schema=openapi.Schema(
title='UserDTO',
type=openapi.TYPE_OBJECT,
properties={
'name': {
'type': openapi.TYPE_STRING,
},
},
),
),
openapi.Parameter(
name='path_param',
in_=openapi.IN_PATH,
description='',
required=True,
type=openapi.TYPE_INTEGER,
),
openapi.Parameter(
name='query_param',
in_=openapi.IN_QUERY,
description='',
required=True,
type=openapi.TYPE_INTEGER,
),
]
responses = openapi.Responses({
'200':
openapi.Response(
description='',
schema=openapi.Schema(
type=openapi.TYPE_OBJECT,
properties={
'name': {
'type': 'string',
},
},
),
),
})
assert operation == openapi.Operation(
operation_id='Controller.post',
responses=responses,
consumes=['application/json; charset=utf-8'],
produces=['application/json; charset=utf-8'],
description='This is post method',
tags=['test_app'],
parameters=parameters,
)