Handle situation where no schema is defined.

Author: mjp4

connexion/operations/openapi.py

@@ -186,7 +186,11 @@ def example_response(self, status_code=None, content_type=None):
         except KeyError:
             pass
 
-        schema = deep_get(self._responses, schema_path)
+        try:
+            schema = deep_get(self._responses, schema_path)
+        except KeyError:
+            return ("No example response or response schema defined.", status_code)
+
         return (build_example_from_schema(schema), status_code)
 
     def get_path_parameter_types(self):

connexion/operations/swagger2.py

@@ -208,7 +208,11 @@ def example_response(self, status_code=None, *args, **kwargs):
         except KeyError:
             pass
 
-        schema = deep_get(self._responses, schema_path)
+        try:
+            schema = deep_get(self._responses, schema_path)
+        except KeyError:
+            return ("No example response or response schema defined.", status_code)
+
         return (build_example_from_schema(schema), status_code)
 
     def body_name(self, content_type: str = None) -> str:

tests/test_mock.py

@@ -280,7 +280,7 @@ def test_mock_resolver_no_examples():
 
     response, status_code = resolver.mock_operation(operation)
     assert status_code == 418
-    assert response == "No example response was defined."
+    assert response == "No example response or response schema defined."
 
 
 def test_mock_resolver_notimplemented():
@@ -317,4 +317,7 @@ def test_mock_resolver_notimplemented():
     )
 
     # check if it is using the mock function
-    assert operation._resolution.function() == ("No example response was defined.", 418)
+    assert operation._resolution.function() == (
+        "No example response or response schema defined.",
+        418,
+    )

tests/test_mock3.py

@@ -103,7 +103,7 @@ def test_mock_resolver_no_examples():
 
     response, status_code = resolver.mock_operation(operation)
     assert status_code == 418
-    assert response == "No example response was defined."
+    assert response == "No example response or response schema defined."
 
 
 def test_mock_resolver_notimplemented():
@@ -133,4 +133,7 @@ def test_mock_resolver_notimplemented():
         resolver=resolver,
     )
     # check if it is using the mock function
-    assert operation._resolution.function() == ("No example response was defined.", 418)
+    assert operation._resolution.function() == (
+        "No example response or response schema defined.",
+        418,
+    )