Fix Idempotency test

aws-powertools · Sep 12, 2024 · 8e96bb1 · 8e96bb1
1 parent 1cf39f2
commit 8e96bb1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/tests/e2e/idempotency/infrastructure.py b/tests/e2e/idempotency/infrastructure.py
@@ -1,4 +1,4 @@
-from aws_cdk import CfnOutput, RemovalPolicy
+from aws_cdk import CfnOutput, Duration, RemovalPolicy
 from aws_cdk import aws_dynamodb as dynamodb
 from aws_cdk.aws_dynamodb import Table
 
@@ -7,7 +7,7 @@
 
 class IdempotencyDynamoDBStack(BaseInfrastructure):
     def create_resources(self):
-        table = self._create_dynamodb_table()
+        table = self._create_dynamodb_table(function_props={"timeout": Duration.seconds(10)})
 
         env_vars = {"IdempotencyTable": table.table_name}
         functions = self.create_lambda_functions(function_props={"environment": env_vars})

diff --git a/tests/e2e/idempotency/test_idempotency_dynamodb.py b/tests/e2e/idempotency/test_idempotency_dynamodb.py
@@ -83,7 +83,7 @@ def test_ttl_caching_expiration_idempotency(ttl_cache_expiration_handler_fn_arn:
 def test_ttl_caching_timeout_idempotency(ttl_cache_timeout_handler_fn_arn: str):
     # GIVEN
     payload_timeout_execution = json.dumps(
-        {"sleep": 5, "message": "Powertools for AWS Lambda (Python) - TTL 1s"},
+        {"sleep": 12, "message": "Powertools for AWS Lambda (Python) - TTL 1s"},
         sort_keys=True,
     )
     payload_working_execution = json.dumps(

diff --git a/tests/e2e/idempotency_redis/infrastructure.py b/tests/e2e/idempotency_redis/infrastructure.py
@@ -1,6 +1,7 @@
 import time
 from typing import Tuple
 
+from aws_cdk import Duration
 from aws_cdk import aws_ec2 as ec2
 from aws_cdk.aws_ec2 import (
     SecurityGroup,
@@ -30,6 +31,7 @@ def create_resources(self) -> None:
                 "environment": env_vars,
                 "vpc": vpc_stack,
                 "security_groups": [security_groups[1]],
+                "timeout": Duration.seconds(10),
             },
         )
 

diff --git a/tests/e2e/idempotency_redis/test_idempotency_redis.py b/tests/e2e/idempotency_redis/test_idempotency_redis.py
@@ -69,7 +69,7 @@ def test_ttl_caching_expiration_idempotency(ttl_cache_expiration_handler_fn_arn:
 def test_ttl_caching_timeout_idempotency(ttl_cache_timeout_handler_fn_arn: str):
     # GIVEN
     payload_timeout_execution = json.dumps(
-        {"sleep": 5, "message": "Powertools for AWS Lambda (Python) - TTL 1s"},
+        {"sleep": 12, "message": "Powertools for AWS Lambda (Python) - TTL 1s"},
         sort_keys=True,
     )
     payload_working_execution = json.dumps(