67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
from datetime import timedelta
|
|
|
|
import app.events.cert_reporting.append_cert as app
|
|
from aws_lambda_powertools.utilities.typing import LambdaContext
|
|
from layercake.dateutils import now
|
|
from layercake.dynamodb import (
|
|
DynamoDBPersistenceLayer,
|
|
SortKey,
|
|
TransactKey,
|
|
)
|
|
|
|
|
|
def test_append_cert(
|
|
seeds,
|
|
dynamodb_persistence_layer: DynamoDBPersistenceLayer,
|
|
lambda_context: LambdaContext,
|
|
):
|
|
expires_at = now() + timedelta(days=360)
|
|
event = {
|
|
'detail': {
|
|
'new_image': {
|
|
'id': 'e45019d8-be7a-4a82-9b37-12a01f0127bb',
|
|
'sk': '0',
|
|
'course': {
|
|
'id': '431',
|
|
'name': 'How to Sing Better',
|
|
},
|
|
'cert': {
|
|
# 'expires_at': '2026-02-10T20:14:42.880991',
|
|
'expires_at': expires_at.isoformat(),
|
|
},
|
|
'user': {
|
|
'id': '1234',
|
|
'name': 'Tobias Summit',
|
|
},
|
|
'org_id': '1e2eaf0e-e319-49eb-ab33-1ddec156dc94',
|
|
'created_at': '2025-01-01T00:00:00-03:06',
|
|
'completed_at': '2025-01-10T00:00:00-03:06',
|
|
}
|
|
}
|
|
}
|
|
assert app.lambda_handler(event, lambda_context) # type: ignore
|
|
|
|
# The reporting month is the month before the certificate expires
|
|
month_start = (expires_at.replace(day=1) - timedelta(days=1)).replace(day=1)
|
|
report_sk = 'MONTH#{}#SCHEDULE#SEND_REPORT_EMAIL'.format(
|
|
month_start.strftime('%Y-%m')
|
|
)
|
|
|
|
r = dynamodb_persistence_layer.collection.get_items(
|
|
TransactKey('CERT#REPORTING#ORG#1e2eaf0e-e319-49eb-ab33-1ddec156dc94')
|
|
+ SortKey(
|
|
sk=report_sk,
|
|
rename_key='report_email',
|
|
)
|
|
+ SortKey(
|
|
sk='MONTH#{}#ENROLLMENT#e45019d8-be7a-4a82-9b37-12a01f0127bb'.format(
|
|
expires_at.strftime('%Y-%m')
|
|
),
|
|
rename_key='enrollment',
|
|
),
|
|
flatten_top=False,
|
|
)
|
|
|
|
assert 'course' in r['enrollment']
|
|
assert 'ttl' in r['report_email']
|