Fix e2e tests

aws-powertools · Oct 1, 2024 · 57921de · 57921de
1 parent 727bac1
commit 57921de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/e2e/idempotency_redis/handlers/response_hook.py b/tests/e2e/idempotency_redis/handlers/response_hook.py
@@ -1,16 +1,16 @@
 import os
 
 from aws_lambda_powertools.utilities.idempotency import (
-    DynamoDBPersistenceLayer,
     IdempotencyConfig,
     idempotent,
 )
 from aws_lambda_powertools.utilities.idempotency.persistence.datarecord import (
     DataRecord,
 )
+from aws_lambda_powertools.utilities.idempotency.persistence.redis import RedisCachePersistenceLayer
 
-TABLE_NAME = os.getenv("IdempotencyTable", "")
-persistence_layer = DynamoDBPersistenceLayer(table_name=TABLE_NAME)
+REDIS_HOST = os.getenv("RedisEndpoint", "")
+persistence_layer = RedisCachePersistenceLayer(host=REDIS_HOST, port=6379)
 
 
 def my_response_hook(response: dict, idempotent_data: DataRecord) -> dict: