def test_response_primitive():
expected_openapi_dict = {
'200': {
'description': 'Test Response',
'content': {
'application/json': {
'schema': {
'type': 'string'
}
}
}
}
}
openapi_response = OpenApiResponse('Test Response', data_type=str)
assert expected_openapi_dict == openapi_response.as_dict()
def test_response_object():
expected_openapi_dict = {
'200': {
'description': 'Test Response',
'content': {
'application/json': {
'schema': {
'$ref': '#/components/schemas/DataclassObject'
}
}
}
}
}
openapi_response = OpenApiResponse('Test Response',
data_type=DataclassObject)
assert expected_openapi_dict == openapi_response.as_dict()
def test_response_empty():
expected_openapi_dict = {'201': {'description': 'Test Empty Response'}}
openapi_response = OpenApiResponse('Test Empty Response', '201')
assert expected_openapi_dict == openapi_response.as_dict()