From fcbfc3b97d39b4e47ef3c5d9cbe45a7324f7fffa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9rgio=20Rafael=20Siqueira?= Date: Tue, 15 Jul 2025 15:50:29 -0300 Subject: [PATCH] improve test --- layercake/tests/test_dynamodb.py | 47 ++++++++++---------------------- 1 file changed, 15 insertions(+), 32 deletions(-) diff --git a/layercake/tests/test_dynamodb.py b/layercake/tests/test_dynamodb.py index 76c6d4d..630c004 100644 --- a/layercake/tests/test_dynamodb.py +++ b/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'),