add get_items

layercake/tests/seeds.jsonl

@@ -2,3 +2,6 @@
 {"id": {"S": "5OxmMjL-ujoR5IMGegQz"}, "sk": {"S": "emails#sergio@somosbeta.com.br"}, "email_verified": {"BOOL": true}, "update_date": {"S": "2024-02-08T16:42:33.776409-03:00"}, "create_date": {"S": "2019-03-25T00:00:00-03:00"}, "email_primary": {"BOOL": true}, "mx_record_exists": {"BOOL": true}, "update_date": {"S": "2023-11-09T12:13:04.308986-03:00"}}
 {"id": {"S": "logs#5OxmMjL-ujoR5IMGegQz"}, "sk": {"S": "2024-02-08T16:42:33.776409-03:00"}, "action": {"S": "OPEN_EMAIL"}}
 {"id": {"S": "logs#5OxmMjL-ujoR5IMGegQz"}, "sk": {"S": "2019-03-25T00:00:00-03:00"}, "action": {"S": "CLICK_EMAIL"}}
+{"id": {"S": "cJtK9SsnJhKPyxESe7g3DG"}, "sk": {"S": "0"}, "name": {"S": "EDUSEG"}, "cnpj": {"S": "15608435000190"}, "email": {"S": "org+15608435000190@users.noreply.betaeducacao.com.br"}}
+{"id": {"S": "cJtK9SsnJhKPyxESe7g3DG"}, "sk": {"S": "payment_policy"}, "due_days": {"N": "90"}}
+{"id": {"S": "cJtK9SsnJhKPyxESe7g3DG"}, "sk": {"S": "billing_policy"}, "billing_day": {"N": "1"}, "payment_method": {"S": "PIX"}}

layercake/tests/test_dynamodb.py

@@ -1,4 +1,5 @@
 from datetime import datetime
+from decimal import Decimal
 from ipaddress import IPv4Address
 
 import pytest
@@ -12,7 +13,9 @@ from layercake.dynamodb import (
     KeyPair,
     PartitionKey,
     PrefixKey,
+    SortKey,
     TransactItems,
+    TransactKey,
     serialize,
 )
 
@@ -189,14 +192,14 @@ def test_collection_delete_item(
     )
 
 
-def test_collection_get_items(
+def test_collection_query(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
     collect = DynamoDBCollection(dynamodb_persistence_layer)
 
     # This data was added from seeds
-    logs = collect.get_items(
+    logs = collect.query(
         PartitionKey(
             ComposeKey('5OxmMjL-ujoR5IMGegQz', prefix='logs'),
         ),
@@ -219,7 +222,7 @@ def test_collection_get_items(
     }
 
     # This data was added from seeds
-    emails = collect.get_items(
+    emails = collect.query(
         KeyPair('5OxmMjL-ujoR5IMGegQz', PrefixKey('emails')),
     )
     assert emails == {
@@ -236,3 +239,27 @@ def test_collection_get_items(
         ],
         'last_key': None,
     }
+
+
+def test_collection_get_items(
+    dynamodb_seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+):
+    collect = DynamoDBCollection(dynamodb_persistence_layer)
+
+    doc = collect.get_items(
+        TransactKey('cJtK9SsnJhKPyxESe7g3DG')
+        + SortKey('0')
+        + SortKey('billing_policy')
+        + SortKey('payment_policy')
+    )
+
+    assert doc == {
+        'sk': '0',
+        'name': 'EDUSEG',
+        'id': 'cJtK9SsnJhKPyxESe7g3DG',
+        'cnpj': '15608435000190',
+        'email': 'org+15608435000190@users.noreply.betaeducacao.com.br',
+        'billing_policy': {'billing_day': Decimal('1'), 'payment_method': 'PIX'},
+        'payment_policy': {'due_days': Decimal('90')},
+    }