m

Author: Lucas McDonald

TestVectors/runtimes/python/src/aws_dbesdk_dynamodb_test_vectors/internaldafny/extern/CreateInterceptedDDBClient.py

@@ -5,21 +5,21 @@
 from aws_dbesdk_dynamodb.encrypted.client import EncryptedClient
 from smithy_dafny_standard_library.internaldafny.generated import Wrappers
 from aws_dbesdk_dynamodb.smithygenerated.aws_cryptography_dbencryptionsdk_dynamodb.errors import _smithy_error_to_dafny_error
-from aws_dbesdk_dynamodb_test_vectors.waiting_boto3_ddb_client import WaitingDynamoClient
+from aws_dbesdk_dynamodb_test_vectors.waiting_boto3_ddb_client import WaitingLocalDynamoClient
 
 class default__:
     @staticmethod
     def CreateVanillaDDBClient():
         try:
-            return aws_cryptography_internal_dynamodb.internaldafny.extern.Com_Amazonaws_Dynamodb.default__.DynamoDBClient(WaitingDynamoClient())
+            return aws_cryptography_internal_dynamodb.internaldafny.extern.Com_Amazonaws_Dynamodb.default__.DynamoDBClient(WaitingLocalDynamoClient())
         except Exception as e:
             return Wrappers.Result_Failure(_smithy_error_to_dafny_error(e))
 
     @staticmethod
     def CreateInterceptedDDBClient(dafny_encryption_config):
         try:
             native_encryption_config = aws_cryptography_dbencryptionsdk_dynamodb_DynamoDbTablesEncryptionConfig(dafny_encryption_config)
-            boto3_client = WaitingDynamoClient()
+            boto3_client = WaitingLocalDynamoClient()
             encrypted_client = EncryptedClient(client = boto3_client, encryption_config = native_encryption_config)
             return aws_cryptography_internal_dynamodb.internaldafny.extern.Com_Amazonaws_Dynamodb.default__.DynamoDBClient(encrypted_client)
         except Exception as e:

TestVectors/runtimes/python/src/aws_dbesdk_dynamodb_test_vectors/waiting_boto3_ddb_client.py

@@ -3,8 +3,8 @@
 class WaitingLocalDynamoClient:
     """
     boto3 DynamoDB client wrapper that wraps `create_table` and `delete_table` methods
-    and connects to localhost:8000.
-    If these methods are called on this client, they will block returning until
+        and connects to localhost:8000.
+    If overridden methods are called on this client, they will block returning until
         the table is created/deleted.
     This is the expected behavior of SDK clients in our Dafny code.
     All other methods besides these are unchanged and will call the boto3 client directly.
@@ -14,7 +14,6 @@ def __init__(self):
 
     def __getattr__(self, name):
         if hasattr(self._client, name):
-            print(f'calling underlyign client {name=}')
             original_method = getattr(self._client, name)
 
             if name == 'create_table':
@@ -46,4 +45,4 @@ def wrapper(*args, **kwargs):
             waiter.wait(TableName=table_name)
             print(f"Table '{table_name}' has been deleted.")
             return response
-        return wrapper
+        return wrapper