improve test

layercake/tests/test_dynamodb.py

@@ -6,7 +6,6 @@ import pytest
 from layercake.dateutils import ttl
 from layercake.dynamodb import (
     ComposeKey,
-    DynamoDBCollection,
     DynamoDBPersistenceLayer,
     KeyPair,
     PartitionKey,
@@ -115,8 +114,7 @@ def test_collection_get_item(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collection = dynamodb_persistence_layer.collection
-    data_notfound = collection.get_item(
+    data_notfound = dynamodb_persistence_layer.collection.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
             sk='tenant',
@@ -127,7 +125,7 @@ def test_collection_get_item(
     assert data_notfound == {}
 
     # This data was added from seeds
-    data = collection.get_item(
+    data = dynamodb_persistence_layer.collection.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
             sk=ComposeKey('sergio@somosbeta.com.br', prefix='emails'),
@@ -147,7 +145,7 @@ def test_collection_get_item(
     class NotFoundError(Exception): ...
 
     with pytest.raises(NotFoundError):
-        collection.get_item(
+        dynamodb_persistence_layer.collection.get_item(
             KeyPair('5OxmMjL-ujoR5IMGegQz', 'notfound'),
             exc_cls=NotFoundError,
         )
@@ -157,10 +155,8 @@ def test_collection_get_item_path_spec(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collection = dynamodb_persistence_layer.collection
-
     # This data was added from seeds
-    data = collection.get_item(
+    data = dynamodb_persistence_layer.collection.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
             sk=SortKey(
@@ -176,9 +172,7 @@ def test_collection_get_item_path_spec(
 def test_collection_put_item(
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-
-    assert collect.put_item(
+    assert dynamodb_persistence_layer.collection.put_item(
         KeyPair(
             '5OxmMjL-ujoR5IMGegQz',
             ComposeKey('6d1044d5-18c5-437c-9219-fc2ace7e5ebc', prefix='orgs'),
@@ -187,7 +181,7 @@ def test_collection_put_item(
         ttl=ttl(days=3),
     )
 
-    data = collect.get_item(
+    data = dynamodb_persistence_layer.collection.get_item(
         KeyPair(
             pk='5OxmMjL-ujoR5IMGegQz',
             sk=ComposeKey('6d1044d5-18c5-437c-9219-fc2ace7e5ebc', prefix='orgs'),
@@ -203,10 +197,8 @@ def test_collection_delete_item(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-
     # This data was added from seeds
-    assert collect.delete_item(
+    assert dynamodb_persistence_layer.collection.delete_item(
         KeyPair(
             '5OxmMjL-ujoR5IMGegQz',
             ComposeKey('sergio@somsbeta.com.br', prefix='emails'),
@@ -218,10 +210,8 @@ def test_collection_query(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-
     # This data was added from seeds
-    logs = collect.query(
+    logs = dynamodb_persistence_layer.collection.query(
         PartitionKey(
             ComposeKey('5OxmMjL-ujoR5IMGegQz', prefix='logs'),
         ),
@@ -244,7 +234,7 @@ def test_collection_query(
     }
 
     # This data was added from seeds
-    emails = collect.query(
+    emails = dynamodb_persistence_layer.collection.query(
         KeyPair('5OxmMjL-ujoR5IMGegQz', PrefixKey('emails')),
     )
     assert emails == {
@@ -267,8 +257,7 @@ def test_collection_get_items(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         TransactKey(
             'cJtK9SsnJhKPyxESe7g3DG', table_name=dynamodb_persistence_layer.table_name
         )
@@ -293,8 +282,7 @@ def test_collection_get_items_not_found(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         TransactKey('not_found')
         + SortKey('0')
         + SortKey('metadata#not_found', path_spec='payment_method')
@@ -307,9 +295,7 @@ def test_collection_get_items_unflatten(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         TransactKey('cJtK9SsnJhKPyxESe7g3DG')
         + SortKey('metadata#billing_policy')
         + SortKey('metadata#payment_policy', remove_prefix='metadata#'),
@@ -329,8 +315,7 @@ def test_collection_get_items_pair(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
         + KeyPair('cpf', '07879819908')
         + KeyPair('email', 'osergiosiqueira@gmail.com')
@@ -359,8 +344,7 @@ def test_collection_get_items_pair_unflatten(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         KeyPair('5OxmMjL-ujoR5IMGegQz', '0')
         + KeyPair('cpf', '07879819908')
         + KeyPair('email', 'osergiosiqueira@gmail.com'),
@@ -392,8 +376,7 @@ def test_collection_get_items_pair_path_spec(
     dynamodb_seeds,
     dynamodb_persistence_layer: DynamoDBPersistenceLayer,
 ):
-    collect = DynamoDBCollection(dynamodb_persistence_layer)
-    doc = collect.get_items(
+    doc = dynamodb_persistence_layer.collection.get_items(
         KeyPair(
             'cpf',
             SortKey('07879819908', path_spec='user_id'),