feat: add environment variable to allow disabling idempotency functionality to make testing easier

Author: Tom McCarthy

aws_lambda_powertools/shared/constants.py

@@ -21,3 +21,5 @@
 
 XRAY_SDK_MODULE: str = "aws_xray_sdk"
 XRAY_SDK_CORE_MODULE: str = "aws_xray_sdk.core"
+
+IDEMPOTENCY_DISABLED_ENV: str = "POWERTOOLS_IDEMPOTENCY_DISABLED"

aws_lambda_powertools/utilities/idempotency/idempotency.py

@@ -3,9 +3,11 @@
 """
 import functools
 import logging
+import os
 from typing import Any, Callable, Dict, Optional, cast
 
 from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
+from aws_lambda_powertools.shared.constants import IDEMPOTENCY_DISABLED_ENV
 from aws_lambda_powertools.shared.types import AnyCallableT
 from aws_lambda_powertools.utilities.idempotency.base import IdempotencyHandler
 from aws_lambda_powertools.utilities.idempotency.config import IdempotencyConfig
@@ -56,6 +58,9 @@ def idempotent(
         >>>     return {"StatusCode": 200}
     """
 
+    if os.getenv(IDEMPOTENCY_DISABLED_ENV):
+        return handler(event, context)
+
     config = config or IdempotencyConfig()
     args = event, context
     idempotency_handler = IdempotencyHandler(
@@ -122,6 +127,9 @@ def process_order(customer_id: str, order: dict, **kwargs):
 
     @functools.wraps(function)
     def decorate(*args, **kwargs):
+        if os.getenv(IDEMPOTENCY_DISABLED_ENV):
+            return function(*args, **kwargs)
+
         payload = kwargs.get(data_keyword_argument)
 
         if payload is None: