Merge branch 'main' into reuse-install_smithy_dafny_codegen_dependencies-actions

Author: rishav-karanjit

Examples/runtimes/java/DDBECwithSDKV2/build.gradle.kts

@@ -73,7 +73,7 @@ dependencies {
     implementation("software.amazon.awssdk:dynamodb")
     implementation("software.amazon.awssdk:dynamodb-enhanced")
     implementation("software.amazon.awssdk:kms")
-
+    
     testImplementation("org.testng:testng:7.5")
 }
 

Examples/runtimes/java/DDBECwithSDKV2/src/test/java/AsymmetricEncryptedItemTest.java

@@ -13,7 +13,7 @@ public class AsymmetricEncryptedItemTest {
   final String partitionKeyValue = "AsymmetricExample-" + UUID.randomUUID();
   final String sortKeyValue = "0";
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void testAsymmetricEncryption() throws Exception {
     try (final DynamoDbClient ddbClient = DynamoDbClient.create()) {
       AsymmetricEncryptedItem.encryptRecord(

Examples/runtimes/java/DDBECwithSDKV2/src/test/java/AwsKmsEncryptedItemTest.java

@@ -14,7 +14,7 @@ public class AwsKmsEncryptedItemTest {
   final String partitionKeyName = "partition_key";
   final String sortKeyName = "sort_key";
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void testAwsKmsEncryption() throws Exception {
     try (final DynamoDbClient ddbClient = DynamoDbClient.create()) {
       AwsKmsEncryptedItem.encryptRecord(

Examples/runtimes/java/DDBECwithSDKV2/src/test/java/RetryAnalyzer.java

@@ -0,0 +1,39 @@
+// Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import java.util.logging.Logger;
+import org.testng.IRetryAnalyzer;
+import org.testng.ITestResult;
+
+// RetryAnalyzer is intentionally duplicated across all examples project as exampels are intentionally self-contained
+// so users of the library can copy a single project without pulling in shared infrastructure.
+public class RetryAnalyzer implements IRetryAnalyzer {
+
+  private static final Logger logger = Logger.getLogger(
+    RetryAnalyzer.class.getName()
+  );
+  private int retryCount = 0;
+  private static final int maxRetryCount = 3;
+
+  @Override
+  public boolean retry(ITestResult result) {
+    if (retryCount < maxRetryCount) {
+      retryCount++;
+      logger.info(
+        String.format(
+          "Retrying test '%s' - attempt %d/%d",
+          result.getName(),
+          retryCount,
+          maxRetryCount
+        )
+      );
+      try {
+        Thread.sleep(5000L * (long) Math.pow(2, retryCount - 1)); // simple exponential backoff
+      } catch (InterruptedException e) {
+        Thread.currentThread().interrupt();
+      }
+      return true;
+    }
+    return false;
+  }
+}

Examples/runtimes/java/DDBECwithSDKV2/src/test/java/SymmetricEncryptedItemTest.java

@@ -16,7 +16,7 @@ public class SymmetricEncryptedItemTest {
   final String partitionKeyValue = "SymmetricExample-" + UUID.randomUUID();
   final String sortKeyValue = "0";
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void testSymmetricEncryption() throws Exception {
     // Generate random keys
     final SecureRandom secureRandom = new SecureRandom();

Examples/runtimes/java/DynamoDbEncryption/src/test/java/software/amazon/cryptography/examples/RetryAnalyzer.java

@@ -0,0 +1,41 @@
+// Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+package software.amazon.cryptography.examples;
+
+import java.util.logging.Logger;
+import org.testng.IRetryAnalyzer;
+import org.testng.ITestResult;
+
+// RetryAnalyzer is intentionally duplicated across all examples project as exampels are intentionally self-contained
+// so users of the library can copy a single project without pulling in shared infrastructure.
+public class RetryAnalyzer implements IRetryAnalyzer {
+
+  private static final Logger logger = Logger.getLogger(
+    RetryAnalyzer.class.getName()
+  );
+  private int retryCount = 0;
+  private static final int maxRetryCount = 3;
+
+  @Override
+  public boolean retry(ITestResult result) {
+    if (retryCount < maxRetryCount) {
+      retryCount++;
+      logger.info(
+        String.format(
+          "Retrying test '%s' - attempt %d/%d",
+          result.getName(),
+          retryCount,
+          maxRetryCount
+        )
+      );
+      try {
+        Thread.sleep(5000L * (long) Math.pow(2, retryCount - 1)); // simple exponential backoff
+      } catch (InterruptedException e) {
+        Thread.currentThread().interrupt();
+      }
+      return true;
+    }
+    return false;
+  }
+}

Examples/runtimes/java/DynamoDbEncryption/src/test/java/software/amazon/cryptography/examples/TestBasicPutGetExample.java

@@ -5,7 +5,7 @@
 
 public class TestBasicPutGetExample {
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void TestPutGet() {
     final String partitionKeyValue = "BasicPutGetExample" + UUID.randomUUID();
     BasicPutGetExample.PutItemGetItem(

Examples/runtimes/java/DynamoDbEncryption/src/test/java/software/amazon/cryptography/examples/TestCreateKeyStoreKeyExample.java

@@ -6,7 +6,7 @@
 
 public class TestCreateKeyStoreKeyExample {
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void TestCreateKeyStoreKeyExample() {
     String keyId = CreateKeyStoreKeyExample.KeyStoreCreateKey(
       TestUtils.TEST_KEYSTORE_NAME,

Examples/runtimes/java/DynamoDbEncryption/src/test/java/software/amazon/cryptography/examples/TestCreateKeyStoreTableExample.java

@@ -4,7 +4,7 @@
 
 public class TestCreateKeyStoreTableExample {
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void TestCreateKeyStoreTableExample() {
     CreateKeyStoreTableExample.KeyStoreCreateTable(
       TestUtils.TEST_KEYSTORE_NAME,

Examples/runtimes/java/DynamoDbEncryption/src/test/java/software/amazon/cryptography/examples/TestGetEncryptedDataKeyDescriptionExample.java

@@ -5,7 +5,7 @@
 
 public class TestGetEncryptedDataKeyDescriptionExample {
 
-  @Test
+  @Test(retryAnalyzer = RetryAnalyzer.class)
   public void TestGetEncryptedDataKeyDescription() {
     final String partitionKeyValue =
       "GetEncryptedDataKeyDescriptionExample" + UUID.randomUUID();