m

Author: Lucas McDonald

DynamoDbEncryption/runtimes/python/test/responses.py

@@ -153,20 +153,25 @@ def basic_transact_write_items_response(items):
         },
     }
 
+
 # No exhaustive response for transact_write_items;
 # The basic_transact_write_items_response is sufficient
 
+
 def basic_transact_get_items_response(items):
     """Get a transact_get_items response in resource (ddb) format for any items."""
     return {"Responses": [{"Item": item} for item in items]}
 
+
 # No exhaustive response for transact_get_items;
 # The basic_transact_get_items_response is sufficient
 
+
 def basic_update_item_response(item):
     """Get an update_item response in resource (ddb) format for any item."""
     return {"Attributes": item}
 
+
 def exhaustive_update_item_response(item):
     """
     Get an update_item response in resource (ddb) format for any item.
@@ -182,10 +187,12 @@ def exhaustive_update_item_response(item):
     }
     return {**base, **additional_keys}
 
+
 def basic_delete_item_response(item):
     """Get a delete_item response in resource (ddb) format for any item."""
     return {"Attributes": item}
 
+
 def exhaustive_delete_item_response(item):
     """
     Get a delete_item response in resource (ddb) format for any item.
@@ -206,6 +213,7 @@ def basic_execute_statement_response(items):
     """Get an execute_statement response in resource (ddb) format for any items."""
     return {"Items": items}
 
+
 def exhaustive_execute_statement_response(items):
     """
     Get an execute_statement response in resource (ddb) format for any items.
@@ -222,17 +230,21 @@ def exhaustive_execute_statement_response(items):
     }
     return {**base, **additional_keys}
 
+
 def basic_execute_transaction_response(items):
     """Get an execute_transaction response in resource (ddb) format for any items."""
     return {"Responses": [{"Item": item} for item in items]}
 
+
 # No exhaustive response for execute_transaction;
 # The basic_execute_transaction_response is sufficient
 
+
 def basic_batch_execute_statement_response(items):
     """Get a batch_execute_statement response in resource (ddb) format for any items."""
     return {"Responses": [{"Item": item} for item in items]}
 
+
 def exhaustive_batch_execute_statement_response(items):
     """
     Get a batch_execute_statement response in resource (ddb) format for any items.
@@ -247,4 +259,4 @@ def exhaustive_batch_execute_statement_response(items):
             "sort_key": items[0]["sort_key"],
         }
     }
-    return base
+    return base

DynamoDbEncryption/runtimes/python/test/unit/internal/test_client_to_resource.py

@@ -55,29 +55,29 @@
     exhaustive_scan_request_dict,
 )
 from ...responses import (
+    basic_batch_execute_statement_response,
     basic_batch_get_item_response,
     basic_batch_write_item_put_response,
     basic_delete_item_response,
-    exhaustive_delete_item_response,
+    basic_execute_statement_response,
+    basic_execute_transaction_response,
     basic_get_item_response,
     basic_put_item_response,
     basic_query_response,
     basic_scan_response,
     basic_transact_get_items_response,
     basic_transact_write_items_response,
     basic_update_item_response,
-    exhaustive_update_item_response,
+    exhaustive_batch_execute_statement_response,
     exhaustive_batch_get_item_response,
     exhaustive_batch_write_item_put_response,
+    exhaustive_delete_item_response,
+    exhaustive_execute_statement_response,
     exhaustive_get_item_response,
     exhaustive_put_item_response,
     exhaustive_query_response,
     exhaustive_scan_response,
-    exhaustive_execute_statement_response,
-    basic_execute_statement_response,
-    basic_execute_transaction_response,
-    basic_batch_execute_statement_response,
-    exhaustive_batch_execute_statement_response,
+    exhaustive_update_item_response,
 )
 
 client_to_resource_converter = ClientShapeToResourceShapeConverter()
@@ -629,7 +629,9 @@ def test_execute_statement_response(use_exhaustive_request):
     return basic_execute_statement_response
 
 
-def test_GIVEN_test_execute_statement_response_WHEN_client_to_resource_THEN_returns_dict_value(test_execute_statement_response, test_ddb_item, test_dict_item):
+def test_GIVEN_test_execute_statement_response_WHEN_client_to_resource_THEN_returns_dict_value(
+    test_execute_statement_response, test_ddb_item, test_dict_item
+):
     # Given: Execute statement response
     ddb_response = test_execute_statement_response([test_ddb_item])
     # When: Converting to resource format
@@ -653,12 +655,15 @@ def test_GIVEN_test_execute_transaction_request_WHEN_client_to_resource_THEN_ret
     # Then: Returns dict value (here, request is not modified)
     assert dict_item == test_execute_transaction_request(test_dict_item)
 
+
 @pytest.fixture
 def test_execute_transaction_response():
     return basic_execute_transaction_response
 
 
-def test_GIVEN_test_execute_transaction_response_WHEN_client_to_resource_THEN_returns_dict_value(test_execute_transaction_response, test_ddb_item, test_dict_item):
+def test_GIVEN_test_execute_transaction_response_WHEN_client_to_resource_THEN_returns_dict_value(
+    test_execute_transaction_response, test_ddb_item, test_dict_item
+):
     # Given: Execute transaction response
     ddb_response = test_execute_transaction_response([test_ddb_item])
     # When: Converting to resource format
@@ -690,7 +695,9 @@ def test_batch_execute_statement_response(use_exhaustive_request):
     return basic_batch_execute_statement_response
 
 
-def test_GIVEN_test_batch_execute_statement_response_WHEN_client_to_resource_THEN_returns_dict_value(test_batch_execute_statement_response, test_ddb_item, test_dict_item):
+def test_GIVEN_test_batch_execute_statement_response_WHEN_client_to_resource_THEN_returns_dict_value(
+    test_batch_execute_statement_response, test_ddb_item, test_dict_item
+):
     # Given: Batch execute statement response
     ddb_response = test_batch_execute_statement_response([test_ddb_item])
     # When: Converting to resource format

DynamoDbEncryption/runtimes/python/test/unit/internal/test_resource_to_client.py

@@ -56,29 +56,29 @@
     exhaustive_scan_request_dict,
 )
 from ...responses import (
+    basic_batch_execute_statement_response,
     basic_batch_get_item_response,
     basic_batch_write_item_put_response,
     basic_delete_item_response,
-    exhaustive_delete_item_response,
+    basic_execute_statement_response,
+    basic_execute_transaction_response,
     basic_get_item_response,
     basic_put_item_response,
     basic_query_response,
     basic_scan_response,
     basic_transact_get_items_response,
     basic_transact_write_items_response,
     basic_update_item_response,
-    exhaustive_update_item_response,
+    exhaustive_batch_execute_statement_response,
     exhaustive_batch_get_item_response,
     exhaustive_batch_write_item_put_response,
+    exhaustive_delete_item_response,
+    exhaustive_execute_statement_response,
     exhaustive_get_item_response,
     exhaustive_put_item_response,
     exhaustive_query_response,
     exhaustive_scan_response,
-    exhaustive_execute_statement_response,
-    basic_execute_statement_response,
-    basic_execute_transaction_response,
-    basic_batch_execute_statement_response,
-    exhaustive_batch_execute_statement_response,
+    exhaustive_update_item_response,
 )
 
 resource_to_client_converter = ResourceShapeToClientShapeConverter(table_name=INTEG_TEST_DEFAULT_DYNAMODB_TABLE_NAME)
@@ -942,7 +942,9 @@ def test_GIVEN_test_execute_statement_response_WHEN_resource_to_client_THEN_retu
     actual_ddb_response = resource_to_client_converter.execute_statement_response(response)
     # Then: Returns dict value
     actual_ddb_response = sort_attribute_list_of_dynamodb_json_lists(actual_ddb_response, "Items")
-    expected_ddb_response = sort_attribute_list_of_dynamodb_json_lists(test_execute_statement_response([test_ddb_item]), "Items")
+    expected_ddb_response = sort_attribute_list_of_dynamodb_json_lists(
+        test_execute_statement_response([test_ddb_item]), "Items"
+    )
 
     assert actual_ddb_response == expected_ddb_response
 
@@ -967,6 +969,7 @@ def test_GIVEN_test_execute_transaction_request_WHEN_resource_to_client_THEN_ret
 def test_execute_transaction_response():
     return basic_execute_transaction_response
 
+
 def test_GIVEN_test_execute_transaction_response_WHEN_resource_to_client_THEN_returns_dict_value(
     test_execute_transaction_response, test_ddb_item, test_dict_item
 ):