Add --local-service-test option and did polymorph

aws · Apr 1, 2024 · 0c638cb · 0c638cb
1 parent 5e6294a
commit 0c638cb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/SharedMakefile.mk b/SharedMakefile.mk
@@ -231,7 +231,7 @@ _polymorph:
 	$(patsubst %, --dependent-model $(PROJECT_ROOT)/%/Model, $($(service_deps_var))) \
 	--namespace $($(namespace_var)) \
 	$(AWS_SDK_CMD) \
-	$(OUTPUT_LOCAL_SERVICE) \
+	$(OUTPUT_LOCAL_SERVICE_$(SERVICE)) \
 	";
 
 # Generates all target runtime code for all namespaces in this project.

diff --git a/...software/amazon/cryptography/dbencryptionsdk/dynamodb/wrapped/TestDynamoDbEncryption.java b/...software/amazon/cryptography/dbencryptionsdk/dynamodb/wrapped/TestDynamoDbEncryption.java
@@ -13,6 +13,8 @@
 import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.CreateDynamoDbEncryptionBranchKeyIdSupplierInput;
 import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.CreateDynamoDbEncryptionBranchKeyIdSupplierOutput;
 import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.Error;
+import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.GetEncryptedDataKeyDescriptionInput;
+import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.GetEncryptedDataKeyDescriptionOutput;
 import software.amazon.cryptography.dbencryptionsdk.dynamodb.internaldafny.types.IDynamoDbEncryptionClient;
 
 public class TestDynamoDbEncryption implements IDynamoDbEncryptionClient {
@@ -38,6 +40,18 @@ public Result<CreateDynamoDbEncryptionBranchKeyIdSupplierOutput, Error> CreateDy
     }
   }
 
+  public Result<GetEncryptedDataKeyDescriptionOutput, Error> GetEncryptedDataKeyDescription(
+      GetEncryptedDataKeyDescriptionInput dafnyInput) {
+    software.amazon.cryptography.dbencryptionsdk.dynamodb.model.GetEncryptedDataKeyDescriptionInput nativeInput = ToNative.GetEncryptedDataKeyDescriptionInput(dafnyInput);
+    try {
+      software.amazon.cryptography.dbencryptionsdk.dynamodb.model.GetEncryptedDataKeyDescriptionOutput nativeOutput = this._impl.GetEncryptedDataKeyDescription(nativeInput);
+      GetEncryptedDataKeyDescriptionOutput dafnyOutput = ToDafny.GetEncryptedDataKeyDescriptionOutput(nativeOutput);
+      return Result.create_Success(dafnyOutput);
+    } catch (RuntimeException ex) {
+      return Result.create_Failure(ToDafny.Error(ex));
+    }
+  }
+
   public interface Builder {
     Builder impl(DynamoDbEncryption impl);