add test mode to checkout

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

@@ -8,6 +8,7 @@ from layercake.dateutils import now
 from layercake.dynamodb import (
     DynamoDBPersistenceLayer,
     KeyPair,
+    SortKey,
 )
 from layercake.extra_types import CreditCard
 
@@ -17,13 +18,6 @@ from iugu import Credentials, Iugu
 
 logger = Logger(__name__)
 dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
-iugu = Iugu(
-    Credentials(
-        IUGU_ACCOUNT_ID,
-        IUGU_API_TOKEN,
-        test_mode=IUGU_TEST_MODE,
-    )
-)
 
 
 @event_source(data_class=EventBridgeEvent)
@@ -36,6 +30,26 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
     credit_card = CreditCard(**new_image['credit_card'])
     now_ = now()
 
+    api_token = dyn.collection.get_item(
+        KeyPair(
+            pk=order_id,
+            sk=SortKey(
+                'METADATA#TEST_MODE',
+                path_spec='iugu_api_token',
+            ),
+        ),
+        default=None,
+        raise_on_error=False,
+    )
+    test_mode = (api_token is not None) or IUGU_TEST_MODE
+    credentials = Credentials(
+        IUGU_ACCOUNT_ID,
+        # Note: `api_token` can be set from the database
+        api_token or IUGU_API_TOKEN,
+        test_mode=test_mode,
+    )
+
+    iugu = Iugu(credentials)
     token = iugu.payment_token(credit_card)
     charge = iugu.charge(
         invoice_id=invoice_id,

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

@@ -24,13 +24,6 @@ from iugu import Credentials, Iugu, Order
 
 logger = Logger(__name__)
 dyn = DynamoDBPersistenceLayer(ORDER_TABLE, dynamodb_client)
-iugu = Iugu(
-    Credentials(
-        IUGU_ACCOUNT_ID,
-        IUGU_API_TOKEN,
-        test_mode=IUGU_TEST_MODE,
-    )
-)
 
 
 @event_source(data_class=EventBridgeEvent)
@@ -43,14 +36,29 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
         TransactKey(order_id)
         + SortKey('ADDRESS', rename_key='address')
         + SortKey('ITEMS', path_spec='items', rename_key='items')
-        + SortKey('CREDIT_CARD#PAYMENT_INTENT', rename_key='credit_card'),
+        + SortKey('CREDIT_CARD#PAYMENT_INTENT', rename_key='credit_card')
+        + SortKey(
+            'METADATA#TEST_MODE',
+            rename_key='iugu_api_token',
+            path_spec='iugu_api_token',
+        ),
         flatten_top=False,
     )
 
+    api_token = r.get('iugu_api_token')
+    test_mode = (api_token is not None) or IUGU_TEST_MODE
+    credentials = Credentials(
+        IUGU_ACCOUNT_ID,
+        # Note: `api_token` can be set from the database
+        api_token or IUGU_API_TOKEN,
+        test_mode=test_mode,
+    )
+
     payment_method = new_image['payment_method']
     is_pix = payment_method == 'PIX'
     is_bank_slip = payment_method == 'BANK_SLIP'
 
+    iugu = Iugu(credentials)
     invoice = iugu.create_invoice(
         order=Order(
             address=r.get('address', {}),
@@ -95,6 +103,15 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
                     cond_expr='attribute_not_exists(sk)',
                 )
 
+            if test_mode:
+                transact.put(
+                    item={
+                        'id': order_id,
+                        'sk': 'SCHEDULE#SELF_DESTRUCTION',
+                        'ttl': ttl(start_dt=now_, days=14),
+                        'created_at': now_,
+                    }
+                )
     except Exception:
         pass