fix: pydantic2 models

Author: rubenfonseca

aws_lambda_powertools/event_handler/openapi/models.py

@@ -373,7 +373,7 @@ class Operation(BaseModel):
     parameters: Optional[List[Union[Parameter, Reference]]] = None
     requestBody: Optional[Union[RequestBody, Reference]] = None
     # Using Any for Specification Extensions
-    responses: Optional[Dict[str, Union[Response, Any]]] = None
+    responses: Optional[Dict[int, Union[Response, Any]]] = None
     callbacks: Optional[Dict[str, Union[Dict[str, "PathItem"], Reference]]] = None
     deprecated: Optional[bool] = None
     security: Optional[List[Dict[str, List[str]]]] = None

tests/functional/event_handler/test_openapi_params.py

@@ -39,8 +39,8 @@ def handler():
     assert get.operationId == "handler__get"
 
     assert get.responses is not None
-    assert "200" in get.responses.keys()
-    response = get.responses["200"]
+    assert 200 in get.responses.keys()
+    response = get.responses[200]
     assert response.description == "Successful Response"
 
     assert JSON_CONTENT_TYPE in response.content
@@ -140,7 +140,7 @@ def handler() -> str:
     get = schema.paths["/"].get
     assert get.parameters is None
 
-    response = get.responses["200"].content[JSON_CONTENT_TYPE]
+    response = get.responses[200].content[JSON_CONTENT_TYPE]
     assert response.schema_.title == "Return"
     assert response.schema_.type == "string"
 
@@ -161,7 +161,7 @@ def handler() -> User:
     get = schema.paths["/"].get
     assert get.parameters is None
 
-    response = get.responses["200"].content[JSON_CONTENT_TYPE]
+    response = get.responses[200].content[JSON_CONTENT_TYPE]
     reference = response.schema_
     assert reference.ref == "#/components/schemas/User"
 
@@ -214,7 +214,7 @@ def handler() -> User:
     get = schema.paths["/"].get
     assert get.parameters is None
 
-    response = get.responses["200"].content[JSON_CONTENT_TYPE]
+    response = get.responses[200].content[JSON_CONTENT_TYPE]
     reference = response.schema_
     assert reference.ref == "#/components/schemas/User"
 

tests/functional/event_handler/test_openapi_responses.py

@@ -10,11 +10,11 @@ def handler():
 
     schema = app.get_openapi_schema()
     responses = schema.paths["/"].get.responses
-    assert "200" in responses.keys()
-    assert responses["200"].description == "Successful Response"
+    assert 200 in responses.keys()
+    assert responses[200].description == "Successful Response"
 
-    assert "422" in responses.keys()
-    assert responses["422"].description == "Validation Error"
+    assert 422 in responses.keys()
+    assert responses[422].description == "Validation Error"
 
 
 def test_openapi_200_response_with_description():
@@ -26,11 +26,11 @@ def handler():
 
     schema = app.get_openapi_schema()
     responses = schema.paths["/"].get.responses
-    assert "200" in responses.keys()
-    assert responses["200"].description == "Custom response"
+    assert 200 in responses.keys()
+    assert responses[200].description == "Custom response"
 
-    assert "422" in responses.keys()
-    assert responses["422"].description == "Validation Error"
+    assert 422 in responses.keys()
+    assert responses[422].description == "Validation Error"
 
 
 def test_openapi_200_custom_response():
@@ -42,8 +42,8 @@ def handler():
 
     schema = app.get_openapi_schema()
     responses = schema.paths["/"].get.responses
-    assert "202" in responses.keys()
-    assert responses["202"].description == "Custom response"
+    assert 202 in responses.keys()
+    assert responses[202].description == "Custom response"
 
-    assert "200" not in responses.keys()
-    assert "422" not in responses.keys()
+    assert 200 not in responses.keys()
+    assert 422 not in responses.keys()