Add e2e tests

Author: leandrodamascena

tests/e2e/parser/handlers/handler_with_model_type_class.py

@@ -0,0 +1,23 @@
+import json
+from typing import Any, Dict, Type, Union
+
+from pydantic import BaseModel
+
+from aws_lambda_powertools.utilities.parser import parse
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+AnyInheritedModel = Union[Type[BaseModel], BaseModel]
+RawDictOrModel = Union[Dict[str, Any], AnyInheritedModel]
+
+
+class ModelWithUnionType(BaseModel):
+    name: str
+    profile: RawDictOrModel
+
+
+def lambda_handler(event: ModelWithUnionType, context: LambdaContext):
+    event = json.dumps(event)
+
+    event_parsed = parse(event=event, model=ModelWithUnionType)
+
+    return {"name": event_parsed.name}

tests/e2e/parser/test_parser.py

@@ -20,6 +20,11 @@ def handler_with_dataclass_arn(infrastructure: dict) -> str:
     return infrastructure.get("HandlerWithDataclass", "")
 
 
+@pytest.fixture
+def handler_with_type_model_class(infrastructure: dict) -> str:
+    return infrastructure.get("HandlerWithModelTypeClass", "")
+
+
 @pytest.mark.xdist_group(name="parser")
 def test_parser_with_basic_model(handler_with_basic_model_arn):
     # GIVEN
@@ -66,3 +71,19 @@ def test_parser_with_dataclass(handler_with_dataclass_arn):
     ret = parser_execution["Payload"].read().decode("utf-8")
 
     assert "powertools" in ret
+
+
+@pytest.mark.xdist_group(name="parser")
+def test_parser_with_type_model(handler_with_type_model_class):
+    # GIVEN
+    payload = json.dumps({"name": "powertools", "profile": {"description": "python", "size": "XXL"}})
+
+    # WHEN
+    parser_execution, _ = data_fetcher.get_lambda_response(
+        lambda_arn=handler_with_type_model_class,
+        payload=payload,
+    )
+
+    ret = parser_execution["Payload"].read().decode("utf-8")
+
+    assert "powertools" in ret