add test_mode

orders-events/app/events/run_self_destruction.py

@@ -0,0 +1,39 @@
+from aws_lambda_powertools import Logger
+from aws_lambda_powertools.utilities.data_classes import (
+    EventBridgeEvent,
+    event_source,
+)
+from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dynamodb import (
+    DynamoDBPersistenceLayer,
+    PartitionKey,
+)
+
+from boto3clients import dynamodb_client
+from config import ORDER_TABLE
+
+logger = Logger(__name__)
+dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
+
+
+@event_source(data_class=EventBridgeEvent)
+@logger.inject_lambda_context
+def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
+    keys = event.detail['keys']
+    order_id = keys['id']
+
+    r = dyn.collection.query(PartitionKey(order_id), limit=150)
+    logger.info('Records found', total_items=len(r['items']), records=r['items'])
+
+    with dyn.batch_writer() as batch:
+        for pair in r['items']:
+            batch.delete_item(
+                Key={
+                    'id': {'S': pair['id']},
+                    'sk': {'S': pair['sk']},
+                }
+            )
+
+        logger.info('Deleted all records')
+
+    return True