doc generation: adapt for error_builder

hapic/doc.py

@@ -50,7 +50,7 @@ def bottle_generate_operations(
 
     if description.errors:
         for error in description.errors:
-            schema_class = type(error.wrapper.schema)
+            schema_class = type(error.wrapper.error_builder)
             method_operations.setdefault('responses', {})\
                 [int(error.wrapper.http_code)] = {
                     'description': int(error.wrapper.http_code),
@@ -153,7 +153,7 @@ class DocGenerator(object):
 
             if description.errors:
                 for error in description.errors:
-                    schemas.append(type(error.wrapper.schema))
+                    schemas.append(type(error.wrapper.error_builder))
 
         for schema in set(schemas):
             spec.definition(schema.__name__, schema=schema)

tests/func/test_doc.py

@@ -170,3 +170,28 @@ class TestDocGeneration(Base):
         assert 'post' in doc['paths']['/upload']
         assert 'tags' in doc['paths']['/upload']['post']
         assert ['foo', 'bar'] == doc['paths']['/upload']['post']['tags']
+
+    def test_func__errors__nominal_case(self):
+        hapic = Hapic()
+        app = bottle.Bottle()
+        hapic.set_context(MyContext(app=app))
+
+        @hapic.with_api_doc()
+        @hapic.handle_exception()
+        def my_controller(hapic_data=None):
+            assert hapic_data
+
+        app.route('/upload', method='POST', callback=my_controller)
+        doc = hapic.generate_doc()
+
+        assert doc.get('paths')
+        assert '/upload' in doc['paths']
+        assert 'post' in doc['paths']['/upload']
+        assert 'responses' in doc['paths']['/upload']['post']
+        assert 500 in doc['paths']['/upload']['post']['responses']
+        assert {
+            'description': 500,
+            'schema': {
+                '$ref': '#/definitions/DefaultErrorBuilder'
+            }
+        } == doc['paths']['/upload']['post']['responses'][500]