Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix @app.param and @app.header producing invalid Swagger/OpenAPI document #654

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
10 changes: 6 additions & 4 deletions flask_restplus/namespace.py
Original file line number Diff line number Diff line change
Expand Up @@ -267,9 +267,10 @@ def param(self, name, description=None, _in='query', **kwargs):
:param str description: a small description
:param str _in: the parameter location `(query|header|formData|body|cookie)`
'''
param = kwargs
param = kwargs.copy()
param['in'] = _in
param['description'] = description
if description is not None:
param['description'] = description
return self.doc(params={name: param})

def response(self, code, description, model=None, **kwargs):
Expand All @@ -291,8 +292,9 @@ def header(self, name, description=None, **kwargs):
:param str description: a description about the header

'''
header = {'description': description}
header.update(kwargs)
header = kwargs.copy()
if description is not None:
header['description'] = description
return self.doc(headers={name: header})

def produces(self, mimetypes):
Expand Down
52 changes: 52 additions & 0 deletions tests/test_swagger.py
Original file line number Diff line number Diff line change
Expand Up @@ -842,6 +842,25 @@ def get(self, age):
assert parameter['in'] == 'formData'
assert parameter['description'] == 'A query string'

def test_explicit_parameters_with_decorator_no_description(self, api, client):
@api.route('/name/')
class ByNameResource(restplus.Resource):
@api.param('q', type='string', _in='formData')
def get(self, age):
return {}

data = client.get_specs()
assert '/name/' in data['paths']

op = data['paths']['/name/']['get']
assert len(op['parameters']) == 1

parameter = op['parameters'][0]
assert parameter['name'] == 'q'
assert parameter['type'] == 'string'
assert parameter['in'] == 'formData'
assert 'description' not in parameter

def test_class_explicit_parameters(self, api, client):
@api.route('/name/<int:age>/', endpoint='by-name', doc={
'params': {
Expand Down Expand Up @@ -876,6 +895,39 @@ def get(self, age):
assert parameter['in'] == 'query'
assert parameter['description'] == 'A query string'

def test_class_explicit_parameters_no_description(self, api, client):
@api.route('/name/<int:age>/', endpoint='by-name', doc={
'params': {
'q': {
'type': 'string',
'in': 'query',
}
}
})
class ByNameResource(restplus.Resource):
def get(self, age):
return {}

data = client.get_specs()
assert '/name/{age}/' in data['paths']

path = data['paths']['/name/{age}/']
assert len(path['parameters']) == 2

by_name = dict((p['name'], p) for p in path['parameters'])

parameter = by_name['age']
assert parameter['name'] == 'age'
assert parameter['type'] == 'integer'
assert parameter['in'] == 'path'
assert parameter['required'] is True

parameter = by_name['q']
assert parameter['name'] == 'q'
assert parameter['type'] == 'string'
assert parameter['in'] == 'query'
assert 'description' not in parameter

def test_explicit_parameters_override(self, api, client):
@api.route('/name/<int:age>/', endpoint='by-name', doc={
'params': {
Expand Down