This commit is contained in:
2025-07-30 12:31:53 -03:00
parent f99ca23a79
commit 80afcb9da3
12 changed files with 873 additions and 211 deletions

View File

@@ -1,5 +1,5 @@
from aws_lambda_powertools.utilities.typing import LambdaContext
from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
import events.billing.append_enrollment as app
@@ -14,9 +14,9 @@ def test_append_enrollment(
'new_image': {
'id': '945e8672-1d72-45c6-b76c-ac06aa8b52ab',
'sk': 'METADATA#SUBSCRIPTION_COVERED',
'billing_day': 5,
'billing_day': 6,
'created_at': '2025-07-23T18:09:22.785678-03:00',
'org_id': 'edp8njvgQuzNkLx2ySNfAD',
'org_id': 'cJtK9SsnJhKPyxESe7g3DG',
}
}
}
@@ -24,13 +24,13 @@ def test_append_enrollment(
assert app.lambda_handler(event, lambda_context) # type: ignore
r = dynamodb_persistence_layer.collection.query(
PartitionKey('BILLING#ORG#edp8njvgQuzNkLx2ySNfAD')
KeyPair('BILLING#ORG#cJtK9SsnJhKPyxESe7g3DG', 'START#2025-06-06#END#2025-07-05')
)
items = r['items']
assert items[0]['sk'] == 'START#2025-06-05#END#2025-07-04#SCHEDULE#AUTO_CLOSE'
assert items[0]['sk'] == 'START#2025-06-06#END#2025-07-05#SCHEDULE#AUTO_CLOSE'
assert (
items[1]['sk']
== 'START#2025-06-05#END#2025-07-04#ENROLLMENT#945e8672-1d72-45c6-b76c-ac06aa8b52ab'
== 'START#2025-06-06#END#2025-07-05#ENROLLMENT#945e8672-1d72-45c6-b76c-ac06aa8b52ab'
)
assert items[2]['sk'] == 'START#2025-06-05#END#2025-07-04'
assert items[2]['sk'] == 'START#2025-06-06#END#2025-07-05'