def test_response_spec(): resp = Response('HTTP_200', HTTP_201=DemoModel) spec = resp.generate_spec() assert spec['200']['description'] == DEFAULT_CODE_DESC['HTTP_200'] assert spec['201']['description'] == DEFAULT_CODE_DESC['HTTP_201'] assert spec['201']['content']['application/json']['schema']['$ref'].split( '/')[-1] == 'DemoModel' assert spec.get(200) is None assert spec.get(404) is None
def test_response_spec(): resp = Response("HTTP_200", HTTP_201=DemoModel) spec = resp.generate_spec() assert spec["200"]["description"] == DEFAULT_CODE_DESC["HTTP_200"] assert spec["201"]["description"] == DEFAULT_CODE_DESC["HTTP_201"] assert spec["422"]["description"] == DEFAULT_CODE_DESC["HTTP_422"] assert (spec["201"]["content"]["application/json"]["schema"]["$ref"].split( "/")[-1] == "DemoModel") assert (spec["422"]["content"]["application/json"]["schema"]["$ref"].split( "/")[-1] == "UnprocessableEntity") assert spec.get(200) is None assert spec.get(404) is None