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