close bolling

order-events/tests/events/billing/test_append_enrollment.py

@@ -1,10 +1,10 @@
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
 
-import events.reporting.add_item as app
+import events.billing.append_enrollment as app
 
 
-def test_add_item(
+def test_append_enrollment(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
     lambda_context: LambdaContext,
@@ -13,7 +13,7 @@ def test_add_item(
         'detail': {
             'new_image': {
                 'id': '945e8672-1d72-45c6-b76c-ac06aa8b52ab',
-                'sk': 'METADATA#BILLING_TERMS',
+                'sk': 'METADATA#SUBSCRIPTION_COVERED',
                 'billing_day': 5,
                 'created_at': '2025-07-23T18:09:22.785678-03:00',
                 'org_id': 'edp8njvgQuzNkLx2ySNfAD',
@@ -26,5 +26,11 @@ def test_add_item(
     r = dynamodb_persistence_layer.collection.query(
         PartitionKey('BILLING#ORG#edp8njvgQuzNkLx2ySNfAD')
     )
+    items = r['items']
 
-    print(r)
+    assert items[0]['sk'] == 'START#2025-06-05#END#2025-07-04#SCHEDULE#AUTO_CLOSE'
+    assert (
+        items[1]['sk']
+        == 'START#2025-06-05#END#2025-07-04#ENROLLMENT#945e8672-1d72-45c6-b76c-ac06aa8b52ab'
+    )
+    assert items[2]['sk'] == 'START#2025-06-05#END#2025-07-04'

order-events/tests/events/billing/test_close_window.py

@@ -0,0 +1,29 @@
+from aws_lambda_powertools.utilities.typing import LambdaContext
+from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
+
+import events.billing.close_window as app
+
+
+def test_append_enrollment(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    lambda_context: LambdaContext,
+):
+    event = {
+        'detail': {
+            'new_image': {
+                'id': 'BILLING#ORG#BES6dmWgTMXRYmfDyYYXUF',
+                'sk': 'START#2025-07-01#END#2025-07-31#SCHEDULE#AUTO_CLOSE',
+                'created_at': '2025-07-24T15:20:52.464244-03:00',
+                'ttl': 1754017200,
+            }
+        }
+    }
+
+    assert app.lambda_handler(event, lambda_context)  # type: ignore
+
+    # r = dynamodb_persistence_layer.collection.query(
+    #     PartitionKey('BILLING#ORG#edp8njvgQuzNkLx2ySNfAD')
+    # )
+
+    # print(r)