improve get items

layercake/tests/test_dynamodb.py

@@ -119,7 +119,7 @@ def test_collection_get_item(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    collect = dynamodb_persistence_layer.collect
     data_notfound = collect.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
@@ -157,6 +157,24 @@ def test_collection_get_item(
         )
 
 
+def test_collection_get_item_path_spec(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = dynamodb_persistence_layer.collect
+
+    # This data was added from seeds
+    data = collect.get_item(
+        KeyPair(
+            pk='5OxmMjL-ujoR5IMGegQz',
+            sk=ComposeKey('sergio@somosbeta.com.br', prefix='emails'),
+        ),
+        'mx_record_exists',
+        default={},
+    )
+    assert data
+
+
 def test_collection_put_item(
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
@@ -181,7 +199,6 @@ def test_collection_put_item(
     assert data['sk'] == 'orgs#6d1044d5-18c5-437c-9219-fc2ace7e5ebc'
     assert 'name' in data
     assert 'ttl' in data
-    assert 'ttl_date' in data
 
 
 def test_collection_delete_item(
@@ -291,3 +308,66 @@ def test_collection_get_items_unflatten(
         },
         'payment_policy': {'due_days': Decimal('90')},
     }
+
+
+def test_collection_get_items_pair(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    doc = collect.get_items(
+        KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
+        + KeyPair('cpf', '07879819908')
+        + KeyPair('email', 'osergiosiqueira@gmail.com')
+    )
+
+    assert doc == {
+        'tenant:org_id': [
+            'cJtK9SsnJhKPyxESe7g3DG',
+            'edp8njvgQuzNkLx2ySNfAD',
+            '8TVSi5oACLxTiT8ycKPmaQ',
+        ],
+        'email_verified': True,
+        'last_login': '2024-02-08T20:53:45.818126-03:00',
+        'sk': '0',
+        'cpf': {'user_id': '5OxmMjL-ujoR5IMGegQz'},
+        'name': 'Sérgio Rafael de Siqueira',
+        'id': '5OxmMjL-ujoR5IMGegQz',
+        'create_date': '2019-03-25T00:00:00-03:00',
+        'cognito:sub': '58efed8d-d276-41a8-8502-4ab8b5a6415e',
+        'update_date': '2024-02-08T16:42:33.776409-03:00',
+        'email': {'user_id': '5OxmMjL-ujoR5IMGegQz'},
+    }
+
+
+def test_collection_get_items_pair_unflatten(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+    doc = collect.get_items(
+        KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
+        + KeyPair('cpf', '07879819908')
+        + KeyPair('email', 'osergiosiqueira@gmail.com'),
+        flatten_top=False,
+    )
+
+    assert doc == {
+        '5OxmMjL-ujoR5IMGegQz': {
+            'tenant:org_id': [
+                'cJtK9SsnJhKPyxESe7g3DG',
+                'edp8njvgQuzNkLx2ySNfAD',
+                '8TVSi5oACLxTiT8ycKPmaQ',
+            ],
+            'email_verified': True,
+            'last_login': '2024-02-08T20:53:45.818126-03:00',
+            'cpf': '07879819908',
+            'name': 'Sérgio Rafael de Siqueira',
+            'create_date': '2019-03-25T00:00:00-03:00',
+            'cognito:sub': '58efed8d-d276-41a8-8502-4ab8b5a6415e',
+            'update_date': '2024-02-08T16:42:33.776409-03:00',
+            'email': 'sergio@somosbeta.com.br',
+        },
+        'cpf': {'user_id': '5OxmMjL-ujoR5IMGegQz'},
+        'email': {'user_id': '5OxmMjL-ujoR5IMGegQz'},
+    }