Adds check for response obj creation in responseschema.

This commit is contained in:
Karen Bradshaw 2015-11-25 09:36:36 -05:00
parent 060b3283b6
commit c77846fce3
1 changed files with 2 additions and 0 deletions

View File

@ -485,6 +485,8 @@ class JSONTranslator(nodes.GenericNodeVisitor):
responses = resource['responses']
status_code = node[0].astext()
filepath = node[1].astext()
if status_code not in responses:
responses[status_code] = new_response
if 'schema' not in responses[status_code]:
responses[status_code]['schema'] = {}
responses[status_code]['schema'] = {'$ref': filepath}