add fix

http-api/tests/routes/test_enrollments.py

@@ -0,0 +1,52 @@
+from http import HTTPMethod, HTTPStatus
+
+from layercake.dynamodb import (
+    ComposeKey,
+    DynamoDBCollection,
+    DynamoDBPersistenceLayer,
+    KeyPair,
+    PartitionKey,
+)
+
+from ..conftest import HttpApiProxy, LambdaContext
+
+
+def test_cancel_enrollment(
+    mock_app,
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = mock_app.lambda_handler(
+        http_api_proxy(
+            raw_path='/enrollments/43ea4475-c369-4f90-b576-135b7df5106b/cancel',
+            method=HTTPMethod.PATCH,
+            headers={'X-Tenant': '*'},
+            body={
+                'id': '43ea4475-c369-4f90-b576-135b7df5106b',
+                'lock_hash': 'f8f7996aa99d50eb85266be5a9fca1db',
+                'course': {
+                    'id': '123',
+                    'name': 'NR-10',
+                    'time_in_days': 720,
+                },
+                'vacancy': {
+                    'sk': 'QV4sXY3DvSTUMGJ4QqsrwJ#43ea4475-c369-4f90-b576-135b7df5106b',
+                    'id': 'vacancies#cJtK9SsnJhKPyxESe7g3DG',
+                },
+            },
+        ),
+        lambda_context,
+    )
+
+    assert r['statusCode'] == HTTPStatus.OK
+
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    enrollment = collect.get_item(KeyPair('43ea4475-c369-4f90-b576-135b7df5106b', '0'))
+    assert enrollment['status'] == 'CANCELED'
+
+    vacancies = collect.query(
+        PartitionKey(ComposeKey('cJtK9SsnJhKPyxESe7g3DG', 'vacancies'))
+    )
+    assert len(vacancies['items']) == 1

http-api/tests/routes/test_orgs.py

@@ -0,0 +1,55 @@
+import json
+from http import HTTPMethod, HTTPStatus
+
+from layercake.dynamodb import (
+    DynamoDBCollection,
+    DynamoDBPersistenceLayer,
+    KeyPair,
+)
+
+from ..conftest import HttpApiProxy, LambdaContext
+
+
+def test_get_policies(
+    mock_app,
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = mock_app.lambda_handler(
+        http_api_proxy(
+            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/policies',
+            method=HTTPMethod.GET,
+            headers={'X-Tenant': '*'},
+        ),
+        lambda_context,
+    )
+    assert r['statusCode'] == HTTPStatus.OK
+    assert json.loads(r['body']) == {
+        'billing_policy': {'billing_day': 1, 'payment_method': 'PIX'},
+        'payment_policy': {'due_days': 90},
+    }
+
+
+def test_put_org(
+    mock_app,
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = mock_app.lambda_handler(
+        http_api_proxy(
+            raw_path='/orgs/cJtK9SsnJhKPyxESe7g3DG/policies',
+            method=HTTPMethod.PUT,
+            headers={'X-Tenant': '*'},
+            body={},
+        ),
+        lambda_context,
+    )
+    assert r['statusCode'] == HTTPStatus.OK
+
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    course = collect.get_item(KeyPair('cJtK9SsnJhKPyxESe7g3DG', '0'))
+    assert course['name'] == 'EDUSEG'

http-api/tests/routes/test_users.py

@@ -1,6 +1,8 @@
 import json
 from http import HTTPMethod, HTTPStatus
 
+from layercake.dynamodb import DynamoDBCollection, DynamoDBPersistenceLayer, KeyPair
+
 from ..conftest import HttpApiProxy, LambdaContext
 
 
@@ -105,3 +107,61 @@ def test_post_user(
     )
 
     assert r['statusCode'] == HTTPStatus.CREATED
+
+
+def test_post_email(
+    mock_app,
+    dynamodb_client,
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = mock_app.lambda_handler(
+        http_api_proxy(
+            raw_path='/users/5OxmMjL-ujoR5IMGegQz/emails',
+            method=HTTPMethod.POST,
+            body={
+                'email': 'sergio+pytest@somosbeta.com.br',
+            },
+        ),
+        lambda_context,
+    )
+
+    assert r['statusCode'] == HTTPStatus.CREATED
+
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    user = collect.get_item(KeyPair('5OxmMjL-ujoR5IMGegQz', '0'))
+    assert user['emails'] == {
+        'sergio@somosbeta.com.br',
+        'osergiosiqueira@gmail.com',
+        'sergio+pytest@somosbeta.com.br',
+    }
+
+
+def test_delete_email(
+    mock_app,
+    dynamodb_client,
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    r = mock_app.lambda_handler(
+        http_api_proxy(
+            raw_path='/users/5OxmMjL-ujoR5IMGegQz/emails',
+            method=HTTPMethod.DELETE,
+            body={
+                'email': 'osergiosiqueira@gmail.com',
+            },
+        ),
+        lambda_context,
+    )
+
+    assert r['statusCode'] == HTTPStatus.OK
+
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    user = collect.get_item(KeyPair('5OxmMjL-ujoR5IMGegQz', '0'))
+    assert user['emails'] == {
+        'sergio@somosbeta.com.br',
+    }