fix enroll and reenroll relationship

konviva-events/tests/test_app.py

@@ -33,12 +33,13 @@ def test_start_progress(
     r = dynamodb_persistence_layer.collection.query(
         PartitionKey('d9da85f2-e09f-472d-9515-3d91d70f1e8a')
     )
-    assert any(item.get('sk') == 'STARTED' for item in r['items'])
+    docx = next((x for x in r['items'] if x['sk'] == '0'), {})
+    assert 'started_at' in docx
     assert any(
         item.get('sk') == 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS'
         for item in r['items']
     )
-    assert len(r['items']) == 3
+    assert len(r['items']) == 2
 
 
 def test_update_progress(
@@ -110,8 +111,9 @@ def test_set_as_completed(
         PartitionKey('6c7e3d9b-f5d1-4da4-9e55-0825bb6ff2b8')
     )
 
-    assert len(r['items']) == 8
-    assert any(item.get('sk') == 'COMPLETED' for item in r['items'])
+    docx = next((x for x in r['items'] if x['sk'] == '0'), {})
+    assert 'completed_at' in docx
+    assert len(r['items']) == 7
     assert any(item.get('sk') == 'LOCK' for item in r['items'])
     assert any(
         item.get('sk') == 'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS'