improve test

This commit is contained in:
2025-09-23 11:37:35 -03:00
parent 9c38e68a34
commit 79d46363ca

View File

@@ -34,6 +34,7 @@ def test_reenroll(
sk='LINKED_ENTITIES#ENROLLMENT', sk='LINKED_ENTITIES#ENROLLMENT',
) )
) )
# Parent knows the child
assert parent_entity['kind'] == 'CHILD' assert parent_entity['kind'] == 'CHILD'
child_entity = dynamodb_persistence_layer.collection.get_item( child_entity = dynamodb_persistence_layer.collection.get_item(
@@ -42,6 +43,6 @@ def test_reenroll(
sk='LINKED_ENTITIES#ENROLLMENT', sk='LINKED_ENTITIES#ENROLLMENT',
) )
) )
# Child knows the parent
assert child_entity['enrollment_id'] == parent_entity['id'] assert child_entity['enrollment_id'] == parent_entity['id']
assert child_entity['kind'] == 'PARENT' assert child_entity['kind'] == 'PARENT'