add enroll to subscribed

api.saladeaula.digital/tests/routes/enrollments/test_enroll.py

@@ -1,5 +1,3 @@
-import json
-import pprint
 from http import HTTPMethod
 
 from layercake.dynamodb import DynamoDBPersistenceLayer, PartitionKey
@@ -26,14 +24,14 @@ def test_enroll(
                         'user': {
                             'id': '15bacf02-1535-4bee-9022-19d106fd7518',
                             'name': 'Eddie Vedder',
-                            'email': 'sergio@somosbeta.com.br',
+                            'email': 'eddie@pearljam.band',
                             'cpf': '07879819908',
                         },
                         'course': {
                             'id': 'c27d1b4f-575c-4b6b-82a1-9b91ff369e0b',
-                            'name': 'NR-10',
+                            'name': 'NR-18 PEMT Plataforma Móvel de Trabalho Aéreo',
                             'access_period': '360',
-                            'unit_price': '100.30',
+                            'unit_price': '149',
                         },
                         'scheduled_for': '2028-01-01',
                     },
@@ -61,11 +59,14 @@ def test_enroll(
         lambda_context,
     )
 
-    body = json.loads(r['body'])
-    pprint.pp(body)
-
     enrolled = dynamodb_persistence_layer.collection.query(
         PartitionKey('d0349bbe-cef3-44f7-b20e-3cb4476ab4c5')
     )
 
-    pprint.pp(enrolled)
+    assert len(enrolled['items']) == 7
+
+    scheduled = dynamodb_persistence_layer.collection.query(
+        PartitionKey('SCHEDULED#ORG#2a8963fc-4694-4fe2-953a-316d1b10f1f5')
+    )
+
+    assert len(scheduled['items']) == 1