Type LambdaContext types in tests

Author: basvandriel

examples/batch_processing/src/getting_started_with_test.py

@@ -11,15 +11,16 @@ def load_event(path: Path):
         return json.load(f)
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
-        aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
+    aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
@@ -29,7 +30,7 @@ def sqs_event():
     return load_event(path=Path("events/sqs_event.json"))
 
 
-def test_app_batch_partial_response(sqs_event, lambda_context):
+def test_app_batch_partial_response(sqs_event, lambda_context: LambdaContext):
     # GIVEN
     processor_result = processor  # access processor for additional assertions
     successful_record = sqs_event["Records"][0]

examples/data_masking/tests/test_lambda_mask.py

@@ -4,22 +4,23 @@
 import test_lambda_mask
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:111111111:function:test"
-        aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:111111111:function:test"
+    aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+
+    def get_remaining_time_in_millis(self) -> int:
+        return 5
 
-        def get_remaining_time_in_millis(self) -> int:
-            return 5
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_encrypt_lambda(lambda_context):
+def test_encrypt_lambda(lambda_context: LambdaContext):
     # GIVEN: A sample event for testing
     event = {"testkey": "testvalue"}
 

examples/event_handler_bedrock_agents/src/assert_bedrock_agent_response.py

@@ -4,19 +4,20 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_lambda_handler(lambda_context):
+def test_lambda_handler(lambda_context: LambdaContext):
     minimal_event = {
         "apiPath": "/current_time",
         "httpMethod": "GET",

examples/event_handler_graphql/src/assert_async_graphql_response.py

@@ -10,20 +10,21 @@
 )
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
 @pytest.mark.asyncio
-async def test_async_direct_resolver(lambda_context):
+async def test_async_direct_resolver(lambda_context: LambdaContext):
     # GIVEN
     fake_event = json.loads(Path("assert_async_graphql_response.json").read_text())
 

examples/event_handler_graphql/src/assert_graphql_response.py

@@ -8,19 +8,20 @@
 from assert_graphql_response_module import Location, app  # instance of AppSyncResolver
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_direct_resolver(lambda_context):
+def test_direct_resolver(lambda_context: LambdaContext):
     # GIVEN
     fake_event = json.loads(Path("assert_graphql_response.json").read_text())
 

examples/event_handler_rest/src/assert_alb_api_resolver_response.py

@@ -4,19 +4,20 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_lambda_handler(lambda_context):
+def test_lambda_handler(lambda_context: LambdaContext):
     minimal_event = {
         "path": "/todos",
         "httpMethod": "GET",

examples/event_handler_rest/src/assert_function_url_api_resolver_response.py

@@ -4,19 +4,20 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_lambda_handler(lambda_context):
+def test_lambda_handler(lambda_context: LambdaContext):
     minimal_event = {
         "rawPath": "/todos",
         "requestContext": {

examples/event_handler_rest/src/assert_http_api_resolver_response.py

@@ -4,19 +4,20 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_lambda_handler(lambda_context):
+def test_lambda_handler(lambda_context: LambdaContext):
     minimal_event = {
         "rawPath": "/todos",
         "requestContext": {

examples/event_handler_rest/src/assert_rest_api_resolver_response.py

@@ -4,19 +4,20 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
-        aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:123456789012:function:test"
+    aws_request_id: str = "da658bd3-2d6f-4e7b-8ec2-937234644fdc"
+
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_lambda_handler(lambda_context):
+def test_lambda_handler(lambda_context: LambdaContext):
     minimal_event = {
         "path": "/todos",
         "httpMethod": "GET",

examples/idempotency/tests/test_disabling_idempotency_utility.py

@@ -4,22 +4,23 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
-        aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
+    aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+
+    def get_remaining_time_in_millis(self) -> int:
+        return 5
 
-        def get_remaining_time_in_millis(self) -> int:
-            return 5
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_idempotent_lambda_handler(monkeypatch, lambda_context):
+def test_idempotent_lambda_handler(monkeypatch, lambda_context: LambdaContext):
     # Set POWERTOOLS_IDEMPOTENCY_DISABLED before calling decorated functions
     monkeypatch.setenv("POWERTOOLS_IDEMPOTENCY_DISABLED", 1)
 

examples/idempotency/tests/test_with_dynamodb_local.py

@@ -5,22 +5,23 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
-        aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
+    aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+
+    def get_remaining_time_in_millis(self) -> int:
+        return 5
 
-        def get_remaining_time_in_millis(self) -> int:
-            return 5
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_idempotent_lambda(lambda_context):
+def test_idempotent_lambda(lambda_context: LambdaContext):
     # Configure the boto3 to use the endpoint for the DynamoDB Local instance
     dynamodb_local_client = boto3.client("dynamodb", endpoint_url="http://localhost:8000")
     app_test_dynamodb_local.persistence_layer.client = dynamodb_local_client

examples/idempotency/tests/test_with_io_operations.py

@@ -5,22 +5,23 @@
 import pytest
 
 
-@pytest.fixture
-def lambda_context():
-    @dataclass
-    class LambdaContext:
-        function_name: str = "test"
-        memory_limit_in_mb: int = 128
-        invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
-        aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+@dataclass
+class LambdaContext:
+    function_name: str = "test"
+    memory_limit_in_mb: int = 128
+    invoked_function_arn: str = "arn:aws:lambda:eu-west-1:809313241:function:test"
+    aws_request_id: str = "52fdfc07-2182-154f-163f-5f0f9a621d72"
+
+    def get_remaining_time_in_millis(self) -> int:
+        return 5
 
-        def get_remaining_time_in_millis(self) -> int:
-            return 5
 
+@pytest.fixture
+def lambda_context() -> LambdaContext:
     return LambdaContext()
 
 
-def test_idempotent_lambda(lambda_context):
+def test_idempotent_lambda(lambda_context: LambdaContext):
     mock_client = MagicMock()
     app_test_io_operations.persistence_layer.client = mock_client
     result = app_test_io_operations.handler({"testkey": "testvalue"}, lambda_context)