m

Author: Lucas McDonald

DynamoDbEncryption/runtimes/python/src/aws_dbesdk_dynamodb/encrypted/client.py

@@ -418,7 +418,7 @@ def update_item(self, **kwargs):
             output_item_to_dict_transform_method=self._client_to_resource_shape_converter.update_item_response,
             output_item_to_ddb_transform_method=self._resource_to_client_shape_converter.update_item_response,
         )
-    
+
     def get_paginator(self, operation_name: str) -> EncryptedPaginator | botocore.client.Paginator:
         """
         Get a paginator from the underlying client.

DynamoDbEncryption/runtimes/python/test/integ/encrypted/test_client.py

@@ -24,6 +24,7 @@
     simple_key_dict,
 )
 from ...requests import (
+    basic_batch_execute_statement_request,
     basic_batch_get_item_request_ddb,
     basic_batch_get_item_request_dict,
     basic_batch_write_item_delete_request_ddb,
@@ -32,6 +33,8 @@
     basic_batch_write_item_put_request_dict,
     basic_delete_item_request_ddb,
     basic_delete_item_request_dict,
+    basic_execute_statement_request,
+    basic_execute_transaction_request,
     basic_get_item_request_ddb,
     basic_get_item_request_dict,
     basic_put_item_request_ddb,
@@ -50,9 +53,6 @@
     basic_update_item_request_ddb_unsigned_attribute,
     basic_update_item_request_dict_signed_attribute,
     basic_update_item_request_dict_unsigned_attribute,
-    basic_execute_statement_request,
-    basic_execute_transaction_request,
-    basic_batch_execute_statement_request,
 )
 from . import sort_dynamodb_json_lists
 
@@ -381,10 +381,14 @@ def test_GIVEN_valid_transact_write_and_get_requests_WHEN_transact_write_and_get
     # Then: transact_write_item delete succeeds
     assert transact_write_delete_response["ResponseMetadata"]["HTTPStatusCode"] == 200
 
+
 @pytest.fixture
 def update_item_request_unsigned_attribute(expect_standard_dictionaries, test_item):
     if expect_standard_dictionaries:
-        return {**basic_update_item_request_dict_unsigned_attribute(test_item), "TableName": INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME}
+        return {
+            **basic_update_item_request_dict_unsigned_attribute(test_item),
+            "TableName": INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME,
+        }
     return basic_update_item_request_ddb_unsigned_attribute(test_item)
 
 
@@ -397,15 +401,21 @@ def test_WHEN_update_item_with_unsigned_attribute_THEN_passes(
     # Then: update_item succeeds
     assert update_response["ResponseMetadata"]["HTTPStatusCode"] == 200
 
+
 @pytest.fixture
 def update_item_request_signed_attribute(expect_standard_dictionaries, test_item):
     if expect_standard_dictionaries:
-        return {**basic_update_item_request_dict_signed_attribute(test_item), "TableName": INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME}
+        return {
+            **basic_update_item_request_dict_signed_attribute(test_item),
+            "TableName": INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME,
+        }
     return basic_update_item_request_ddb_signed_attribute(test_item)
 
 
 def test_WHEN_update_item_with_signed_attribute_THEN_raises_DynamoDbEncryptionTransformsException(
-    client, update_item_request_signed_attribute, encrypted,
+    client,
+    update_item_request_signed_attribute,
+    encrypted,
 ):
     """Test that update_item raises DynamoDbEncryptionTransformsException."""
     if not encrypted:
@@ -415,9 +425,7 @@ def test_WHEN_update_item_with_signed_attribute_THEN_raises_DynamoDbEncryptionTr
     # Then: DynamoDbEncryptionTransformsException is raised
     with pytest.raises(DynamoDbEncryptionTransformsException):
         # When: Calling update_item
-        client.update_item(
-            **update_item_request_signed_attribute
-        )
+        client.update_item(**update_item_request_signed_attribute)
 
 
 @pytest.fixture
@@ -426,7 +434,9 @@ def execute_statement_request():
 
 
 def test_WHEN_execute_statement_THEN_raises_DynamoDbEncryptionTransformsException(
-    client, execute_statement_request, encrypted,
+    client,
+    execute_statement_request,
+    encrypted,
 ):
     """Test that execute_statement raises DynamoDbEncryptionTransformsException."""
     if not encrypted:
@@ -436,9 +446,7 @@ def test_WHEN_execute_statement_THEN_raises_DynamoDbEncryptionTransformsExceptio
     # Then: DynamoDbEncryptionTransformsException is raised
     with pytest.raises(DynamoDbEncryptionTransformsException):
         # When: Calling update_item
-        client.execute_statement(
-            **execute_statement_request
-        )
+        client.execute_statement(**execute_statement_request)
 
 
 @pytest.fixture
@@ -447,7 +455,9 @@ def execute_transaction_request():
 
 
 def test_WHEN_execute_transaction_THEN_raises_DynamoDbEncryptionTransformsException(
-    client, execute_transaction_request, encrypted,
+    client,
+    execute_transaction_request,
+    encrypted,
 ):
     """Test that execute_transaction raises DynamoDbEncryptionTransformsException."""
     if not encrypted:
@@ -457,9 +467,7 @@ def test_WHEN_execute_transaction_THEN_raises_DynamoDbEncryptionTransformsExcept
     # Then: DynamoDbEncryptionTransformsException is raised
     with pytest.raises(DynamoDbEncryptionTransformsException):
         # When: Calling update_item
-        client.execute_transaction(
-            **execute_transaction_request
-        )
+        client.execute_transaction(**execute_transaction_request)
 
 
 @pytest.fixture
@@ -468,7 +476,9 @@ def batch_execute_statement_request():
 
 
 def test_WHEN_batch_execute_statement_THEN_raises_DynamoDbEncryptionTransformsException(
-    client, batch_execute_statement_request, encrypted,
+    client,
+    batch_execute_statement_request,
+    encrypted,
 ):
     """Test that batch_execute_statement raises DynamoDbEncryptionTransformsException."""
     if not encrypted:
@@ -478,9 +488,7 @@ def test_WHEN_batch_execute_statement_THEN_raises_DynamoDbEncryptionTransformsEx
     # Then: DynamoDbEncryptionTransformsException is raised
     with pytest.raises(DynamoDbEncryptionTransformsException):
         # When: Calling update_item
-        client.batch_execute_statement(
-            **batch_execute_statement_request
-        )
+        client.batch_execute_statement(**batch_execute_statement_request)
 
 
 def test_WHEN_get_paginator_THEN_correct_paginator_is_returned():

DynamoDbEncryption/runtimes/python/test/requests.py

@@ -78,6 +78,7 @@ def base_update_item_request_unsigned_attribute(item):
         "ExpressionAttributeNames": {"#attr3": ":attribute3"},
     }
 
+
 def basic_execute_statement_request():
     """Base structure for execute_statement requests."""
     return {"Statement": "SELECT * FROM " + INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME}
@@ -521,4 +522,3 @@ def basic_update_item_request_dict_unsigned_attribute(item):
     """Get an update_item request in dict format for any item."""
     base = base_update_item_request_unsigned_attribute(item)
     return base
-