add fix

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',
+    }