Addressing Simon's feedback

Author: leandrodamascena

tests/unit/event_handler/__init__.py

@@ -0,0 +1,18 @@
+import pytest
+from pydantic import __version__
+
+
+@pytest.fixture(scope="session")
+def pydanticv1_only():
+
+    version = __version__.split(".")
+    if version[0] != "1":
+        pytest.skip("pydanticv1 test only")
+
+
+@pytest.fixture(scope="session")
+def pydanticv2_only():
+
+    version = __version__.split(".")
+    if version[0] != "2":
+        pytest.skip("pydanticv2 test only")

tests/unit/event_handler/_pydantic/__init__.py

@@ -0,0 +1,35 @@
+import pytest
+
+from aws_lambda_powertools.event_handler.openapi.models import OpenAPIExtensions
+
+
+@pytest.mark.usefixtures("pydanticv1_only")
+def test_openapi_extensions_with_dict():
+    # GIVEN we create an OpenAPIExtensions object with a dict
+    extensions = OpenAPIExtensions(openapi_extensions={"x-amazon-apigateway": {"foo": "bar"}})
+
+    # THEN we get a dict with the extension
+    assert extensions.dict(exclude_none=True) == {"x-amazon-apigateway": {"foo": "bar"}}
+
+
+@pytest.mark.usefixtures("pydanticv1_only")
+def test_openapi_extensions_with_proxy_models():
+
+    # GIVEN we create an models using OpenAPIExtensions as a "Proxy" Model
+    class MyModelFoo(OpenAPIExtensions):
+        foo: str
+
+    class MyModelBar(OpenAPIExtensions):
+        bar: str
+        foo: MyModelFoo
+
+    value_to_serialize = MyModelBar(
+        bar="bar",
+        foo=MyModelFoo(foo="foo"),
+        openapi_extensions={"x-amazon-apigateway": {"foo": "bar"}},
+    )
+
+    value_to_return = value_to_serialize.dict(exclude_none=True)
+
+    # THEN we get a dict with the value serialized
+    assert value_to_return == {"bar": "bar", "foo": {"foo": "foo"}, "x-amazon-apigateway": {"foo": "bar"}}

tests/unit/event_handler/_pydantic/conftest.py

@@ -0,0 +1,35 @@
+import pytest
+
+from aws_lambda_powertools.event_handler.openapi.models import OpenAPIExtensions
+
+
+@pytest.mark.usefixtures("pydanticv2_only")
+def test_openapi_extensions_with_dict():
+    # GIVEN we create an OpenAPIExtensions object with a dict
+    extensions = OpenAPIExtensions(openapi_extensions={"x-amazon-apigateway": {"foo": "bar"}})
+
+    # THEN we get a dict with the extension
+    assert extensions.model_dump(exclude_none=True) == {"x-amazon-apigateway": {"foo": "bar"}}
+
+
+@pytest.mark.usefixtures("pydanticv2_only")
+def test_openapi_extensions_with_proxy_models():
+
+    # GIVEN we create an models using OpenAPIExtensions as a "Proxy" Model
+    class MyModelFoo(OpenAPIExtensions):
+        foo: str
+
+    class MyModelBar(OpenAPIExtensions):
+        bar: str
+        foo: MyModelFoo
+
+    value_to_serialize = MyModelBar(
+        bar="bar",
+        foo=MyModelFoo(foo="foo"),
+        openapi_extensions={"x-amazon-apigateway": {"foo": "bar"}},
+    )
+
+    value_to_return = value_to_serialize.model_dump(exclude_none=True)
+
+    # THEN we get a dict with the value serialized
+    assert value_to_return == {"bar": "bar", "foo": {"foo": "foo"}, "x-amazon-apigateway": {"foo": "bar"}}