Version Count

Author: imabhichow

Examples/java/src/main/java/software/amazon/cryptography/example/DdbHelper.java

@@ -32,10 +32,13 @@ public static void createHappyCaseId(
     @Nonnull String kmsKeyArn,
     @Nonnull String branchKeyId,
     @Nonnull KeyStoreAdmin admin,
-    @Nonnull HierarchyVersion hierarchyVersion
+    @Nonnull HierarchyVersion hierarchyVersion,
+    @Nonnull Integer versionCount
   ) {
     CreateKeyExample.CreateKey(kmsKeyArn, branchKeyId, admin, hierarchyVersion);
-    VersionKeyExample.VersionKey(kmsKeyArn, branchKeyId, admin);
+    for (int i = 0; i < versionCount; i++) {
+      VersionKeyExample.VersionKey(kmsKeyArn, branchKeyId, admin);
+    }
   }
 
   public static boolean deleteKeyStoreDdbItem(

Examples/java/src/test/java/software/amazon/cryptography/example/hierarchy/mutations/MutationKmsAccessInFlightTestRunner.java

@@ -44,7 +44,8 @@ public static void createHappyCaseId(
       kmsKeyArn,
       branchKeyId,
       admin,
-      HierarchyVersion.v1
+      HierarchyVersion.v1,
+      1
     );
     if (hierarchyVersion == HierarchyVersion.v2) {
       Mutations mutations = Mutations