improve report

enrollments-events/tests/events/test_enroll.py

@@ -20,14 +20,13 @@ def test_enroll(
     assert app.lambda_handler(event, lambda_context)  # type: ignore
 
     # Parent knows the child
-    order_entity = dynamodb_persistence_layer.collection.get_item(
-        KeyPair(order_id, 'LINKED_ENTITIES#ORDER#CHILD')
+    r = dynamodb_persistence_layer.collection.query(
+        KeyPair(order_id, 'LINKED_ENTITIES#CHILD')
     )
-    assert order_entity
+    *_, enrollment_id = r['items'][0]['sk'].split('#')
 
     # Child knows the parent
-    enrollment_entity = dynamodb_persistence_layer.collection.get_item(
-        KeyPair(order_entity['enrollment_id'], 'LINKED_ENTITIES#ORDER#PARENT'),
+    enrollment = dynamodb_persistence_layer.collection.get_item(
+        KeyPair(enrollment_id, f'LINKED_ENTITIES#PARENT#ORDER#{order_id}'),
     )
-
-    assert enrollment_entity['order_id'] == order_id
+    assert enrollment