rename key

layercake/tests/test_dynamodb.py

@@ -166,9 +166,11 @@ def test_collection_get_item_path_spec(
     data = collect.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
-            sk=ComposeKey('sergio@somosbeta.com.br', prefix='emails'),
+            sk=SortKey(
+                ComposeKey('sergio@somosbeta.com.br', prefix='emails'),
+                path_spec='mx_record_exists',
+            ),
         ),
-        'mx_record_exists',
         default={},
     )
     assert data
@@ -287,6 +289,20 @@ def test_collection_get_items(
     }
 
 
+def test_collection_get_items_not_found(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    doc = collect.get_items(
+        TransactKey('not_found')
+        + SortKey('0')
+        + SortKey('metadata#not_found', path_spec='payment_method')
+    )
+
+    assert doc == {}
+
+
 def test_collection_get_items_unflatten(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
@@ -378,15 +394,20 @@ def test_collection_get_items_pair_path_spec(
 ):
     collect = DynamoDBCollection(dynamodb_persistence_layer)
     doc = collect.get_items(
-        KeyPair('cpf', SortKey('07879819908', path_spec='user_id', retain_key=True))
+        KeyPair(
+            'cpf',
+            SortKey('07879819908', path_spec='user_id'),
+            rename_key='user_id',
+        )
         + KeyPair(
             'email',
-            SortKey('osergiosiqueira@gmail.com', path_spec='user_id', retain_key=True),
+            SortKey('osergiosiqueira@gmail.com', path_spec='user_id'),
+            retain_key=True,
         ),
         flatten_top=False,
     )
 
     assert doc == {
-        'cpf': '5OxmMjL-ujoR5IMGegQz',
+        'user_id': '5OxmMjL-ujoR5IMGegQz',
         'email': '5OxmMjL-ujoR5IMGegQz',
     }