improve report

enrollments-events/tests/events/test_reenroll_if_failed.py

@@ -8,11 +8,11 @@ def test_reenroll(
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
     lambda_context: LambdaContext,
 ):
-    parent_id = '294e9864-8284-4287-b153-927b15d90900'
+    enrollment_id = '294e9864-8284-4287-b153-927b15d90900'
     event = {
         'detail': {
             'new_image': {
-                'id': parent_id,
+                'id': enrollment_id,
                 'sk': '0',
                 'user': {
                     'id': '2beb8642-aab4-4088-86d4-2966fac7c570',
@@ -30,19 +30,19 @@ def test_reenroll(
     assert app.lambda_handler(event, lambda_context)  # type: ignore
 
     # Parent knows the child
-    current_entity = dynamodb_persistence_layer.collection.get_item(
+    r = dynamodb_persistence_layer.collection.query(
         KeyPair(
-            pk=parent_id,
-            sk='LINKED_ENTITIES#ENROLLMENT#CHILD',
+            pk=enrollment_id,
+            sk='LINKED_ENTITIES#CHILD',
         )
     )
-    assert current_entity
+    *_, child_id = r['items'][0]['sk'].split('#')
 
     # Child knows the parent
-    new_entity = dynamodb_persistence_layer.collection.get_item(
+    child = dynamodb_persistence_layer.collection.get_item(
         KeyPair(
-            pk=current_entity['enrollment_id'],
-            sk='LINKED_ENTITIES#ENROLLMENT#PARENT',
+            pk=child_id,
+            sk=f'LINKED_ENTITIES#PARENT#ENROLLMENT#{enrollment_id}',
         )
     )
-    assert new_entity['enrollment_id'] == parent_id
+    assert child