add dynamodb collection

layercake/tests/test_dynamodb.py

@@ -5,6 +5,7 @@ import pytest
 from botocore.exceptions import ClientError
 
 from layercake.dynamodb import (
+    DynamoDBCollection,
     DynamoDBPersistenceLayer,
     Key,
     KeyPair,
@@ -37,35 +38,57 @@ def test_keypair():
     assert KeyPair('123', 'abc') == {'id': '123', 'sk': 'abc'}
 
 
-def test_transact_write_items(dynamodb_client):
-    user_layer = DynamoDBPersistenceLayer('pytest', dynamodb_client)
-    transact = TransactItems(user_layer.table_name)
+def test_transact_write_items(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    transact = TransactItems(dynamodb_persistence_layer.table_name)
+    transact.put(item=KeyPair('5OxmMjL-ujoR5IMGegQz', '0'))
+    transact.put(item=KeyPair('cpf', '07879819908'))
     transact.put(
-        item={
-            'id': '5OxmMjL-ujoR5IMGegQz',
-            'sk': '0',
-        }
-    )
-    transact.put(
-        item={
-            'id': 'cpf',
-            'sk': '07879819908',
-        }
-    )
-    transact.put(
-        item={
-            'id': 'email',
-            'sk': 'sergio@somosbeta.com.br',
-        },
+        item=KeyPair('email', 'sergio@somosbeta.com.br'),
         cond_expr='attribute_not_exists(sk)',
     )
     transact.put(
-        item={
-            'id': '5OxmMjL-ujoR5IMGegQz',
-            'sk': 'email:sergio@somosbeta.com.br',
-        },
+        item=KeyPair('5OxmMjL-ujoR5IMGegQz', 'emails#sergio@somosbeta.com.br'),
         cond_expr='attribute_not_exists(sk)',
     )
 
     with pytest.raises(ClientError):
-        user_layer.transact_write_items(transact)
+        dynamodb_persistence_layer.transact_write_items(transact)
+
+
+def test_collection(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    tenant_item = collect.get_item(
+        key=KeyPair(
+            pk='5OxmMjL-ujoR5IMGegQz',
+            sk=Key('tenant'),
+        ),
+        raise_if_missing=False,
+        default={},
+    )
+    assert tenant_item == {}
+
+    email_item = collect.get_item(
+        key=KeyPair(
+            pk='5OxmMjL-ujoR5IMGegQz',
+            sk=Key('sergio@somosbeta.com.br', prefix='emails'),
+        ),
+        default={},
+    )
+    assert email_item == {
+        'email_verified': True,
+        'mx_record_exists': True,
+        'sk': 'emails#sergio@somosbeta.com.br',
+        'email_primary': True,
+        'id': '5OxmMjL-ujoR5IMGegQz',
+        'create_date': '2019-03-25T00:00:00-03:00',
+        'update_date': '2023-11-09T12:13:04.308986-03:00',
+    }
+
+    with pytest.raises(DynamoDBCollection.MissingError):
+        collect.get_item(key=KeyPair('5OxmMjL-ujoR5IMGegQz', 'notfound'))