refactor(e2e-tests): use standard collections for types + refactor code (#6505)

* Using generics types + enabling ruff

* Using generics types + enabling ruff

Author: leandrodamascena

tests/e2e/conftest.py

@@ -1,11 +1,18 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
 import pytest
 
 from tests.e2e.utils.infrastructure import call_once
 from tests.e2e.utils.lambda_layer.powertools_layer import LocalLambdaPowertoolsLayer
 
+if TYPE_CHECKING:
+    from collections.abc import Generator
+
 
 @pytest.fixture(scope="session", autouse=True)
-def lambda_layer_build(tmp_path_factory: pytest.TempPathFactory, worker_id: str) -> str:
+def lambda_layer_build(tmp_path_factory: pytest.TempPathFactory, worker_id: str) -> Generator[Any, Any, Any]:
     """Build Lambda Layer once before stacks are created
 
     Parameters

tests/e2e/data_masking/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from tests.e2e.data_masking.infrastructure import DataMaskingStack

tests/e2e/data_masking/handlers/basic_handler.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from aws_lambda_powertools import Logger
 from aws_lambda_powertools.utilities.data_masking import DataMasking
 from aws_lambda_powertools.utilities.data_masking.provider.kms.aws_encryption_sdk import AWSEncryptionSDKProvider
@@ -17,7 +19,4 @@ def lambda_handler(event, context):
     data_masker = DataMasking(provider=AWSEncryptionSDKProvider(keys=[kms_key]))
     value = [1, 2, "string", 4.5]
     encrypted_data = data_masker.encrypt(value)
-    response = {}
-    response["encrypted_data"] = encrypted_data
-
-    return response
+    return {"encrypted_data": encrypted_data}

tests/e2e/data_masking/infrastructure.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import aws_cdk.aws_kms as kms
 from aws_cdk import CfnOutput, Duration
 from aws_cdk import aws_iam as iam

tests/e2e/data_masking/test_e2e_data_masking.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import json
 from uuid import uuid4
 

tests/e2e/event_handler/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from tests.e2e.event_handler.infrastructure import EventHandlerStack

tests/e2e/event_handler/handlers/alb_handler.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from aws_lambda_powertools.event_handler import (
     ALBResolver,
     CORSConfig,

tests/e2e/event_handler/handlers/alb_handler_with_body_none.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from aws_lambda_powertools.event_handler import (
     ALBResolver,
     Response,

tests/e2e/event_handler/handlers/api_gateway_http_handler.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from aws_lambda_powertools.event_handler import (
     APIGatewayHttpResolver,
     CORSConfig,

tests/e2e/event_handler/handlers/api_gateway_rest_handler.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from aws_lambda_powertools.event_handler import (
     APIGatewayRestResolver,
     CORSConfig,