refactor(e2e): make table name dynamic

Author: heitorlessa

tests/e2e/idempotency/handlers/parallel_execution_handler.py

@@ -1,8 +1,10 @@
+import os
 import time
 
 from aws_lambda_powertools.utilities.idempotency import DynamoDBPersistenceLayer, idempotent
 
-persistence_layer = DynamoDBPersistenceLayer(table_name="IdempotencyTable")
+TABLE_NAME = os.getenv("IdempotencyTable", "")
+persistence_layer = DynamoDBPersistenceLayer(table_name=TABLE_NAME)
 
 
 @idempotent(persistence_store=persistence_layer)

tests/e2e/idempotency/handlers/ttl_cache_expiration_handler.py

@@ -1,8 +1,10 @@
+import os
 import time
 
 from aws_lambda_powertools.utilities.idempotency import DynamoDBPersistenceLayer, IdempotencyConfig, idempotent
 
-persistence_layer = DynamoDBPersistenceLayer(table_name="IdempotencyTable")
+TABLE_NAME = os.getenv("IdempotencyTable", "")
+persistence_layer = DynamoDBPersistenceLayer(table_name=TABLE_NAME)
 config = IdempotencyConfig(expires_after_seconds=20)
 
 

tests/e2e/idempotency/handlers/ttl_cache_timeout_handler.py

@@ -1,8 +1,10 @@
+import os
 import time
 
 from aws_lambda_powertools.utilities.idempotency import DynamoDBPersistenceLayer, IdempotencyConfig, idempotent
 
-persistence_layer = DynamoDBPersistenceLayer(table_name="IdempotencyTable")
+TABLE_NAME = os.getenv("IdempotencyTable", "")
+persistence_layer = DynamoDBPersistenceLayer(table_name=TABLE_NAME)
 config = IdempotencyConfig(expires_after_seconds=1)
 
 

tests/e2e/idempotency/infrastructure.py

@@ -1,30 +1,31 @@
-from typing import Dict
-
 from aws_cdk import CfnOutput, RemovalPolicy
 from aws_cdk import aws_dynamodb as dynamodb
-from aws_cdk.aws_lambda import Function
+from aws_cdk.aws_dynamodb import Table
 
 from tests.e2e.utils.infrastructure import BaseInfrastructure
 
 
 class IdempotencyDynamoDBStack(BaseInfrastructure):
     def create_resources(self):
-        functions = self.create_lambda_functions()
-        self._create_dynamodb_table(functions=functions)
+        table = self._create_dynamodb_table()
+
+        env_vars = {"IdempotencyTable": table.table_name}
+        functions = self.create_lambda_functions(function_props={"environment": env_vars})
+
+        table.grant_read_write_data(functions["TtlCacheExpirationHandler"])
+        table.grant_read_write_data(functions["TtlCacheTimeoutHandler"])
+        table.grant_read_write_data(functions["ParallelExecutionHandler"])
 
-    def _create_dynamodb_table(self, functions: Dict[str, Function]):
+    def _create_dynamodb_table(self) -> Table:
         table = dynamodb.Table(
             self.stack,
             "Idempotency",
-            table_name="IdempotencyTable",
             removal_policy=RemovalPolicy.DESTROY,
             partition_key=dynamodb.Attribute(name="id", type=dynamodb.AttributeType.STRING),
             time_to_live_attribute="expiration",
             billing_mode=dynamodb.BillingMode.PAY_PER_REQUEST,
         )
 
-        table.grant_read_write_data(functions["TtlCacheExpirationHandler"])
-        table.grant_read_write_data(functions["TtlCacheTimeoutHandler"])
-        table.grant_read_write_data(functions["ParallelExecutionHandler"])
-
         CfnOutput(self.stack, "DynamoDBTable", value=table.table_name)
+
+        return table