finish seat

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

@@ -5,8 +5,27 @@ from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, PartitionKey
 
 from ...conftest import HttpApiProxy, LambdaContext
 
+# Check the seeds, if necessary.
+org_id = '2a0f83b6-9d72-4fc0-952c-acbcfba39016'
+seat = {
+    'id': '389a282f-0a1e-4c9e-9502-d3131b1c2e57',
+    'user': {
+        'id': '15bacf02-1535-4bee-9022-19d106fd7518',
+        'name': 'Eddie Vedder',
+        'email': 'eddie@pearljam.band',
+        'cpf': '07879819908',
+    },
+    'course': {
+        'id': 'c27d1b4f-575c-4b6b-82a1-9b91ff369e0b',
+        'name': 'NR-18 PEMT Plataforma Móvel de Trabalho Aéreo',
+        'access_period': '360',
+        'unit_price': '149',
+    },
+    'seat': {'order_id': 'c556e2f2-f65b-4959-ad04-e789de107ac5'},
+}
 
-def test_enroll(
+
+def test_enroll_from_subscription(
     app,
     seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
@@ -18,7 +37,7 @@ def test_enroll(
             raw_path='/enrollments',
             method=HTTPMethod.POST,
             body={
-                'org_id': '2a8963fc-4694-4fe2-953a-316d1b10f1f5',
+                'org_id': org_id,
                 'subscription': {
                     'billing_day': 6,
                 },
@@ -74,7 +93,77 @@ def test_enroll(
     assert len(enrolled['items']) == 7
 
     scheduled = dynamodb_persistence_layer.collection.query(
-        PartitionKey('SCHEDULED#ORG#2a8963fc-4694-4fe2-953a-316d1b10f1f5')
+        PartitionKey(f'SCHEDULED#ORG#{org_id}')
     )
 
     assert len(scheduled['items']) == 1
+
+
+def test_enroll_for_from_seats(
+    app,
+    seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = app.lambda_handler(
+        http_api_proxy(
+            raw_path='/enrollments',
+            method=HTTPMethod.POST,
+            body={
+                'org_id': org_id,
+                'enrollments': [{**seat}],
+            },
+        ),
+        lambda_context,
+    )
+    body = json.loads(r['body'])
+    assert r['statusCode'] == HTTPStatus.OK
+    item = body['enrolled'][0]
+    assert item['status'] == 'success'
+
+    r = dynamodb_persistence_layer.collection.get_item(
+        KeyPair(
+            'c556e2f2-f65b-4959-ad04-e789de107ac5',
+            'ENROLLMENT#389a282f-0a1e-4c9e-9502-d3131b1c2e57',
+        )
+    )
+    assert r['status'] == 'EXECUTED'
+
+
+def test_schedule_for_from_seats(
+    app,
+    seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = app.lambda_handler(
+        http_api_proxy(
+            raw_path='/enrollments',
+            method=HTTPMethod.POST,
+            body={
+                'org_id': org_id,
+                'enrollments': [
+                    {
+                        **seat,
+                        'scheduled_for': '2028-01-01',
+                    }
+                ],
+            },
+        ),
+        lambda_context,
+    )
+    body = json.loads(r['body'])
+    assert r['statusCode'] == HTTPStatus.OK
+
+    item = body['scheduled'][0]
+    assert item['status'] == 'success'
+
+    r = dynamodb_persistence_layer.collection.get_item(
+        KeyPair(
+            'c556e2f2-f65b-4959-ad04-e789de107ac5',
+            'ENROLLMENT#389a282f-0a1e-4c9e-9502-d3131b1c2e57',
+        )
+    )
+    assert r['status'] == 'SCHEDULED'