a

Author: Ran Isenberg

service/handlers/handle_create_order.py

@@ -14,7 +14,7 @@
 from service.handlers.utils.rest_api_resolver import ORDERS_PATH, app
 from service.logic.create_order import create_order
 from service.models.input import CreateOrderRequest
-from service.models.output import CreateOrderOutput, InternalServerErrorOutput
+from service.models.output import CreateOrderOutput, InternalServerErrorOutput, InvalidRestApiRequest
 
 
 @app.post(
@@ -25,11 +25,15 @@
     responses={
         200: {
             'description': 'The created order',
-            'content': {'application/json': {'schema': {'$ref': '#/components/schemas/CreateOrderOutput'}}},
+            'content': {'application/json': {'model': CreateOrderOutput}},
+        },
+        442: {
+            'description': 'Invalid create order request',
+            'content': {'application/json': {'model': InvalidRestApiRequest}},
         },
         501: {
             'description': 'Internal server error',
-            'content': {'application/json': {'schema': InternalServerErrorOutput.model_json_schema()}},
+            'content': {'application/json': {'model': InternalServerErrorOutput}},
         },
     },
     tags=['CRUD'],

service/models/output.py

@@ -1,3 +1,5 @@
+from typing import List
+
 from pydantic import BaseModel
 
 from service.models.order import Order
@@ -12,3 +14,11 @@ class CreateOrderOutput(Order):
 
 class InternalServerErrorOutput(BaseModel):
     error: str = 'internal server error'
+
+
+class Error(BaseModel):
+    x: str
+
+
+class InvalidRestApiRequest(BaseModel):
+    details: List[Error]