add test mode to enrollments

orders-events/app/events/billing/send_email_on_closing.py

@@ -43,7 +43,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     new_image = event.detail['new_image']
     # Key pattern `BILLING#ORG#{org_id}`
     *_, org_id = new_image['id'].split('#')
-    # Key pattern `START#{start_period}#END#{v}
+    # Key pattern `START#{start_period}#END#{end_period}
     _, start_period, _, end_period, *_ = new_image['sk'].split('#')
 
     emailmsg = Message(

orders-events/app/events/payments/create_invoice.py

@@ -102,16 +102,6 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                 transact.delete(
                     key=KeyPair(order_id, 'CREDIT_CARD#PAYMENT_INTENT'),
                 )
-
-            if test_mode:
-                transact.put(
-                    item={
-                        'id': order_id,
-                        'sk': 'SCHEDULED#SELF_DESTRUCTION',
-                        'ttl': ttl(start_dt=now_, days=7),
-                        'created_at': now_,
-                    }
-                )
     except Exception:
         pass
 

orders-events/app/events/stopgap/__init__.py

@@ -1,4 +0,0 @@
-"""
-Stopgap events. Everything here is a quick fix and should be replaced with
-proper solutions.
-"""

orders-events/app/events/stopgap/remove_slots.py

@@ -1,71 +0,0 @@
-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,
-    KeyPair,
-)
-
-from boto3clients import dynamodb_client
-from config import ENROLLMENT_TABLE, ORDER_TABLE, USER_TABLE
-
-logger = Logger(__name__)
-user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
-order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
-enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
-
-
-@event_source(data_class=EventBridgeEvent)
-@logger.inject_lambda_context
-def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
-    """Remove slots if the tenant has a `metadata#billing_policy` and
-    the total is greater than zero."""
-    new_image = event.detail['new_image']
-    data = order_layer.get_item(KeyPair(new_image['id'], '0'))
-    org_id = data.get('tenant_id')
-
-    if not org_id:
-        return False
-
-    policy = user_layer.collection.get_item(
-        KeyPair(pk=org_id, sk='metadata#billing_policy'),
-        raise_on_error=False,
-        default=False,
-    )
-
-    # Skip if billing policy is missing or order is less than or equal to zero
-    if not policy or data['total'] <= 0:
-        logger.info('Missing billing policy')
-        return False
-
-    logger.info(f'Billing policy from Org ID "{org_id}" found', policy=policy)
-
-    result = enrollment_layer.collection.query(
-        KeyPair(
-            f'vacancies#{org_id}',
-            new_image['id'],
-        ),
-        limit=150,
-    )
-
-    logger.info(
-        'Slots found',
-        total_items=len(result['items']),
-        slots=result['items'],
-    )
-
-    with enrollment_layer.batch_writer() as batch:
-        for pair in result['items']:
-            batch.delete_item(
-                Key={
-                    'id': {'S': pair['id']},
-                    'sk': {'S': pair['sk']},
-                }
-            )
-
-        logger.info('Deleted all slots')
-
-    return True

orders-events/app/events/stopgap/set_as_paid.py

@@ -1,47 +0,0 @@
-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.dateutils import now
-from layercake.dynamodb import (
-    DynamoDBPersistenceLayer,
-    KeyPair,
-)
-
-from boto3clients import dynamodb_client
-from config import ORDER_TABLE
-
-logger = Logger(__name__)
-order_layer = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
-
-
-@event_source(data_class=EventBridgeEvent)
-@logger.inject_lambda_context
-def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
-    """Set to `PAID` if the status is `PENDING` and the total is zero."""
-    new_image = event.detail['new_image']
-    now_ = now()
-
-    with order_layer.transact_writer() as transact:
-        transact.update(
-            key=KeyPair(new_image['id'], '0'),
-            update_expr='SET #status = :status, updated_at = :updated_at',
-            expr_attr_names={
-                '#status': 'status',
-            },
-            expr_attr_values={
-                ':status': 'PAID',
-                ':updated_at': now_,
-            },
-        )
-        transact.put(
-            item={
-                'id': new_image['id'],
-                'sk': 'paid_at',
-                'paid_at': now_,
-            }
-        )
-
-    return True